OSSClient.m 89 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206120712081209121012111212121312141215121612171218121912201221122212231224122512261227122812291230123112321233123412351236123712381239124012411242124312441245124612471248124912501251125212531254125512561257125812591260126112621263126412651266126712681269127012711272127312741275127612771278127912801281128212831284128512861287128812891290129112921293129412951296129712981299130013011302130313041305130613071308130913101311131213131314131513161317131813191320132113221323132413251326132713281329133013311332133313341335133613371338133913401341134213431344134513461347134813491350135113521353135413551356135713581359136013611362136313641365136613671368136913701371137213731374137513761377137813791380138113821383138413851386138713881389139013911392139313941395139613971398139914001401140214031404140514061407140814091410141114121413141414151416141714181419142014211422142314241425142614271428142914301431143214331434143514361437143814391440144114421443144414451446144714481449145014511452145314541455145614571458145914601461146214631464146514661467146814691470147114721473147414751476147714781479148014811482148314841485148614871488148914901491149214931494149514961497149814991500150115021503150415051506150715081509151015111512151315141515151615171518151915201521152215231524152515261527152815291530153115321533153415351536153715381539154015411542154315441545154615471548154915501551155215531554155515561557155815591560156115621563156415651566156715681569157015711572157315741575157615771578157915801581158215831584158515861587158815891590159115921593159415951596159715981599160016011602160316041605160616071608160916101611161216131614161516161617161816191620162116221623162416251626162716281629163016311632163316341635163616371638163916401641164216431644164516461647164816491650165116521653165416551656165716581659166016611662166316641665166616671668166916701671167216731674167516761677167816791680168116821683168416851686168716881689169016911692169316941695169616971698169917001701170217031704170517061707170817091710171117121713171417151716171717181719172017211722172317241725172617271728172917301731173217331734173517361737173817391740174117421743174417451746174717481749175017511752175317541755175617571758175917601761176217631764176517661767176817691770177117721773177417751776177717781779178017811782178317841785178617871788178917901791179217931794179517961797179817991800180118021803180418051806180718081809181018111812181318141815181618171818181918201821182218231824182518261827182818291830183118321833183418351836183718381839184018411842184318441845184618471848184918501851185218531854185518561857185818591860186118621863186418651866186718681869187018711872187318741875187618771878187918801881188218831884188518861887188818891890189118921893189418951896189718981899190019011902190319041905190619071908190919101911191219131914191519161917191819191920192119221923192419251926192719281929193019311932193319341935193619371938193919401941194219431944194519461947194819491950195119521953195419551956195719581959196019611962196319641965196619671968196919701971197219731974197519761977197819791980198119821983198419851986198719881989199019911992199319941995199619971998199920002001200220032004200520062007200820092010201120122013201420152016201720182019202020212022202320242025202620272028202920302031203220332034203520362037203820392040204120422043204420452046204720482049205020512052205320542055205620572058205920602061206220632064206520662067
  1. //
  2. // OSSClient.m
  3. // oss_ios_sdk
  4. //
  5. // Created by zhouzhuo on 8/16/15.
  6. // Copyright (c) 2015 aliyun.com. All rights reserved.
  7. //
  8. #import "OSSClient.h"
  9. #import "OSSDefine.h"
  10. #import "OSSModel.h"
  11. #import "OSSUtil.h"
  12. #import "OSSLog.h"
  13. #import "OSSBolts.h"
  14. #import "OSSNetworking.h"
  15. #import "OSSXMLDictionary.h"
  16. #import "OSSReachabilityManager.h"
  17. #import "OSSIPv6Adapter.h"
  18. #import "OSSNetworkingRequestDelegate.h"
  19. #import "OSSAllRequestNeededMessage.h"
  20. #import "OSSURLRequestRetryHandler.h"
  21. #import "OSSHttpResponseParser.h"
  22. #import "OSSGetObjectACLRequest.h"
  23. #import "OSSDeleteMultipleObjectsRequest.h"
  24. #import "OSSGetBucketInfoRequest.h"
  25. #import "OSSPutSymlinkRequest.h"
  26. #import "OSSGetSymlinkRequest.h"
  27. #import "OSSRestoreObjectRequest.h"
  28. static NSString * const kClientRecordNameWithCommonPrefix = @"oss_partInfos_storage_name";
  29. static NSString * const kClientRecordNameWithCRC64Suffix = @"-crc64";
  30. static NSString * const kClientRecordNameWithSequentialSuffix = @"-sequential";
  31. static NSUInteger const kClientMaximumOfChunks = 5000; //max part number
  32. static NSString * const kClientErrorMessageForEmptyFile = @"the length of file should not be 0!";
  33. static NSString * const kClientErrorMessageForCancelledTask = @"This task has been cancelled!";
  34. /**
  35. * extend OSSRequest to include the ref to networking request object
  36. */
  37. @interface OSSRequest ()
  38. @property (nonatomic, strong) OSSNetworkingRequestDelegate * requestDelegate;
  39. @end
  40. @interface OSSClient()
  41. - (void)enableCRC64WithFlag:(OSSRequestCRCFlag)flag requestDelegate:(OSSNetworkingRequestDelegate *)delegate;
  42. - (OSSTask *)preChecksForRequest:(OSSMultipartUploadRequest *)request;
  43. - (void)checkRequestCrc64Setting:(OSSRequest *)request;
  44. - (OSSTask *)checkNecessaryParamsOfRequest:(OSSMultipartUploadRequest *)request;
  45. - (OSSTask *)checkPartSizeForRequest:(OSSMultipartUploadRequest *)request;
  46. - (NSUInteger)judgePartSizeForMultipartRequest:(OSSMultipartUploadRequest *)request fileSize:(unsigned long long)fileSize;
  47. - (unsigned long long)getSizeWithFilePath:(nonnull NSString *)filePath error:(NSError **)error;
  48. - (NSString *)readUploadIdForRequest:(OSSResumableUploadRequest *)request recordFilePath:(NSString **)recordFilePath sequential:(BOOL)sequential;
  49. - (NSMutableDictionary *)localPartInfosDictoryWithUploadId:(NSString *)uploadId;
  50. - (OSSTask *)persistencePartInfos:(NSDictionary *)partInfos withUploadId:(NSString *)uploadId;
  51. - (OSSTask *)checkFileSizeWithRequest:(OSSMultipartUploadRequest *)request;
  52. + (NSError *)cancelError;
  53. @end
  54. @implementation OSSClient
  55. static NSObject *lock;
  56. - (instancetype)initWithEndpoint:(NSString *)endpoint credentialProvider:(id<OSSCredentialProvider>)credentialProvider {
  57. return [self initWithEndpoint:endpoint credentialProvider:credentialProvider clientConfiguration:[OSSClientConfiguration new]];
  58. }
  59. - (instancetype)initWithEndpoint:(NSString *)endpoint
  60. credentialProvider:(id<OSSCredentialProvider>)credentialProvider
  61. clientConfiguration:(OSSClientConfiguration *)conf {
  62. if (self = [super init]) {
  63. if (!lock) {
  64. lock = [NSObject new];
  65. }
  66. // Monitor the network. If the network type is changed, recheck the IPv6 status.
  67. [OSSReachabilityManager shareInstance];
  68. NSOperationQueue * queue = [NSOperationQueue new];
  69. // using for resumable upload and compat old interface
  70. queue.maxConcurrentOperationCount = 3;
  71. _ossOperationExecutor = [OSSExecutor executorWithOperationQueue:queue];
  72. if (![endpoint oss_isNotEmpty]) {
  73. [NSException raise:NSInvalidArgumentException
  74. format:@"endpoint should not be nil or empty!"];
  75. }
  76. if ([endpoint rangeOfString:@"://"].location == NSNotFound) {
  77. endpoint = [@"https://" stringByAppendingString:endpoint];
  78. }
  79. NSURL *endpointURL = [NSURL URLWithString:endpoint];
  80. if ([endpointURL.scheme.lowercaseString isEqualToString:@"https"]) {
  81. if ([[OSSIPv6Adapter getInstance] isIPv4Address: endpointURL.host] || [[OSSIPv6Adapter getInstance] isIPv6Address: endpointURL.host]) {
  82. [NSException raise:NSInvalidArgumentException
  83. format:@"unsupported format of endpoint, please use right endpoint format!"];
  84. }
  85. }
  86. self.endpoint = [endpoint oss_trim];
  87. self.credentialProvider = credentialProvider;
  88. self.clientConfiguration = conf;
  89. OSSNetworkingConfiguration * netConf = [OSSNetworkingConfiguration new];
  90. if (conf) {
  91. netConf.maxRetryCount = conf.maxRetryCount;
  92. netConf.timeoutIntervalForRequest = conf.timeoutIntervalForRequest;
  93. netConf.timeoutIntervalForResource = conf.timeoutIntervalForResource;
  94. netConf.enableBackgroundTransmitService = conf.enableBackgroundTransmitService;
  95. netConf.backgroundSessionIdentifier = conf.backgroundSesseionIdentifier;
  96. netConf.proxyHost = conf.proxyHost;
  97. netConf.proxyPort = conf.proxyPort;
  98. netConf.maxConcurrentRequestCount = conf.maxConcurrentRequestCount;
  99. }
  100. self.networking = [[OSSNetworking alloc] initWithConfiguration:netConf];
  101. }
  102. return self;
  103. }
  104. - (OSSTask *)invokeRequest:(OSSNetworkingRequestDelegate *)request requireAuthentication:(BOOL)requireAuthentication {
  105. /* if content-type haven't been set, we set one */
  106. if (!request.allNeededMessage.contentType.oss_isNotEmpty
  107. && ([request.allNeededMessage.httpMethod isEqualToString:@"POST"] || [request.allNeededMessage.httpMethod isEqualToString:@"PUT"])) {
  108. request.allNeededMessage.contentType = [OSSUtil detemineMimeTypeForFilePath:request.uploadingFileURL.path uploadName:request.allNeededMessage.objectKey];
  109. }
  110. // Checks if the endpoint is in the excluded CName list.
  111. [self.clientConfiguration.cnameExcludeList enumerateObjectsUsingBlock:^(NSString *exclude, NSUInteger idx, BOOL * _Nonnull stop) {
  112. if ([self.endpoint hasSuffix:exclude]) {
  113. request.allNeededMessage.isHostInCnameExcludeList = YES;
  114. *stop = YES;
  115. }
  116. }];
  117. id<OSSRequestInterceptor> uaSetting = [[OSSUASettingInterceptor alloc] initWithClientConfiguration:self.clientConfiguration];
  118. [request.interceptors addObject:uaSetting];
  119. /* check if the authentication is required */
  120. if (requireAuthentication) {
  121. id<OSSRequestInterceptor> signer = [[OSSSignerInterceptor alloc] initWithCredentialProvider:self.credentialProvider];
  122. [request.interceptors addObject:signer];
  123. }
  124. request.isHttpdnsEnable = self.clientConfiguration.isHttpdnsEnable;
  125. return [_networking sendRequest:request];
  126. }
  127. #pragma implement restful apis
  128. - (OSSTask *)getService:(OSSGetServiceRequest *)request {
  129. OSSNetworkingRequestDelegate * requestDelegate = request.requestDelegate;
  130. requestDelegate.responseParser = [[OSSHttpResponseParser alloc] initForOperationType:OSSOperationTypeGetService];
  131. OSSAllRequestNeededMessage *neededMsg = [[OSSAllRequestNeededMessage alloc] init];
  132. neededMsg.endpoint = self.endpoint;
  133. neededMsg.httpMethod = OSSHTTPMethodGET;
  134. neededMsg.params = [request requestParams];
  135. requestDelegate.allNeededMessage = neededMsg;
  136. requestDelegate.operType = OSSOperationTypeGetService;
  137. return [self invokeRequest:requestDelegate requireAuthentication:request.isAuthenticationRequired];
  138. }
  139. # pragma mark - Private Methods
  140. - (void)enableCRC64WithFlag:(OSSRequestCRCFlag)flag requestDelegate:(OSSNetworkingRequestDelegate *)delegate
  141. {
  142. switch (flag) {
  143. case OSSRequestCRCOpen:
  144. delegate.crc64Verifiable = YES;
  145. break;
  146. case OSSRequestCRCClosed:
  147. delegate.crc64Verifiable = NO;
  148. break;
  149. default:
  150. delegate.crc64Verifiable = self.clientConfiguration.crc64Verifiable;
  151. break;
  152. }
  153. }
  154. - (OSSTask *)preChecksForRequest:(OSSMultipartUploadRequest *)request
  155. {
  156. OSSTask *preTask = [self checkFileSizeWithRequest:request];
  157. if (preTask) {
  158. return preTask;
  159. }
  160. preTask = [self checkNecessaryParamsOfRequest:request];
  161. if (preTask) {
  162. return preTask;
  163. }
  164. preTask = [self checkPartSizeForRequest:request];
  165. if (preTask) {
  166. return preTask;
  167. }
  168. return preTask;
  169. }
  170. - (void)checkRequestCrc64Setting:(OSSRequest *)request
  171. {
  172. if (request.crcFlag == OSSRequestCRCUninitialized)
  173. {
  174. if (self.clientConfiguration.crc64Verifiable)
  175. {
  176. request.crcFlag = OSSRequestCRCOpen;
  177. }else
  178. {
  179. request.crcFlag = OSSRequestCRCClosed;
  180. }
  181. }
  182. }
  183. - (OSSTask *)checkNecessaryParamsOfRequest:(OSSMultipartUploadRequest *)request
  184. {
  185. NSError *error = nil;
  186. if (![request.objectKey oss_isNotEmpty]) {
  187. error = [NSError errorWithDomain:OSSClientErrorDomain
  188. code:OSSClientErrorCodeInvalidArgument
  189. userInfo:@{OSSErrorMessageTOKEN: @"checkNecessaryParamsOfRequest requires nonnull objectKey!"}];
  190. }else if (![request.bucketName oss_isNotEmpty]) {
  191. error = [NSError errorWithDomain:OSSClientErrorDomain
  192. code:OSSClientErrorCodeInvalidArgument
  193. userInfo:@{OSSErrorMessageTOKEN: @"checkNecessaryParamsOfRequest requires nonnull bucketName!"}];
  194. }else if (![request.uploadingFileURL.path oss_isNotEmpty]) {
  195. error = [NSError errorWithDomain:OSSClientErrorDomain
  196. code:OSSClientErrorCodeInvalidArgument
  197. userInfo:@{OSSErrorMessageTOKEN: @"checkNecessaryParamsOfRequest requires nonnull uploadingFileURL!"}];
  198. }
  199. OSSTask *errorTask = nil;
  200. if (error) {
  201. errorTask = [OSSTask taskWithError:error];
  202. }
  203. return errorTask;
  204. }
  205. - (OSSTask *)checkPartSizeForRequest:(OSSMultipartUploadRequest *)request
  206. {
  207. OSSTask *errorTask = nil;
  208. unsigned long long fileSize = [self getSizeWithFilePath:request.uploadingFileURL.path error:nil];
  209. if (request.partSize == 0 || (fileSize > 102400 && request.partSize < 102400)) {
  210. NSError *error = [NSError errorWithDomain:OSSClientErrorDomain
  211. code:OSSClientErrorCodeInvalidArgument
  212. userInfo:@{OSSErrorMessageTOKEN: @"Part size must be greater than equal to 100KB"}];
  213. errorTask = [OSSTask taskWithError:error];
  214. }
  215. return errorTask;
  216. }
  217. - (NSUInteger)judgePartSizeForMultipartRequest:(OSSMultipartUploadRequest *)request fileSize:(unsigned long long)fileSize
  218. {
  219. #pragma clang diagnostic push
  220. #pragma clang diagnostic ignored "-Wshorten-64-to-32"
  221. BOOL divisible = (fileSize % request.partSize == 0);
  222. NSUInteger partCount = (fileSize / request.partSize) + (divisible? 0 : 1);
  223. if(partCount > kClientMaximumOfChunks)
  224. {
  225. request.partSize = fileSize / kClientMaximumOfChunks;
  226. partCount = kClientMaximumOfChunks;
  227. }
  228. return partCount;
  229. #pragma clang diagnostic pop
  230. }
  231. - (unsigned long long)getSizeWithFilePath:(nonnull NSString *)filePath error:(NSError **)error
  232. {
  233. NSFileManager *fm = [NSFileManager defaultManager];
  234. NSDictionary *attributes = [fm attributesOfItemAtPath:filePath error:error];
  235. return attributes.fileSize;
  236. }
  237. - (NSString *)readUploadIdForRequest:(OSSResumableUploadRequest *)request recordFilePath:(NSString **)recordFilePath sequential:(BOOL)sequential
  238. {
  239. NSString *uploadId = nil;
  240. NSString *record = [NSString stringWithFormat:@"%@%@%@%lu", request.md5String, request.bucketName, request.objectKey, (unsigned long)request.partSize];
  241. if (sequential) {
  242. record = [record stringByAppendingString:kClientRecordNameWithSequentialSuffix];
  243. }
  244. if (request.crcFlag == OSSRequestCRCOpen) {
  245. record = [record stringByAppendingString:kClientRecordNameWithCRC64Suffix];
  246. }
  247. NSData *data = [record dataUsingEncoding:NSUTF8StringEncoding];
  248. NSString *recordFileName = [OSSUtil dataMD5String:data];
  249. *recordFilePath = [request.recordDirectoryPath stringByAppendingPathComponent: recordFileName];
  250. NSFileManager *fileManager = [NSFileManager defaultManager];
  251. if ([fileManager fileExistsAtPath: *recordFilePath]) {
  252. NSFileHandle * read = [NSFileHandle fileHandleForReadingAtPath:*recordFilePath];
  253. uploadId = [[NSString alloc] initWithData:[read readDataToEndOfFile] encoding:NSUTF8StringEncoding];
  254. [read closeFile];
  255. } else {
  256. [fileManager createFileAtPath:*recordFilePath contents:nil attributes:nil];
  257. }
  258. return uploadId;
  259. }
  260. #pragma mark - sequential multipart upload
  261. - (NSMutableDictionary *)localPartInfosDictoryWithUploadId:(NSString *)uploadId
  262. {
  263. NSMutableDictionary *localPartInfoDict = nil;
  264. NSString *partInfosDirectory = [[NSString oss_documentDirectory] stringByAppendingPathComponent:kClientRecordNameWithCommonPrefix];
  265. NSString *partInfosPath = [partInfosDirectory stringByAppendingPathComponent:uploadId];
  266. BOOL isDirectory;
  267. NSFileManager *defaultFM = [NSFileManager defaultManager];
  268. if (!([defaultFM fileExistsAtPath:partInfosDirectory isDirectory:&isDirectory] && isDirectory))
  269. {
  270. if (![defaultFM createDirectoryAtPath:partInfosDirectory
  271. withIntermediateDirectories:NO
  272. attributes:nil error:nil]) {
  273. OSSLogError(@"create Directory(%@) failed!",partInfosDirectory);
  274. };
  275. }
  276. if (![defaultFM fileExistsAtPath:partInfosPath])
  277. {
  278. if (![defaultFM createFileAtPath:partInfosPath
  279. contents:nil
  280. attributes:nil])
  281. {
  282. OSSLogError(@"create local partInfo file failed!");
  283. }
  284. }
  285. localPartInfoDict = [[NSMutableDictionary alloc] initWithContentsOfURL:[NSURL fileURLWithPath:partInfosPath]];
  286. return localPartInfoDict;
  287. }
  288. - (OSSTask *)persistencePartInfos:(NSDictionary *)partInfos withUploadId:(NSString *)uploadId
  289. {
  290. NSString *filePath = [[[NSString oss_documentDirectory] stringByAppendingPathComponent:kClientRecordNameWithCommonPrefix] stringByAppendingPathComponent:uploadId];
  291. if (![partInfos writeToFile:filePath atomically:YES])
  292. {
  293. NSError *error = [NSError errorWithDomain:OSSClientErrorDomain
  294. code:OSSClientErrorCodeFileCantWrite
  295. userInfo:@{OSSErrorMessageTOKEN: @"uploadId for this task can't be stored persistentially!"}];
  296. OSSLogDebug(@"[Error]: %@", error);
  297. return [OSSTask taskWithError:error];
  298. }
  299. return nil;
  300. }
  301. - (OSSTask *)checkPutObjectFileURL:(OSSPutObjectRequest *)request {
  302. NSError *error = nil;
  303. if (!request.uploadingFileURL || ![request.uploadingFileURL.path oss_isNotEmpty]) {
  304. error = [NSError errorWithDomain:OSSClientErrorDomain
  305. code:OSSClientErrorCodeInvalidArgument
  306. userInfo:@{OSSErrorMessageTOKEN: @"Please check your request's uploadingFileURL!"}];
  307. } else {
  308. NSFileManager *dfm = [NSFileManager defaultManager];
  309. NSDictionary *attributes = [dfm attributesOfItemAtPath:request.uploadingFileURL.path error:&error];
  310. unsigned long long fileSize = [attributes[NSFileSize] unsignedLongLongValue];
  311. if (!error && fileSize == 0) {
  312. error = [NSError errorWithDomain:OSSClientErrorDomain
  313. code:OSSClientErrorCodeInvalidArgument
  314. userInfo:@{OSSErrorMessageTOKEN: kClientErrorMessageForEmptyFile}];
  315. }
  316. }
  317. if (error) {
  318. return [OSSTask taskWithError:error];
  319. } else {
  320. return [OSSTask taskWithResult:nil];
  321. }
  322. }
  323. - (OSSTask *)checkFileSizeWithRequest:(OSSMultipartUploadRequest *)request {
  324. NSError *error = nil;
  325. if (!request.uploadingFileURL || ![request.uploadingFileURL.path oss_isNotEmpty]) {
  326. error = [NSError errorWithDomain:OSSClientErrorDomain
  327. code:OSSClientErrorCodeInvalidArgument
  328. userInfo:@{OSSErrorMessageTOKEN: @"Please check your request's uploadingFileURL!"}];
  329. }
  330. else
  331. {
  332. NSFileManager *dfm = [NSFileManager defaultManager];
  333. NSDictionary *attributes = [dfm attributesOfItemAtPath:request.uploadingFileURL.path error:&error];
  334. unsigned long long fileSize = [attributes[NSFileSize] unsignedLongLongValue];
  335. if (!error && fileSize == 0) {
  336. error = [NSError errorWithDomain:OSSClientErrorDomain
  337. code:OSSClientErrorCodeInvalidArgument
  338. userInfo:@{OSSErrorMessageTOKEN: kClientErrorMessageForEmptyFile}];
  339. }
  340. }
  341. if (error) {
  342. return [OSSTask taskWithError:error];
  343. } else {
  344. return nil;
  345. }
  346. }
  347. + (NSError *)cancelError{
  348. static NSError *error = nil;
  349. static dispatch_once_t onceToken;
  350. dispatch_once(&onceToken, ^{
  351. error = [NSError errorWithDomain:OSSClientErrorDomain
  352. code:OSSClientErrorCodeTaskCancelled
  353. userInfo:@{OSSErrorMessageTOKEN: kClientErrorMessageForCancelledTask}];
  354. });
  355. return error;
  356. }
  357. - (void)dealloc{
  358. [self.networking.session invalidateAndCancel];
  359. }
  360. @end
  361. @implementation OSSClient (Bucket)
  362. - (OSSTask *)createBucket:(OSSCreateBucketRequest *)request {
  363. OSSNetworkingRequestDelegate *requestDelegate = request.requestDelegate;
  364. NSMutableDictionary *headerParams = [NSMutableDictionary dictionary];
  365. [headerParams oss_setObject:request.xOssACL forKey:OSSHttpHeaderBucketACL];
  366. if (request.location) {
  367. requestDelegate.uploadingData = [OSSUtil constructHttpBodyForCreateBucketWithLocation:request.location];
  368. }
  369. requestDelegate.responseParser = [[OSSHttpResponseParser alloc] initForOperationType:OSSOperationTypeCreateBucket];
  370. NSString *bodyString = [NSString stringWithFormat:@"<?xml version='1.0' encoding='UTF-8'?><CreateBucketConfiguration><StorageClass>%@</StorageClass></CreateBucketConfiguration>", request.storageClassAsString];
  371. requestDelegate.uploadingData = [bodyString dataUsingEncoding:NSUTF8StringEncoding];
  372. NSString *md5String = [OSSUtil base64Md5ForData:requestDelegate.uploadingData];
  373. OSSAllRequestNeededMessage *neededMsg = [[OSSAllRequestNeededMessage alloc] init];
  374. neededMsg.endpoint = self.endpoint;
  375. neededMsg.httpMethod = OSSHTTPMethodPUT;
  376. neededMsg.bucketName = request.bucketName;
  377. neededMsg.headerParams = headerParams;
  378. neededMsg.contentMd5 = md5String;
  379. requestDelegate.allNeededMessage = neededMsg;
  380. requestDelegate.operType = OSSOperationTypeCreateBucket;
  381. return [self invokeRequest:requestDelegate requireAuthentication:request.isAuthenticationRequired];
  382. }
  383. - (OSSTask *)deleteBucket:(OSSDeleteObjectRequest *)request {
  384. OSSNetworkingRequestDelegate * requestDelegate = request.requestDelegate;
  385. requestDelegate.responseParser = [[OSSHttpResponseParser alloc] initForOperationType:OSSOperationTypeDeleteBucket];
  386. OSSAllRequestNeededMessage *neededMsg = [[OSSAllRequestNeededMessage alloc] init];
  387. neededMsg.endpoint = self.endpoint;
  388. neededMsg.httpMethod = OSSHTTPMethodDELETE;
  389. neededMsg.bucketName = request.bucketName;
  390. requestDelegate.allNeededMessage = neededMsg;
  391. requestDelegate.operType = OSSOperationTypeDeleteBucket;
  392. return [self invokeRequest:requestDelegate requireAuthentication:request.isAuthenticationRequired];
  393. }
  394. - (OSSTask *)getBucket:(OSSGetBucketRequest *)request {
  395. OSSNetworkingRequestDelegate * requestDelegate = request.requestDelegate;
  396. requestDelegate.responseParser = [[OSSHttpResponseParser alloc] initForOperationType:OSSOperationTypeGetBucket];
  397. OSSAllRequestNeededMessage *neededMsg = [[OSSAllRequestNeededMessage alloc] init];
  398. neededMsg.endpoint = self.endpoint;
  399. neededMsg.httpMethod = OSSHTTPMethodGET;
  400. neededMsg.bucketName = request.bucketName;
  401. neededMsg.params = request.requestParams;
  402. requestDelegate.allNeededMessage = neededMsg;
  403. requestDelegate.operType = OSSOperationTypeGetBucket;
  404. return [self invokeRequest:requestDelegate requireAuthentication:request.isAuthenticationRequired];
  405. }
  406. - (OSSTask *)getBucketInfo:(OSSGetBucketInfoRequest *)request {
  407. OSSNetworkingRequestDelegate * requestDelegate = request.requestDelegate;
  408. requestDelegate.responseParser = [[OSSHttpResponseParser alloc] initForOperationType:OSSOperationTypeGetBucketInfo];
  409. OSSAllRequestNeededMessage *neededMsg = [[OSSAllRequestNeededMessage alloc] init];
  410. neededMsg.endpoint = self.endpoint;
  411. neededMsg.httpMethod = OSSHTTPMethodGET;
  412. neededMsg.bucketName = request.bucketName;
  413. neededMsg.params = request.requestParams;
  414. requestDelegate.allNeededMessage = neededMsg;
  415. requestDelegate.operType = OSSOperationTypeGetBucketInfo;
  416. return [self invokeRequest:requestDelegate requireAuthentication:request.isAuthenticationRequired];
  417. }
  418. - (OSSTask *)getBucketACL:(OSSGetBucketACLRequest *)request {
  419. OSSNetworkingRequestDelegate * requestDelegate = request.requestDelegate;
  420. requestDelegate.responseParser = [[OSSHttpResponseParser alloc] initForOperationType:OSSOperationTypeGetBucketACL];
  421. OSSAllRequestNeededMessage *neededMsg = [[OSSAllRequestNeededMessage alloc] init];
  422. neededMsg.endpoint = self.endpoint;
  423. neededMsg.httpMethod = OSSHTTPMethodGET;
  424. neededMsg.bucketName = request.bucketName;
  425. neededMsg.params = request.requestParams;
  426. requestDelegate.allNeededMessage = neededMsg;
  427. requestDelegate.operType = OSSOperationTypeGetBucketACL;
  428. return [self invokeRequest:requestDelegate requireAuthentication:request.isAuthenticationRequired];
  429. }
  430. @end
  431. @implementation OSSClient (Object)
  432. - (OSSTask *)headObject:(OSSHeadObjectRequest *)request {
  433. OSSNetworkingRequestDelegate * requestDelegate = request.requestDelegate;
  434. requestDelegate.responseParser = [[OSSHttpResponseParser alloc] initForOperationType:OSSOperationTypeHeadObject];
  435. OSSAllRequestNeededMessage *neededMsg = [[OSSAllRequestNeededMessage alloc] init];
  436. neededMsg.endpoint = self.endpoint;
  437. neededMsg.httpMethod = OSSHTTPMethodHEAD;
  438. neededMsg.bucketName = request.bucketName;
  439. neededMsg.objectKey = request.objectKey;
  440. requestDelegate.allNeededMessage = neededMsg;
  441. requestDelegate.operType = OSSOperationTypeHeadObject;
  442. return [self invokeRequest:requestDelegate requireAuthentication:request.isAuthenticationRequired];
  443. }
  444. - (OSSTask *)getObject:(OSSGetObjectRequest *)request {
  445. OSSNetworkingRequestDelegate * requestDelegate = request.requestDelegate;
  446. NSString * rangeString = nil;
  447. if (request.range) {
  448. rangeString = [request.range toHeaderString];
  449. }
  450. if (request.downloadProgress) {
  451. requestDelegate.downloadProgress = request.downloadProgress;
  452. }
  453. if (request.onRecieveData) {
  454. requestDelegate.onRecieveData = request.onRecieveData;
  455. }
  456. NSMutableDictionary * params = [NSMutableDictionary dictionary];
  457. [params oss_setObject:request.xOssProcess forKey:OSSHttpQueryProcess];
  458. [self enableCRC64WithFlag:request.crcFlag requestDelegate:requestDelegate];
  459. OSSHttpResponseParser *responseParser = [[OSSHttpResponseParser alloc] initForOperationType:OSSOperationTypeGetObject];
  460. responseParser.crc64Verifiable = requestDelegate.crc64Verifiable;
  461. requestDelegate.responseParser = responseParser;
  462. requestDelegate.responseParser.downloadingFileURL = request.downloadToFileURL;
  463. OSSAllRequestNeededMessage *neededMsg = [[OSSAllRequestNeededMessage alloc] init];
  464. neededMsg.endpoint = self.endpoint;
  465. neededMsg.httpMethod = OSSHTTPMethodGET;
  466. neededMsg.bucketName = request.bucketName;
  467. neededMsg.objectKey = request.objectKey;
  468. neededMsg.range = rangeString;
  469. neededMsg.params = params;
  470. neededMsg.headerParams = request.headerFields;
  471. requestDelegate.allNeededMessage = neededMsg;
  472. requestDelegate.operType = OSSOperationTypeGetObject;
  473. return [self invokeRequest:requestDelegate requireAuthentication:request.isAuthenticationRequired];
  474. }
  475. - (OSSTask *)getObjectACL:(OSSGetObjectACLRequest *)request
  476. {
  477. OSSNetworkingRequestDelegate *requestDelegate = request.requestDelegate;
  478. requestDelegate.responseParser = [[OSSHttpResponseParser alloc] initForOperationType:OSSOperationTypeGetObjectACL];
  479. NSMutableDictionary *params = [NSMutableDictionary dictionary];
  480. [params oss_setObject:@"" forKey:@"acl"];
  481. OSSAllRequestNeededMessage *neededMsg = [[OSSAllRequestNeededMessage alloc] init];
  482. neededMsg.endpoint = self.endpoint;
  483. neededMsg.httpMethod = OSSHTTPMethodGET;
  484. neededMsg.bucketName = request.bucketName;
  485. neededMsg.objectKey = request.objectName;
  486. neededMsg.params = params;
  487. requestDelegate.allNeededMessage = neededMsg;
  488. requestDelegate.operType = OSSOperationTypeGetObjectACL;
  489. return [self invokeRequest:requestDelegate requireAuthentication:request.isAuthenticationRequired];
  490. }
  491. - (OSSTask *)putObject:(OSSPutObjectRequest *)request
  492. {
  493. OSSNetworkingRequestDelegate * requestDelegate = request.requestDelegate;
  494. NSMutableDictionary * headerParams = [NSMutableDictionary dictionaryWithDictionary:request.objectMeta];
  495. [self enableCRC64WithFlag:request.crcFlag requestDelegate:requestDelegate];
  496. if (request.uploadingData) {
  497. requestDelegate.uploadingData = request.uploadingData;
  498. if (requestDelegate.crc64Verifiable)
  499. {
  500. NSMutableData *mutableData = [NSMutableData dataWithData:request.uploadingData];
  501. requestDelegate.contentCRC = [NSString stringWithFormat:@"%llu",[mutableData oss_crc64]];
  502. }
  503. }
  504. if (request.uploadingFileURL) {
  505. OSSTask *checkIfEmptyTask = [self checkPutObjectFileURL:request];
  506. if (checkIfEmptyTask.error) {
  507. return checkIfEmptyTask;
  508. }
  509. requestDelegate.uploadingFileURL = request.uploadingFileURL;
  510. }
  511. if (request.uploadProgress) {
  512. requestDelegate.uploadProgress = request.uploadProgress;
  513. }
  514. if (request.uploadRetryCallback) {
  515. requestDelegate.retryCallback = request.uploadRetryCallback;
  516. }
  517. [headerParams oss_setObject:[request.callbackParam base64JsonString] forKey:OSSHttpHeaderXOSSCallback];
  518. [headerParams oss_setObject:[request.callbackVar base64JsonString] forKey:OSSHttpHeaderXOSSCallbackVar];
  519. [headerParams oss_setObject:request.contentDisposition forKey:OSSHttpHeaderContentDisposition];
  520. [headerParams oss_setObject:request.contentEncoding forKey:OSSHttpHeaderContentEncoding];
  521. [headerParams oss_setObject:request.expires forKey:OSSHttpHeaderExpires];
  522. [headerParams oss_setObject:request.cacheControl forKey:OSSHttpHeaderCacheControl];
  523. OSSHttpResponseParser *responseParser = [[OSSHttpResponseParser alloc] initForOperationType:OSSOperationTypePutObject];
  524. responseParser.crc64Verifiable = requestDelegate.crc64Verifiable;
  525. requestDelegate.responseParser = responseParser;
  526. OSSAllRequestNeededMessage *neededMsg = [[OSSAllRequestNeededMessage alloc] init];
  527. neededMsg.endpoint = self.endpoint;
  528. neededMsg.httpMethod = OSSHTTPMethodPUT;
  529. neededMsg.bucketName = request.bucketName;
  530. neededMsg.objectKey = request.objectKey;
  531. neededMsg.contentMd5 = request.contentMd5;
  532. neededMsg.contentType = request.contentType;
  533. neededMsg.headerParams = headerParams;
  534. neededMsg.contentSHA1 = request.contentSHA1;
  535. requestDelegate.allNeededMessage = neededMsg;
  536. requestDelegate.operType = OSSOperationTypePutObject;
  537. return [self invokeRequest:requestDelegate requireAuthentication:request.isAuthenticationRequired];
  538. }
  539. - (OSSTask *)putObjectACL:(OSSPutObjectACLRequest *)request {
  540. OSSNetworkingRequestDelegate * requestDelegate = request.requestDelegate;
  541. if (request.uploadRetryCallback) {
  542. requestDelegate.retryCallback = request.uploadRetryCallback;
  543. }
  544. NSMutableDictionary * headerParams = [NSMutableDictionary dictionary];
  545. [headerParams oss_setObject:request.acl forKey:OSSHttpHeaderObjectACL];
  546. NSMutableDictionary * params = [NSMutableDictionary dictionary];
  547. [params oss_setObject:@"" forKey:@"acl"];
  548. requestDelegate.responseParser = [[OSSHttpResponseParser alloc] initForOperationType:OSSOperationTypePutObjectACL];
  549. OSSAllRequestNeededMessage *neededMsg = [[OSSAllRequestNeededMessage alloc] init];
  550. neededMsg.endpoint = self.endpoint;
  551. neededMsg.httpMethod = OSSHTTPMethodPUT;
  552. neededMsg.bucketName = request.bucketName;
  553. neededMsg.objectKey = request.objectKey;
  554. neededMsg.params = params;
  555. neededMsg.headerParams = headerParams;
  556. requestDelegate.allNeededMessage = neededMsg;
  557. requestDelegate.operType = OSSOperationTypePutObjectACL;
  558. return [self invokeRequest:requestDelegate requireAuthentication:request.isAuthenticationRequired];
  559. }
  560. - (OSSTask *)appendObject:(OSSAppendObjectRequest *)request
  561. {
  562. return [self appendObject:request withCrc64ecma:nil];
  563. }
  564. - (OSSTask *)appendObject:(OSSAppendObjectRequest *)request withCrc64ecma:(nullable NSString *)crc64ecma
  565. {
  566. OSSNetworkingRequestDelegate * requestDelegate = request.requestDelegate;
  567. requestDelegate.lastCRC = crc64ecma;
  568. [self enableCRC64WithFlag:request.crcFlag requestDelegate:requestDelegate];
  569. if (request.uploadingData)
  570. {
  571. requestDelegate.uploadingData = request.uploadingData;
  572. if (requestDelegate.crc64Verifiable)
  573. {
  574. NSMutableData *mutableData = [NSMutableData dataWithData:request.uploadingData];
  575. requestDelegate.contentCRC = [NSString stringWithFormat:@"%llu",[mutableData oss_crc64]];
  576. }
  577. }
  578. if (request.uploadingFileURL) {
  579. requestDelegate.uploadingFileURL = request.uploadingFileURL;
  580. }
  581. if (request.uploadProgress) {
  582. requestDelegate.uploadProgress = request.uploadProgress;
  583. }
  584. NSMutableDictionary * headerParams = [NSMutableDictionary dictionaryWithDictionary:request.objectMeta];
  585. [headerParams oss_setObject:request.contentDisposition forKey:OSSHttpHeaderContentDisposition];
  586. [headerParams oss_setObject:request.contentEncoding forKey:OSSHttpHeaderContentEncoding];
  587. [headerParams oss_setObject:request.expires forKey:OSSHttpHeaderExpires];
  588. [headerParams oss_setObject:request.cacheControl forKey:OSSHttpHeaderCacheControl];
  589. NSMutableDictionary* params = [NSMutableDictionary dictionary];
  590. [params oss_setObject:@"" forKey:@"append"];
  591. [params oss_setObject:[@(request.appendPosition) stringValue] forKey:@"position"];
  592. OSSHttpResponseParser *responseParser = [[OSSHttpResponseParser alloc] initForOperationType:OSSOperationTypeAppendObject];
  593. responseParser.crc64Verifiable = requestDelegate.crc64Verifiable;
  594. requestDelegate.responseParser = responseParser;
  595. OSSAllRequestNeededMessage *neededMsg = [[OSSAllRequestNeededMessage alloc] init];
  596. neededMsg.endpoint = self.endpoint;
  597. neededMsg.httpMethod = OSSHTTPMethodPOST;
  598. neededMsg.bucketName = request.bucketName;
  599. neededMsg.objectKey = request.objectKey;
  600. neededMsg.contentType = request.contentType;
  601. neededMsg.contentMd5 = request.contentMd5;
  602. neededMsg.headerParams = headerParams;
  603. neededMsg.params = params;
  604. neededMsg.contentSHA1 = request.contentSHA1;
  605. requestDelegate.allNeededMessage = neededMsg;
  606. requestDelegate.operType = OSSOperationTypeAppendObject;
  607. return [self invokeRequest:requestDelegate requireAuthentication:request.isAuthenticationRequired];
  608. }
  609. - (OSSTask *)deleteObject:(OSSDeleteObjectRequest *)request {
  610. OSSNetworkingRequestDelegate * requestDelegate = request.requestDelegate;
  611. requestDelegate.responseParser = [[OSSHttpResponseParser alloc] initForOperationType:OSSOperationTypePutObject];
  612. OSSAllRequestNeededMessage *neededMsg = [[OSSAllRequestNeededMessage alloc] init];
  613. neededMsg.endpoint = self.endpoint;
  614. neededMsg.httpMethod = OSSHTTPMethodDELETE;
  615. neededMsg.bucketName = request.bucketName;
  616. neededMsg.objectKey = request.objectKey;
  617. requestDelegate.allNeededMessage = neededMsg;
  618. requestDelegate.operType = OSSOperationTypeDeleteObject;
  619. return [self invokeRequest:requestDelegate requireAuthentication:request.isAuthenticationRequired];
  620. }
  621. - (OSSTask *)deleteMultipleObjects:(OSSDeleteMultipleObjectsRequest *)request
  622. {
  623. if ([request.keys count] == 0) {
  624. NSError *error = [NSError errorWithDomain:OSSClientErrorDomain
  625. code:OSSClientErrorCodeInvalidArgument
  626. userInfo:@{OSSErrorMessageTOKEN: @"keys should not be empty"}];
  627. return [OSSTask taskWithError:error];
  628. }
  629. OSSNetworkingRequestDelegate * requestDelegate = request.requestDelegate;
  630. requestDelegate.uploadingData = [OSSUtil constructHttpBodyForDeleteMultipleObjects:request.keys quiet:request.quiet];
  631. requestDelegate.responseParser = [[OSSHttpResponseParser alloc] initForOperationType:OSSOperationTypeDeleteMultipleObjects];
  632. NSString *md5String = [OSSUtil base64Md5ForData:requestDelegate.uploadingData];
  633. NSMutableDictionary *params = [NSMutableDictionary dictionary];
  634. [params oss_setObject:@"" forKey:@"delete"];
  635. [params oss_setObject:request.encodingType forKey:@"encoding-type"];
  636. OSSAllRequestNeededMessage *neededMsg = [[OSSAllRequestNeededMessage alloc] init];
  637. neededMsg.endpoint = self.endpoint;
  638. neededMsg.httpMethod = OSSHTTPMethodPOST;
  639. neededMsg.bucketName = request.bucketName;
  640. neededMsg.contentMd5 = md5String;
  641. neededMsg.params = params;
  642. requestDelegate.allNeededMessage = neededMsg;
  643. requestDelegate.operType = OSSOperationTypeDeleteMultipleObjects;
  644. return [self invokeRequest:requestDelegate requireAuthentication:request.isAuthenticationRequired];
  645. }
  646. - (OSSTask *)copyObject:(OSSCopyObjectRequest *)request {
  647. NSString *copySourceHeader = nil;
  648. if (request.sourceCopyFrom) {
  649. copySourceHeader = request.sourceCopyFrom;
  650. } else {
  651. if (![request.sourceBucketName oss_isNotEmpty]) {
  652. NSError *error = [NSError errorWithDomain:OSSClientErrorDomain code:OSSClientErrorCodeInvalidArgument userInfo:@{NSLocalizedDescriptionKey: @"sourceBucketName should not be empty!"}];
  653. return [OSSTask taskWithError:error];
  654. }
  655. if (![request.sourceObjectKey oss_isNotEmpty]) {
  656. NSError *error = [NSError errorWithDomain:OSSClientErrorDomain code:OSSClientErrorCodeInvalidArgument userInfo:@{NSLocalizedDescriptionKey: @"sourceObjectKey should not be empty!"}];
  657. return [OSSTask taskWithError:error];
  658. }
  659. copySourceHeader = [NSString stringWithFormat:@"/%@/%@",request.bucketName, request.sourceObjectKey.oss_urlEncodedString];
  660. }
  661. OSSNetworkingRequestDelegate * requestDelegate = request.requestDelegate;
  662. NSMutableDictionary * headerParams = [NSMutableDictionary dictionaryWithDictionary:request.objectMeta];
  663. [headerParams oss_setObject:copySourceHeader forKey:OSSHttpHeaderCopySource];
  664. requestDelegate.responseParser = [[OSSHttpResponseParser alloc] initForOperationType:OSSOperationTypeCopyObject];
  665. OSSAllRequestNeededMessage *neededMsg = [[OSSAllRequestNeededMessage alloc] init];
  666. neededMsg.endpoint = self.endpoint;
  667. neededMsg.httpMethod = OSSHTTPMethodPUT;
  668. neededMsg.bucketName = request.bucketName;
  669. neededMsg.objectKey = request.objectKey;
  670. neededMsg.contentType = request.contentType;
  671. neededMsg.contentMd5 = request.contentMd5;
  672. neededMsg.headerParams = headerParams;
  673. neededMsg.contentSHA1 = request.contentSHA1;
  674. requestDelegate.allNeededMessage = neededMsg;
  675. requestDelegate.operType = OSSOperationTypeCopyObject;
  676. return [self invokeRequest:requestDelegate requireAuthentication:request.isAuthenticationRequired];
  677. }
  678. - (OSSTask *)putSymlink:(OSSPutSymlinkRequest *)request {
  679. OSSNetworkingRequestDelegate * requestDelegate = request.requestDelegate;
  680. requestDelegate.responseParser = [[OSSHttpResponseParser alloc] initForOperationType:OSSOperationTypePutSymlink];
  681. NSMutableDictionary *headerFields = [NSMutableDictionary dictionary];
  682. [headerFields oss_setObject:[request.targetObjectName oss_urlEncodedString] forKey:OSSHttpHeaderSymlinkTarget];
  683. if (request.objectMeta) {
  684. [headerFields addEntriesFromDictionary:request.objectMeta];
  685. }
  686. OSSAllRequestNeededMessage *neededMsg = [[OSSAllRequestNeededMessage alloc] init];
  687. neededMsg.endpoint = self.endpoint;
  688. neededMsg.httpMethod = OSSHTTPMethodPUT;
  689. neededMsg.bucketName = request.bucketName;
  690. neededMsg.objectKey = request.objectKey;
  691. neededMsg.params = request.requestParams;
  692. neededMsg.headerParams = headerFields;
  693. requestDelegate.allNeededMessage = neededMsg;
  694. requestDelegate.operType = OSSOperationTypePutSymlink;
  695. return [self invokeRequest:requestDelegate requireAuthentication:request.isAuthenticationRequired];
  696. }
  697. - (OSSTask *)getSymlink:(OSSGetSymlinkRequest *)request {
  698. OSSNetworkingRequestDelegate * requestDelegate = request.requestDelegate;
  699. requestDelegate.responseParser = [[OSSHttpResponseParser alloc] initForOperationType:OSSOperationTypeGetSymlink];
  700. OSSAllRequestNeededMessage *neededMsg = [[OSSAllRequestNeededMessage alloc] init];
  701. neededMsg.endpoint = self.endpoint;
  702. neededMsg.httpMethod = OSSHTTPMethodGET;
  703. neededMsg.bucketName = request.bucketName;
  704. neededMsg.objectKey = request.objectKey;
  705. neededMsg.params = request.requestParams;
  706. requestDelegate.allNeededMessage = neededMsg;
  707. requestDelegate.operType = OSSOperationTypeGetSymlink;
  708. return [self invokeRequest:requestDelegate requireAuthentication:request.isAuthenticationRequired];
  709. }
  710. - (OSSTask *)restoreObject:(OSSRestoreObjectRequest *)request {
  711. OSSNetworkingRequestDelegate * requestDelegate = request.requestDelegate;
  712. requestDelegate.responseParser = [[OSSHttpResponseParser alloc] initForOperationType:OSSOperationTypeRestoreObject];
  713. OSSAllRequestNeededMessage *neededMsg = [[OSSAllRequestNeededMessage alloc] init];
  714. neededMsg.endpoint = self.endpoint;
  715. neededMsg.httpMethod = OSSHTTPMethodPOST;
  716. neededMsg.bucketName = request.bucketName;
  717. neededMsg.objectKey = request.objectKey;
  718. neededMsg.params = request.requestParams;
  719. requestDelegate.allNeededMessage = neededMsg;
  720. requestDelegate.operType = OSSOperationTypeRestoreObject;
  721. return [self invokeRequest:requestDelegate requireAuthentication:request.isAuthenticationRequired];
  722. }
  723. @end
  724. @implementation OSSClient (MultipartUpload)
  725. - (OSSTask *)listMultipartUploads:(OSSListMultipartUploadsRequest *)request {
  726. OSSNetworkingRequestDelegate * requestDelegate = request.requestDelegate;
  727. NSMutableDictionary *params = [[NSMutableDictionary alloc] initWithDictionary:[request requestParams]];
  728. [params oss_setObject:@"" forKey:@"uploads"];
  729. requestDelegate.responseParser = [[OSSHttpResponseParser alloc] initForOperationType:OSSOperationTypeListMultipartUploads];
  730. OSSAllRequestNeededMessage *neededMsg = [[OSSAllRequestNeededMessage alloc] init];
  731. neededMsg.endpoint = self.endpoint;
  732. neededMsg.httpMethod = OSSHTTPMethodGET;
  733. neededMsg.bucketName = request.bucketName;
  734. neededMsg.params = params;
  735. requestDelegate.allNeededMessage = neededMsg;
  736. requestDelegate.operType = OSSOperationTypeListMultipartUploads;
  737. return [self invokeRequest:requestDelegate requireAuthentication:request.isAuthenticationRequired];
  738. }
  739. - (OSSTask *)multipartUploadInit:(OSSInitMultipartUploadRequest *)request {
  740. OSSNetworkingRequestDelegate * requestDelegate = request.requestDelegate;
  741. NSMutableDictionary * headerParams = [NSMutableDictionary dictionaryWithDictionary:request.objectMeta];
  742. [headerParams oss_setObject:request.contentDisposition forKey:OSSHttpHeaderContentDisposition];
  743. [headerParams oss_setObject:request.contentEncoding forKey:OSSHttpHeaderContentEncoding];
  744. [headerParams oss_setObject:request.expires forKey:OSSHttpHeaderExpires];
  745. [headerParams oss_setObject:request.cacheControl forKey:OSSHttpHeaderCacheControl];
  746. NSMutableDictionary *params = [NSMutableDictionary dictionary];
  747. [params oss_setObject:@"" forKey:@"uploads"];
  748. if (request.sequential) {
  749. [params oss_setObject:@"" forKey:@"sequential"];
  750. }
  751. requestDelegate.responseParser = [[OSSHttpResponseParser alloc] initForOperationType:OSSOperationTypeInitMultipartUpload];
  752. OSSAllRequestNeededMessage *neededMsg = [[OSSAllRequestNeededMessage alloc] init];
  753. neededMsg.endpoint = self.endpoint;
  754. neededMsg.httpMethod = OSSHTTPMethodPOST;
  755. neededMsg.bucketName = request.bucketName;
  756. neededMsg.objectKey = request.objectKey;
  757. neededMsg.contentType = request.contentType;
  758. neededMsg.params = params;
  759. neededMsg.headerParams = headerParams;
  760. requestDelegate.allNeededMessage = neededMsg;
  761. requestDelegate.operType = OSSOperationTypeInitMultipartUpload;
  762. return [self invokeRequest:requestDelegate requireAuthentication:request.isAuthenticationRequired];
  763. }
  764. - (OSSTask *)uploadPart:(OSSUploadPartRequest *)request {
  765. OSSNetworkingRequestDelegate * requestDelegate = request.requestDelegate;
  766. NSMutableDictionary * params = [NSMutableDictionary dictionary];
  767. [params oss_setObject:[@(request.partNumber) stringValue] forKey:@"partNumber"];
  768. [params oss_setObject:request.uploadId forKey:@"uploadId"];
  769. [self enableCRC64WithFlag:request.crcFlag requestDelegate:requestDelegate];
  770. if (request.uploadPartData) {
  771. requestDelegate.uploadingData = request.uploadPartData;
  772. if (requestDelegate.crc64Verifiable)
  773. {
  774. NSMutableData *mutableData = [NSMutableData dataWithData:request.uploadPartData];
  775. requestDelegate.contentCRC = [NSString stringWithFormat:@"%llu",[mutableData oss_crc64]];
  776. }
  777. }
  778. if (request.uploadPartFileURL) {
  779. requestDelegate.uploadingFileURL = request.uploadPartFileURL;
  780. }
  781. if (request.uploadPartProgress) {
  782. requestDelegate.uploadProgress = request.uploadPartProgress;
  783. }
  784. OSSHttpResponseParser *responseParser = [[OSSHttpResponseParser alloc] initForOperationType:OSSOperationTypeUploadPart];
  785. responseParser.crc64Verifiable = requestDelegate.crc64Verifiable;
  786. requestDelegate.responseParser = responseParser;
  787. OSSAllRequestNeededMessage *neededMsg = [[OSSAllRequestNeededMessage alloc] init];
  788. neededMsg.endpoint = self.endpoint;
  789. neededMsg.httpMethod = OSSHTTPMethodPUT;
  790. neededMsg.bucketName = request.bucketName;
  791. neededMsg.objectKey = request.objectkey;
  792. neededMsg.contentMd5 = request.contentMd5;
  793. neededMsg.params = params;
  794. neededMsg.contentSHA1 = request.contentSHA1;
  795. requestDelegate.allNeededMessage = neededMsg;
  796. requestDelegate.operType = OSSOperationTypeUploadPart;
  797. return [self invokeRequest:requestDelegate requireAuthentication:request.isAuthenticationRequired];
  798. }
  799. - (OSSTask *)completeMultipartUpload:(OSSCompleteMultipartUploadRequest *)request
  800. {
  801. OSSNetworkingRequestDelegate * requestDelegate = request.requestDelegate;
  802. NSMutableDictionary * headerParams = [NSMutableDictionary dictionary];
  803. if (request.partInfos) {
  804. requestDelegate.uploadingData = [OSSUtil constructHttpBodyFromPartInfos:request.partInfos];
  805. }
  806. [headerParams oss_setObject:[request.callbackParam base64JsonString] forKey:OSSHttpHeaderXOSSCallback];
  807. [headerParams oss_setObject:[request.callbackVar base64JsonString] forKey:OSSHttpHeaderXOSSCallbackVar];
  808. if (request.completeMetaHeader) {
  809. [headerParams addEntriesFromDictionary:request.completeMetaHeader];
  810. }
  811. NSMutableDictionary * params = [NSMutableDictionary dictionaryWithObjectsAndKeys:request.uploadId, @"uploadId", nil];
  812. OSSHttpResponseParser *responseParser = [[OSSHttpResponseParser alloc] initForOperationType:OSSOperationTypeCompleteMultipartUpload];
  813. responseParser.crc64Verifiable = requestDelegate.crc64Verifiable;
  814. requestDelegate.responseParser = responseParser;
  815. OSSAllRequestNeededMessage *neededMsg = [[OSSAllRequestNeededMessage alloc] init];
  816. neededMsg.endpoint = self.endpoint;
  817. neededMsg.httpMethod = OSSHTTPMethodPOST;
  818. neededMsg.bucketName = request.bucketName;
  819. neededMsg.objectKey = request.objectKey;
  820. neededMsg.contentMd5 = request.contentMd5;
  821. neededMsg.headerParams = headerParams;
  822. neededMsg.params = params;
  823. neededMsg.contentSHA1 = request.contentSHA1;
  824. requestDelegate.allNeededMessage = neededMsg;
  825. requestDelegate.operType = OSSOperationTypeCompleteMultipartUpload;
  826. return [self invokeRequest:requestDelegate requireAuthentication:request.isAuthenticationRequired];
  827. }
  828. - (OSSTask *)listParts:(OSSListPartsRequest *)request {
  829. OSSNetworkingRequestDelegate * requestDelegate = request.requestDelegate;
  830. NSMutableDictionary *params = [NSMutableDictionary dictionary];
  831. [params oss_setObject: request.uploadId forKey: @"uploadId"];
  832. [params oss_setObject: [NSString stringWithFormat:@"%d",request.partNumberMarker] forKey: @"part-number-marker"];
  833. requestDelegate.responseParser = [[OSSHttpResponseParser alloc] initForOperationType:OSSOperationTypeListMultipart];
  834. OSSAllRequestNeededMessage *neededMsg = [[OSSAllRequestNeededMessage alloc] init];
  835. neededMsg.endpoint = self.endpoint;
  836. neededMsg.httpMethod = OSSHTTPMethodGET;
  837. neededMsg.bucketName = request.bucketName;
  838. neededMsg.objectKey = request.objectKey;
  839. neededMsg.params = params;
  840. requestDelegate.allNeededMessage = neededMsg;
  841. requestDelegate.operType = OSSOperationTypeListMultipart;
  842. return [self invokeRequest:requestDelegate requireAuthentication:request.isAuthenticationRequired];
  843. }
  844. - (OSSTask *)abortMultipartUpload:(OSSAbortMultipartUploadRequest *)request {
  845. OSSNetworkingRequestDelegate * requestDelegate = request.requestDelegate;
  846. NSMutableDictionary * params = [NSMutableDictionary dictionaryWithObjectsAndKeys:request.uploadId, @"uploadId", nil];
  847. requestDelegate.responseParser = [[OSSHttpResponseParser alloc] initForOperationType:OSSOperationTypeAbortMultipartUpload];
  848. OSSAllRequestNeededMessage *neededMsg = [[OSSAllRequestNeededMessage alloc] init];
  849. neededMsg.endpoint = self.endpoint;
  850. neededMsg.httpMethod = OSSHTTPMethodDELETE;
  851. neededMsg.bucketName = request.bucketName;
  852. neededMsg.objectKey = request.objectKey;
  853. neededMsg.params = params;
  854. requestDelegate.allNeededMessage = neededMsg;
  855. requestDelegate.operType = OSSOperationTypeAbortMultipartUpload;
  856. return [self invokeRequest:requestDelegate requireAuthentication:request.isAuthenticationRequired];
  857. }
  858. - (OSSTask *)abortResumableMultipartUpload:(OSSResumableUploadRequest *)request
  859. {
  860. return [self abortMultipartUpload:request sequential:NO resumable:YES];
  861. }
  862. - (OSSTask *)abortMultipartUpload:(OSSMultipartUploadRequest *)request sequential:(BOOL)sequential resumable:(BOOL)resumable {
  863. OSSTask *errorTask = nil;
  864. if(resumable) {
  865. OSSResumableUploadRequest *resumableRequest = (OSSResumableUploadRequest *)request;
  866. NSString *nameInfoString = [NSString stringWithFormat:@"%@%@%@%lu",request.md5String, resumableRequest.bucketName, resumableRequest.objectKey, (unsigned long)resumableRequest.partSize];
  867. if (sequential) {
  868. nameInfoString = [nameInfoString stringByAppendingString:kClientRecordNameWithSequentialSuffix];
  869. }
  870. if (request.crcFlag == OSSRequestCRCOpen) {
  871. nameInfoString = [nameInfoString stringByAppendingString:kClientRecordNameWithCRC64Suffix];
  872. }
  873. NSData *data = [nameInfoString dataUsingEncoding:NSUTF8StringEncoding];
  874. NSString *recordFileName = [OSSUtil dataMD5String:data];
  875. NSString *recordFilePath = [NSString stringWithFormat:@"%@/%@",resumableRequest.recordDirectoryPath,recordFileName];
  876. NSFileManager *fileManager = [NSFileManager defaultManager];
  877. NSString *partInfosFilePath = [[[NSString oss_documentDirectory] stringByAppendingPathComponent:kClientRecordNameWithCommonPrefix] stringByAppendingPathComponent:resumableRequest.uploadId];
  878. if([fileManager fileExistsAtPath:recordFilePath])
  879. {
  880. NSError *error;
  881. if (![fileManager removeItemAtPath:recordFilePath error:&error])
  882. {
  883. OSSLogDebug(@"[OSSSDKError]: %@", error);
  884. }
  885. }
  886. if ([fileManager fileExistsAtPath:partInfosFilePath]) {
  887. NSError *error;
  888. if (![fileManager removeItemAtPath:partInfosFilePath error:&error])
  889. {
  890. OSSLogDebug(@"[OSSSDKError]: %@", error);
  891. }
  892. }
  893. OSSAbortMultipartUploadRequest * abort = [OSSAbortMultipartUploadRequest new];
  894. abort.bucketName = request.bucketName;
  895. abort.objectKey = request.objectKey;
  896. if (request.uploadId) {
  897. abort.uploadId = request.uploadId;
  898. } else {
  899. abort.uploadId = [[NSString alloc] initWithData:[[NSFileHandle fileHandleForReadingAtPath:recordFilePath] readDataToEndOfFile] encoding:NSUTF8StringEncoding];
  900. }
  901. errorTask = [self abortMultipartUpload:abort];
  902. }else
  903. {
  904. OSSAbortMultipartUploadRequest * abort = [OSSAbortMultipartUploadRequest new];
  905. abort.bucketName = request.bucketName;
  906. abort.objectKey = request.objectKey;
  907. abort.uploadId = request.uploadId;
  908. errorTask = [self abortMultipartUpload:abort];
  909. }
  910. return errorTask;
  911. }
  912. - (OSSTask *)multipartUpload:(OSSMultipartUploadRequest *)request {
  913. return [self multipartUpload: request resumable: NO sequential: NO];
  914. }
  915. - (OSSTask *)processCompleteMultipartUpload:(OSSMultipartUploadRequest *)request partInfos:(NSArray<OSSPartInfo *> *)partInfos clientCrc64:(uint64_t)clientCrc64 recordFilePath:(NSString *)recordFilePath localPartInfosPath:(NSString *)localPartInfosPath
  916. {
  917. OSSCompleteMultipartUploadRequest * complete = [OSSCompleteMultipartUploadRequest new];
  918. complete.bucketName = request.bucketName;
  919. complete.objectKey = request.objectKey;
  920. complete.uploadId = request.uploadId;
  921. complete.partInfos = partInfos;
  922. complete.crcFlag = request.crcFlag;
  923. complete.contentSHA1 = request.contentSHA1;
  924. if (request.completeMetaHeader != nil) {
  925. complete.completeMetaHeader = request.completeMetaHeader;
  926. }
  927. if (request.callbackParam != nil) {
  928. complete.callbackParam = request.callbackParam;
  929. }
  930. if (request.callbackVar != nil) {
  931. complete.callbackVar = request.callbackVar;
  932. }
  933. OSSTask * completeTask = [self completeMultipartUpload:complete];
  934. [completeTask waitUntilFinished];
  935. if (completeTask.error) {
  936. OSSLogVerbose(@"completeTask.error %@: ",completeTask.error);
  937. return completeTask;
  938. } else
  939. {
  940. if(recordFilePath && [[NSFileManager defaultManager] fileExistsAtPath:recordFilePath])
  941. {
  942. NSError *deleteError;
  943. if (![[NSFileManager defaultManager] removeItemAtPath:recordFilePath error:&deleteError])
  944. {
  945. OSSLogError(@"delete localUploadIdPath failed!Error: %@",deleteError);
  946. }
  947. }
  948. if (localPartInfosPath && [[NSFileManager defaultManager] fileExistsAtPath:localPartInfosPath])
  949. {
  950. NSError *deleteError;
  951. if (![[NSFileManager defaultManager] removeItemAtPath:localPartInfosPath error:&deleteError])
  952. {
  953. OSSLogError(@"delete localPartInfosPath failed!Error: %@",deleteError);
  954. }
  955. }
  956. OSSCompleteMultipartUploadResult * completeResult = completeTask.result;
  957. if (complete.crcFlag == OSSRequestCRCOpen && completeResult.remoteCRC64ecma)
  958. {
  959. uint64_t remote_crc64 = 0;
  960. NSScanner *scanner = [NSScanner scannerWithString:completeResult.remoteCRC64ecma];
  961. if ([scanner scanUnsignedLongLong:&remote_crc64])
  962. {
  963. OSSLogVerbose(@"resumableUpload local_crc64 %llu",clientCrc64);
  964. OSSLogVerbose(@"resumableUpload remote_crc64 %llu", remote_crc64);
  965. if (remote_crc64 != clientCrc64)
  966. {
  967. NSString *errorMessage = [NSString stringWithFormat:@"local_crc64(%llu) is not equal to remote_crc64(%llu)!",clientCrc64,remote_crc64];
  968. NSError *error = [NSError errorWithDomain:OSSClientErrorDomain
  969. code:OSSClientErrorCodeInvalidCRC
  970. userInfo:@{OSSErrorMessageTOKEN:errorMessage}];
  971. return [OSSTask taskWithError:error];
  972. }
  973. }
  974. }
  975. OSSResumableUploadResult * result = [OSSResumableUploadResult new];
  976. result.requestId = completeResult.requestId;
  977. result.httpResponseCode = completeResult.httpResponseCode;
  978. result.httpResponseHeaderFields = completeResult.httpResponseHeaderFields;
  979. result.serverReturnJsonString = completeResult.serverReturnJsonString;
  980. result.remoteCRC64ecma = completeResult.remoteCRC64ecma;
  981. return [OSSTask taskWithResult:result];
  982. }
  983. }
  984. - (OSSTask *)resumableUpload:(OSSResumableUploadRequest *)request
  985. {
  986. return [self multipartUpload: request resumable: YES sequential: NO];
  987. }
  988. - (OSSTask *)processListPartsWithObjectKey:(nonnull NSString *)objectKey bucket:(nonnull NSString *)bucket uploadId:(NSString * _Nonnull *)uploadId uploadedParts:(nonnull NSMutableArray *)uploadedParts uploadedLength:(NSUInteger *)uploadedLength totalSize:(unsigned long long)totalSize partSize:(NSUInteger)partSize
  989. {
  990. BOOL isTruncated = NO;
  991. int nextPartNumberMarker = 0;
  992. do {
  993. OSSListPartsRequest * listParts = [OSSListPartsRequest new];
  994. listParts.bucketName = bucket;
  995. listParts.objectKey = objectKey;
  996. listParts.uploadId = *uploadId;
  997. listParts.partNumberMarker = nextPartNumberMarker;
  998. OSSTask * listPartsTask = [self listParts:listParts];
  999. [listPartsTask waitUntilFinished];
  1000. if (listPartsTask.error)
  1001. {
  1002. isTruncated = NO;
  1003. [uploadedParts removeAllObjects];
  1004. if ([listPartsTask.error.domain isEqualToString: OSSServerErrorDomain] && labs(listPartsTask.error.code) == 404)
  1005. {
  1006. OSSLogVerbose(@"local record existes but the remote record is deleted");
  1007. *uploadId = nil;
  1008. } else
  1009. {
  1010. return listPartsTask;
  1011. }
  1012. }
  1013. else
  1014. {
  1015. OSSListPartsResult *res = listPartsTask.result;
  1016. isTruncated = res.isTruncated;
  1017. nextPartNumberMarker = res.nextPartNumberMarker;
  1018. OSSLogVerbose(@"resumableUpload listpart ok");
  1019. if (res.parts.count > 0) {
  1020. [uploadedParts addObjectsFromArray:res.parts];
  1021. }
  1022. }
  1023. } while (isTruncated);
  1024. __block NSUInteger firstPartSize = 0;
  1025. __block NSUInteger bUploadedLength = 0;
  1026. [uploadedParts enumerateObjectsUsingBlock:^(NSDictionary *part, NSUInteger idx, BOOL * _Nonnull stop) {
  1027. unsigned long long iPartSize = 0;
  1028. NSString *partSizeString = [part objectForKey:OSSSizeXMLTOKEN];
  1029. NSScanner *scanner = [NSScanner scannerWithString:partSizeString];
  1030. [scanner scanUnsignedLongLong:&iPartSize];
  1031. #pragma clang diagnostic push
  1032. #pragma clang diagnostic ignored "-Wshorten-64-to-32"
  1033. bUploadedLength += iPartSize;
  1034. if (idx == 0)
  1035. {
  1036. firstPartSize = iPartSize;
  1037. }
  1038. #pragma clang diagnostic pop
  1039. }];
  1040. *uploadedLength = bUploadedLength;
  1041. if (totalSize < bUploadedLength)
  1042. {
  1043. NSError *error = [NSError errorWithDomain:OSSClientErrorDomain
  1044. code:OSSClientErrorCodeCannotResumeUpload
  1045. userInfo:@{OSSErrorMessageTOKEN: @"The uploading file is inconsistent with before"}];
  1046. return [OSSTask taskWithError: error];
  1047. }
  1048. else if (firstPartSize != 0 && firstPartSize != partSize && totalSize != firstPartSize)
  1049. {
  1050. NSError *error = [NSError errorWithDomain:OSSClientErrorDomain
  1051. code:OSSClientErrorCodeCannotResumeUpload
  1052. userInfo:@{OSSErrorMessageTOKEN: @"The part size setting is inconsistent with before"}];
  1053. return [OSSTask taskWithError: error];
  1054. }
  1055. return nil;
  1056. }
  1057. - (OSSTask *)processResumableInitMultipartUpload:(OSSInitMultipartUploadRequest *)request recordFilePath:(NSString *)recordFilePath
  1058. {
  1059. OSSTask *task = [self multipartUploadInit:request];
  1060. [task waitUntilFinished];
  1061. if(task.result && [recordFilePath oss_isNotEmpty])
  1062. {
  1063. OSSInitMultipartUploadResult *result = task.result;
  1064. if (![result.uploadId oss_isNotEmpty])
  1065. {
  1066. NSString *errorMessage = [NSString stringWithFormat:@"Can not get uploadId!"];
  1067. NSError *error = [NSError errorWithDomain:OSSServerErrorDomain
  1068. code:OSSClientErrorCodeNilUploadid userInfo:@{OSSErrorMessageTOKEN: errorMessage}];
  1069. return [OSSTask taskWithError:error];
  1070. }
  1071. NSFileManager *defaultFM = [NSFileManager defaultManager];
  1072. if (![defaultFM fileExistsAtPath:recordFilePath])
  1073. {
  1074. if (![defaultFM createFileAtPath:recordFilePath contents:nil attributes:nil]) {
  1075. NSError *error = [NSError errorWithDomain:OSSClientErrorDomain
  1076. code:OSSClientErrorCodeFileCantWrite
  1077. userInfo:@{OSSErrorMessageTOKEN: @"uploadId for this task can't be stored persistentially!"}];
  1078. OSSLogDebug(@"[Error]: %@", error);
  1079. return [OSSTask taskWithError:error];
  1080. }
  1081. }
  1082. NSFileHandle * write = [NSFileHandle fileHandleForWritingAtPath:recordFilePath];
  1083. [write writeData:[result.uploadId dataUsingEncoding:NSUTF8StringEncoding]];
  1084. [write closeFile];
  1085. }
  1086. return task;
  1087. }
  1088. - (OSSTask *)upload:(OSSMultipartUploadRequest *)request
  1089. uploadIndex:(NSMutableArray *)alreadyUploadIndex
  1090. uploadPart:(NSMutableArray *)alreadyUploadPart
  1091. count:(NSUInteger)partCout
  1092. uploadedLength:(NSUInteger *)uploadedLength
  1093. fileSize:(unsigned long long)uploadFileSize
  1094. {
  1095. NSOperationQueue *queue = [[NSOperationQueue alloc] init];
  1096. [queue setMaxConcurrentOperationCount: 5];
  1097. NSObject *localLock = [[NSObject alloc] init];
  1098. OSSRequestCRCFlag crcFlag = request.crcFlag;
  1099. __block OSSTask *errorTask;
  1100. __block NSMutableDictionary *localPartInfos = nil;
  1101. if (crcFlag == OSSRequestCRCOpen) {
  1102. localPartInfos = [self localPartInfosDictoryWithUploadId:request.uploadId];
  1103. }
  1104. if (!localPartInfos) {
  1105. localPartInfos = [NSMutableDictionary dictionary];
  1106. }
  1107. NSError *readError;
  1108. NSFileHandle *fileHande = [NSFileHandle fileHandleForReadingFromURL:request.uploadingFileURL error:&readError];
  1109. if (readError) {
  1110. return [OSSTask taskWithError: readError];
  1111. }
  1112. NSData * uploadPartData;
  1113. NSInteger realPartLength = request.partSize;
  1114. __block BOOL hasError = NO;
  1115. for (NSUInteger idx = 1; idx <= partCout; idx++)
  1116. {
  1117. if (request.isCancelled)
  1118. {
  1119. [queue cancelAllOperations];
  1120. break;
  1121. }
  1122. if ([alreadyUploadIndex containsObject:@(idx)])
  1123. {
  1124. continue;
  1125. }
  1126. // while operationCount >= 5,the loop will stay here
  1127. while (queue.operationCount >= 5) {
  1128. [NSThread sleepForTimeInterval: 0.15f];
  1129. }
  1130. if (idx == partCout) {
  1131. #pragma clang diagnostic push
  1132. #pragma clang diagnostic ignored "-Wshorten-64-to-32"
  1133. realPartLength = uploadFileSize - request.partSize * (idx - 1);
  1134. #pragma clang diagnostic pop
  1135. }
  1136. @autoreleasepool
  1137. {
  1138. [fileHande seekToFileOffset: request.partSize * (idx - 1)];
  1139. uploadPartData = [fileHande readDataOfLength:realPartLength];
  1140. NSBlockOperation *operation = [NSBlockOperation blockOperationWithBlock:^{
  1141. OSSTask *uploadPartErrorTask = nil;
  1142. [self executePartUpload:request
  1143. totalBytesExpectedToSend:uploadFileSize
  1144. totalBytesSent:uploadedLength
  1145. index:idx
  1146. partData:uploadPartData
  1147. alreadyUploadPart:alreadyUploadPart
  1148. localParts:localPartInfos
  1149. errorTask:&uploadPartErrorTask];
  1150. if (uploadPartErrorTask != nil) {
  1151. @synchronized(localLock) {
  1152. if (!hasError) {
  1153. hasError = YES;
  1154. errorTask = uploadPartErrorTask;
  1155. }
  1156. }
  1157. uploadPartErrorTask = nil;
  1158. }
  1159. }];
  1160. [queue addOperation:operation];
  1161. }
  1162. }
  1163. [fileHande closeFile];
  1164. [queue waitUntilAllOperationsAreFinished];
  1165. localLock = nil;
  1166. if (!errorTask && request.isCancelled) {
  1167. errorTask = [OSSTask taskWithError:[OSSClient cancelError]];
  1168. }
  1169. return errorTask;
  1170. }
  1171. - (void)executePartUpload:(OSSMultipartUploadRequest *)request totalBytesExpectedToSend:(unsigned long long)totalBytesExpectedToSend totalBytesSent:(NSUInteger *)totalBytesSent index:(NSUInteger)idx partData:(NSData *)partData alreadyUploadPart:(NSMutableArray *)uploadedParts localParts:(NSMutableDictionary *)localParts errorTask:(OSSTask **)errorTask
  1172. {
  1173. NSUInteger bytesSent = partData.length;
  1174. OSSUploadPartRequest * uploadPart = [OSSUploadPartRequest new];
  1175. uploadPart.bucketName = request.bucketName;
  1176. uploadPart.objectkey = request.objectKey;
  1177. #pragma clang diagnostic push
  1178. #pragma clang diagnostic ignored "-Wshorten-64-to-32"
  1179. uploadPart.partNumber = idx;
  1180. #pragma clang diagnostic pop
  1181. uploadPart.uploadId = request.uploadId;
  1182. uploadPart.uploadPartData = partData;
  1183. uploadPart.contentMd5 = [OSSUtil base64Md5ForData:partData];
  1184. uploadPart.crcFlag = request.crcFlag;
  1185. OSSTask * uploadPartTask = [self uploadPart:uploadPart];
  1186. [uploadPartTask waitUntilFinished];
  1187. if (uploadPartTask.error) {
  1188. if (labs(uploadPartTask.error.code) != 409) {
  1189. *errorTask = uploadPartTask;
  1190. }
  1191. } else {
  1192. OSSUploadPartResult * result = uploadPartTask.result;
  1193. OSSPartInfo * partInfo = [OSSPartInfo new];
  1194. #pragma clang diagnostic push
  1195. #pragma clang diagnostic ignored "-Wshorten-64-to-32"
  1196. partInfo.partNum = idx;
  1197. #pragma clang diagnostic pop
  1198. partInfo.eTag = result.eTag;
  1199. partInfo.size = bytesSent;
  1200. uint64_t crc64OfPart;
  1201. @try {
  1202. NSScanner *scanner = [NSScanner scannerWithString:result.remoteCRC64ecma];
  1203. [scanner scanUnsignedLongLong:&crc64OfPart];
  1204. partInfo.crc64 = crc64OfPart;
  1205. } @catch (NSException *exception) {
  1206. OSSLogError(@"multipart upload error with nil remote crc64!");
  1207. }
  1208. @synchronized(lock){
  1209. [uploadedParts addObject:partInfo];
  1210. if (request.crcFlag == OSSRequestCRCOpen)
  1211. {
  1212. [self processForLocalPartInfos:localParts
  1213. partInfo:partInfo
  1214. uploadId:request.uploadId];
  1215. [self persistencePartInfos:localParts
  1216. withUploadId:request.uploadId];
  1217. }
  1218. *totalBytesSent += bytesSent;
  1219. if (request.uploadProgress)
  1220. {
  1221. request.uploadProgress(bytesSent, *totalBytesSent, totalBytesExpectedToSend);
  1222. }
  1223. }
  1224. }
  1225. }
  1226. - (void)processForLocalPartInfos:(NSMutableDictionary *)localPartInfoDict partInfo:(OSSPartInfo *)partInfo uploadId:(NSString *)uploadId
  1227. {
  1228. NSDictionary *partInfoDict = [partInfo entityToDictionary];
  1229. NSString *keyString = [NSString stringWithFormat:@"%i",partInfo.partNum];
  1230. [localPartInfoDict oss_setObject:partInfoDict forKey:keyString];
  1231. }
  1232. - (OSSTask *)sequentialMultipartUpload:(OSSResumableUploadRequest *)request
  1233. {
  1234. return [self multipartUpload:request resumable:YES sequential:YES];
  1235. }
  1236. - (OSSTask *)multipartUpload:(OSSMultipartUploadRequest *)request resumable:(BOOL)resumable sequential:(BOOL)sequential
  1237. {
  1238. if (resumable) {
  1239. if (![request isKindOfClass:[OSSResumableUploadRequest class]]) {
  1240. NSError *typoError = [NSError errorWithDomain:OSSClientErrorDomain
  1241. code:OSSClientErrorCodeInvalidArgument
  1242. userInfo:@{OSSErrorMessageTOKEN: @"resumable multipart request should use instance of class OSSMultipartUploadRequest!"}];
  1243. return [OSSTask taskWithError: typoError];
  1244. }
  1245. }
  1246. [self checkRequestCrc64Setting:request];
  1247. OSSTask *preTask = [self preChecksForRequest:request];
  1248. if (preTask) {
  1249. return preTask;
  1250. }
  1251. return [[OSSTask taskWithResult:nil] continueWithExecutor:self.ossOperationExecutor withBlock:^id(OSSTask *task) {
  1252. __block NSUInteger uploadedLength = 0;
  1253. uploadedLength = 0;
  1254. __block OSSTask * errorTask;
  1255. __block NSString *uploadId;
  1256. NSError *error;
  1257. unsigned long long uploadFileSize = [self getSizeWithFilePath:request.uploadingFileURL.path error:&error];
  1258. if (error) {
  1259. return [OSSTask taskWithError:error];
  1260. }
  1261. NSUInteger partCount = [self judgePartSizeForMultipartRequest:request fileSize:uploadFileSize];
  1262. if (partCount > 1 && request.partSize < 102400) {
  1263. NSError *checkPartSizeError = [NSError errorWithDomain:OSSClientErrorDomain
  1264. code:OSSClientErrorCodeInvalidArgument
  1265. userInfo:@{OSSErrorMessageTOKEN: @"Part size must be greater than equal to 100KB"}];
  1266. return [OSSTask taskWithError:checkPartSizeError];
  1267. }
  1268. if (request.isCancelled) {
  1269. return [OSSTask taskWithError:[OSSClient cancelError]];
  1270. }
  1271. NSString *recordFilePath = nil;
  1272. NSMutableArray * uploadedPart = [NSMutableArray array];
  1273. NSString *localPartInfosPath = nil;
  1274. NSDictionary *localPartInfos = nil;
  1275. NSMutableArray<OSSPartInfo *> *uploadedPartInfos = [NSMutableArray array];
  1276. NSMutableArray * alreadyUploadIndex = [NSMutableArray array];
  1277. if (resumable) {
  1278. OSSResumableUploadRequest *resumableRequest = (OSSResumableUploadRequest *)request;
  1279. NSString *recordDirectoryPath = resumableRequest.recordDirectoryPath;
  1280. request.md5String = [OSSUtil fileMD5String:request.uploadingFileURL.path];
  1281. if ([recordDirectoryPath oss_isNotEmpty])
  1282. {
  1283. uploadId = [self readUploadIdForRequest:resumableRequest recordFilePath:&recordFilePath sequential:sequential];
  1284. OSSLogVerbose(@"local uploadId: %@,recordFilePath: %@",uploadId, recordFilePath);
  1285. }
  1286. if([uploadId oss_isNotEmpty])
  1287. {
  1288. localPartInfosPath = [[[NSString oss_documentDirectory] stringByAppendingPathComponent:kClientRecordNameWithCommonPrefix] stringByAppendingPathComponent:uploadId];
  1289. localPartInfos = [[NSDictionary alloc] initWithContentsOfFile:localPartInfosPath];
  1290. OSSTask *listPartTask = [self processListPartsWithObjectKey:request.objectKey
  1291. bucket:request.bucketName
  1292. uploadId:&uploadId
  1293. uploadedParts:uploadedPart
  1294. uploadedLength:&uploadedLength
  1295. totalSize:uploadFileSize
  1296. partSize:request.partSize];
  1297. if (listPartTask.error)
  1298. {
  1299. return listPartTask;
  1300. }
  1301. }
  1302. [uploadedPart enumerateObjectsUsingBlock:^(NSDictionary *partInfo, NSUInteger idx, BOOL * _Nonnull stop) {
  1303. unsigned long long remotePartNumber = 0;
  1304. NSString *partNumberString = [partInfo objectForKey: OSSPartNumberXMLTOKEN];
  1305. NSScanner *scanner = [NSScanner scannerWithString: partNumberString];
  1306. [scanner scanUnsignedLongLong: &remotePartNumber];
  1307. NSString *remotePartEtag = [partInfo objectForKey:OSSETagXMLTOKEN];
  1308. unsigned long long remotePartSize = 0;
  1309. NSString *partSizeString = [partInfo objectForKey:OSSSizeXMLTOKEN];
  1310. scanner = [NSScanner scannerWithString:partSizeString];
  1311. [scanner scanUnsignedLongLong:&remotePartSize];
  1312. #pragma clang diagnostic push
  1313. #pragma clang diagnostic ignored "-Wshorten-64-to-32"
  1314. OSSPartInfo * info = [[OSSPartInfo alloc] init];
  1315. info.partNum = remotePartNumber;
  1316. info.size = remotePartSize;
  1317. info.eTag = remotePartEtag;
  1318. #pragma clang diagnostic pop
  1319. NSDictionary *tPartInfo = [localPartInfos objectForKey: [@(remotePartNumber) stringValue]];
  1320. info.crc64 = [tPartInfo[@"crc64"] unsignedLongLongValue];
  1321. [uploadedPartInfos addObject:info];
  1322. [alreadyUploadIndex addObject:@(remotePartNumber)];
  1323. }];
  1324. if ([alreadyUploadIndex count] > 0 && request.uploadProgress && uploadFileSize) {
  1325. request.uploadProgress(0, uploadedLength, uploadFileSize);
  1326. }
  1327. }
  1328. if (![uploadId oss_isNotEmpty]) {
  1329. OSSInitMultipartUploadRequest *initRequest = [OSSInitMultipartUploadRequest new];
  1330. initRequest.bucketName = request.bucketName;
  1331. initRequest.objectKey = request.objectKey;
  1332. initRequest.contentType = request.contentType;
  1333. initRequest.objectMeta = request.completeMetaHeader;
  1334. initRequest.sequential = sequential;
  1335. initRequest.crcFlag = request.crcFlag;
  1336. OSSTask *task = [self processResumableInitMultipartUpload:initRequest
  1337. recordFilePath:recordFilePath];
  1338. if (task.error)
  1339. {
  1340. return task;
  1341. }
  1342. OSSInitMultipartUploadResult *initResult = (OSSInitMultipartUploadResult *)task.result;
  1343. uploadId = initResult.uploadId;
  1344. }
  1345. request.uploadId = uploadId;
  1346. localPartInfosPath = [[[NSString oss_documentDirectory] stringByAppendingPathComponent:kClientRecordNameWithCommonPrefix] stringByAppendingPathComponent:uploadId];
  1347. if (request.isCancelled)
  1348. {
  1349. if(resumable)
  1350. {
  1351. OSSResumableUploadRequest *resumableRequest = (OSSResumableUploadRequest *)request;
  1352. if (resumableRequest.deleteUploadIdOnCancelling) {
  1353. OSSTask *abortTask = [self abortMultipartUpload:request sequential:sequential resumable:resumable];
  1354. [abortTask waitUntilFinished];
  1355. }
  1356. }
  1357. return [OSSTask taskWithError:[OSSClient cancelError]];
  1358. }
  1359. if (sequential) {
  1360. errorTask = [self sequentialUpload:request
  1361. uploadIndex:alreadyUploadIndex
  1362. uploadPart:uploadedPartInfos
  1363. count:partCount
  1364. uploadedLength:&uploadedLength
  1365. fileSize:uploadFileSize];
  1366. } else {
  1367. errorTask = [self upload:request
  1368. uploadIndex:alreadyUploadIndex
  1369. uploadPart:uploadedPartInfos
  1370. count:partCount
  1371. uploadedLength:&uploadedLength
  1372. fileSize:uploadFileSize];
  1373. }
  1374. if(errorTask.error)
  1375. {
  1376. OSSTask *abortTask;
  1377. if(resumable)
  1378. {
  1379. OSSResumableUploadRequest *resumableRequest = (OSSResumableUploadRequest *)request;
  1380. if (resumableRequest.deleteUploadIdOnCancelling || errorTask.error.code == OSSClientErrorCodeFileCantWrite) {
  1381. abortTask = [self abortMultipartUpload:request sequential:sequential resumable:resumable];
  1382. }
  1383. }else
  1384. {
  1385. abortTask =[self abortMultipartUpload:request sequential:sequential resumable:resumable];
  1386. }
  1387. [abortTask waitUntilFinished];
  1388. return errorTask;
  1389. }
  1390. [uploadedPartInfos sortUsingComparator:^NSComparisonResult(OSSPartInfo *part1,OSSPartInfo* part2) {
  1391. if(part1.partNum < part2.partNum){
  1392. return NSOrderedAscending;
  1393. }else if(part1.partNum > part2.partNum){
  1394. return NSOrderedDescending;
  1395. }else{
  1396. return NSOrderedSame;
  1397. }
  1398. }];
  1399. // 如果开启了crc64的校验
  1400. uint64_t local_crc64 = 0;
  1401. if (request.crcFlag == OSSRequestCRCOpen)
  1402. {
  1403. for (NSUInteger index = 0; index< uploadedPartInfos.count; index++)
  1404. {
  1405. uint64_t partCrc64 = uploadedPartInfos[index].crc64;
  1406. int64_t partSize = uploadedPartInfos[index].size;
  1407. #pragma clang diagnostic push
  1408. #pragma clang diagnostic ignored "-Wshorten-64-to-32"
  1409. local_crc64 = [OSSUtil crc64ForCombineCRC1:local_crc64 CRC2:partCrc64 length:partSize];
  1410. #pragma clang diagnostic pop
  1411. }
  1412. }
  1413. return [self processCompleteMultipartUpload:request
  1414. partInfos:uploadedPartInfos
  1415. clientCrc64:local_crc64
  1416. recordFilePath:recordFilePath
  1417. localPartInfosPath:localPartInfosPath];
  1418. }];
  1419. }
  1420. - (OSSTask *)sequentialUpload:(OSSMultipartUploadRequest *)request
  1421. uploadIndex:(NSMutableArray *)alreadyUploadIndex
  1422. uploadPart:(NSMutableArray *)alreadyUploadPart
  1423. count:(NSUInteger)partCout
  1424. uploadedLength:(NSUInteger *)uploadedLength
  1425. fileSize:(unsigned long long)uploadFileSize
  1426. {
  1427. OSSRequestCRCFlag crcFlag = request.crcFlag;
  1428. __block OSSTask *errorTask;
  1429. __block NSMutableDictionary *localPartInfos = nil;
  1430. if (crcFlag == OSSRequestCRCOpen) {
  1431. localPartInfos = [self localPartInfosDictoryWithUploadId:request.uploadId];
  1432. }
  1433. if (!localPartInfos) {
  1434. localPartInfos = [NSMutableDictionary dictionary];
  1435. }
  1436. NSError *readError;
  1437. NSFileHandle *fileHande = [NSFileHandle fileHandleForReadingFromURL:request.uploadingFileURL error:&readError];
  1438. if (readError) {
  1439. return [OSSTask taskWithError: readError];
  1440. }
  1441. NSUInteger realPartLength = request.partSize;
  1442. for (int i = 1; i <= partCout; i++) {
  1443. if (errorTask) {
  1444. break;
  1445. }
  1446. if (request.isCancelled) {
  1447. errorTask = [OSSTask taskWithError:[OSSClient cancelError]];
  1448. break;
  1449. }
  1450. if ([alreadyUploadIndex containsObject:@(i)]) {
  1451. continue;
  1452. }
  1453. realPartLength = request.partSize;
  1454. [fileHande seekToFileOffset:request.partSize * (i - 1)];
  1455. if (i == partCout) {
  1456. #pragma clang diagnostic push
  1457. #pragma clang diagnostic ignored "-Wshorten-64-to-32"
  1458. realPartLength = uploadFileSize - request.partSize * (i - 1);
  1459. #pragma clang diagnostic pop
  1460. }
  1461. NSData *uploadPartData = [fileHande readDataOfLength:realPartLength];
  1462. @autoreleasepool {
  1463. OSSUploadPartRequest * uploadPart = [OSSUploadPartRequest new];
  1464. uploadPart.bucketName = request.bucketName;
  1465. uploadPart.objectkey = request.objectKey;
  1466. uploadPart.partNumber = i;
  1467. uploadPart.uploadId = request.uploadId;
  1468. uploadPart.uploadPartData = uploadPartData;
  1469. uploadPart.contentMd5 = [OSSUtil base64Md5ForData:uploadPartData];
  1470. uploadPart.crcFlag = request.crcFlag;
  1471. OSSTask * uploadPartTask = [self uploadPart:uploadPart];
  1472. [uploadPartTask waitUntilFinished];
  1473. if (uploadPartTask.error) {
  1474. if (labs(uploadPartTask.error.code) != 409) {
  1475. errorTask = uploadPartTask;
  1476. break;
  1477. } else {
  1478. NSDictionary *partDict = uploadPartTask.error.userInfo;
  1479. OSSPartInfo *partInfo = [[OSSPartInfo alloc] init];
  1480. partInfo.eTag = partDict[@"PartEtag"];
  1481. #pragma clang diagnostic push
  1482. #pragma clang diagnostic ignored "-Wshorten-64-to-32"
  1483. partInfo.partNum = [(NSString *)partDict[@"PartNumber"] integerValue];
  1484. partInfo.size = realPartLength;
  1485. #pragma clang diagnostic push
  1486. partInfo.crc64 = [[uploadPartData mutableCopy] oss_crc64];
  1487. [alreadyUploadPart addObject:partInfo];
  1488. }
  1489. } else {
  1490. OSSUploadPartResult * result = uploadPartTask.result;
  1491. OSSPartInfo * partInfo = [OSSPartInfo new];
  1492. partInfo.partNum = i;
  1493. partInfo.eTag = result.eTag;
  1494. partInfo.size = realPartLength;
  1495. uint64_t crc64OfPart;
  1496. @try {
  1497. NSScanner *scanner = [NSScanner scannerWithString:result.remoteCRC64ecma];
  1498. [scanner scanUnsignedLongLong:&crc64OfPart];
  1499. partInfo.crc64 = crc64OfPart;
  1500. } @catch (NSException *exception) {
  1501. OSSLogError(@"multipart upload error with nil remote crc64!");
  1502. }
  1503. [alreadyUploadPart addObject:partInfo];
  1504. if (crcFlag == OSSRequestCRCOpen)
  1505. {
  1506. [self processForLocalPartInfos:localPartInfos
  1507. partInfo:partInfo
  1508. uploadId:request.uploadId];
  1509. [self persistencePartInfos:localPartInfos
  1510. withUploadId:request.uploadId];
  1511. }
  1512. @synchronized(lock) {
  1513. *uploadedLength += realPartLength;
  1514. if (request.uploadProgress)
  1515. {
  1516. request.uploadProgress(realPartLength, *uploadedLength, uploadFileSize);
  1517. }
  1518. }
  1519. }
  1520. }
  1521. }
  1522. [fileHande closeFile];
  1523. return errorTask;
  1524. }
  1525. @end
  1526. @implementation OSSClient (PresignURL)
  1527. - (OSSTask *)presignConstrainURLWithBucketName:(NSString *)bucketName
  1528. withObjectKey:(NSString *)objectKey
  1529. withExpirationInterval:(NSTimeInterval)interval {
  1530. return [self presignConstrainURLWithBucketName:bucketName
  1531. withObjectKey:objectKey
  1532. withExpirationInterval:interval
  1533. withParameters:@{}];
  1534. }
  1535. - (OSSTask *)presignConstrainURLWithBucketName:(NSString *)bucketName
  1536. withObjectKey:(NSString *)objectKey
  1537. withExpirationInterval:(NSTimeInterval)interval
  1538. withParameters:(NSDictionary *)parameters {
  1539. return [self presignConstrainURLWithBucketName: bucketName
  1540. withObjectKey: objectKey
  1541. httpMethod: @"GET"
  1542. withExpirationInterval: interval
  1543. withParameters: parameters];
  1544. }
  1545. - (OSSTask *)presignConstrainURLWithBucketName:(NSString *)bucketName
  1546. withObjectKey:(NSString *)objectKey
  1547. httpMethod:(NSString *)method
  1548. withExpirationInterval:(NSTimeInterval)interval
  1549. withParameters:(NSDictionary *)parameters
  1550. {
  1551. return [[OSSTask taskWithResult:nil] continueWithBlock:^id(OSSTask *task) {
  1552. NSString * resource = [NSString stringWithFormat:@"/%@/%@", bucketName, objectKey];
  1553. NSString * expires = [@((int64_t)[[NSDate oss_clockSkewFixedDate] timeIntervalSince1970] + interval) stringValue];
  1554. NSMutableDictionary * params = [NSMutableDictionary dictionary];
  1555. if (parameters.count > 0) {
  1556. [params addEntriesFromDictionary:parameters];
  1557. }
  1558. NSString * wholeSign = nil;
  1559. OSSFederationToken *token = nil;
  1560. NSError *error = nil;
  1561. if ([self.credentialProvider isKindOfClass:[OSSFederationCredentialProvider class]]) {
  1562. token = [(OSSFederationCredentialProvider *)self.credentialProvider getToken:&error];
  1563. if (error) {
  1564. return [OSSTask taskWithError:error];
  1565. }
  1566. } else if ([self.credentialProvider isKindOfClass:[OSSStsTokenCredentialProvider class]]) {
  1567. token = [(OSSStsTokenCredentialProvider *)self.credentialProvider getToken];
  1568. }
  1569. if ([self.credentialProvider isKindOfClass:[OSSFederationCredentialProvider class]]
  1570. || [self.credentialProvider isKindOfClass:[OSSStsTokenCredentialProvider class]])
  1571. {
  1572. [params oss_setObject:token.tToken forKey:@"security-token"];
  1573. resource = [NSString stringWithFormat:@"%@?%@", resource, [OSSUtil populateSubresourceStringFromParameter:params]];
  1574. NSString * string2sign = [NSString stringWithFormat:@"%@\n\n\n%@\n%@", method, expires, resource];
  1575. wholeSign = [OSSUtil sign:string2sign withToken:token];
  1576. } else {
  1577. NSString * subresource = [OSSUtil populateSubresourceStringFromParameter:params];
  1578. if ([subresource length] > 0) {
  1579. resource = [NSString stringWithFormat:@"%@?%@", resource, [OSSUtil populateSubresourceStringFromParameter:params]];
  1580. }
  1581. NSString * string2sign = [NSString stringWithFormat:@"%@\n\n\n%@\n%@", method, expires, resource];
  1582. wholeSign = [self.credentialProvider sign:string2sign error:&error];
  1583. if (error) {
  1584. return [OSSTask taskWithError:error];
  1585. }
  1586. }
  1587. NSArray * splitResult = [wholeSign componentsSeparatedByString:@":"];
  1588. if ([splitResult count] != 2
  1589. || ![((NSString *)[splitResult objectAtIndex:0]) hasPrefix:@"OSS "]) {
  1590. return [OSSTask taskWithError:[NSError errorWithDomain:OSSClientErrorDomain
  1591. code:OSSClientErrorCodeSignFailed
  1592. userInfo:@{OSSErrorMessageTOKEN: @"the returned signature is invalid"}]];
  1593. }
  1594. NSString * accessKey = [(NSString *)[splitResult objectAtIndex:0] substringFromIndex:4];
  1595. NSString * signature = [splitResult objectAtIndex:1];
  1596. NSURL * endpointURL = [NSURL URLWithString:self.endpoint];
  1597. NSString * host = endpointURL.host;
  1598. if ([OSSUtil isOssOriginBucketHost:host]) {
  1599. host = [NSString stringWithFormat:@"%@.%@", bucketName, host];
  1600. }
  1601. [params oss_setObject:signature forKey:@"Signature"];
  1602. [params oss_setObject:accessKey forKey:@"OSSAccessKeyId"];
  1603. [params oss_setObject:expires forKey:@"Expires"];
  1604. NSString * stringURL = [NSString stringWithFormat:@"%@://%@/%@?%@",
  1605. endpointURL.scheme,
  1606. host,
  1607. [OSSUtil encodeURL:objectKey],
  1608. [OSSUtil populateQueryStringFromParameter:params]];
  1609. return [OSSTask taskWithResult:stringURL];
  1610. }];
  1611. }
  1612. - (OSSTask *)presignPublicURLWithBucketName:(NSString *)bucketName
  1613. withObjectKey:(NSString *)objectKey {
  1614. return [self presignPublicURLWithBucketName:bucketName
  1615. withObjectKey:objectKey
  1616. withParameters:@{}];
  1617. }
  1618. - (OSSTask *)presignPublicURLWithBucketName:(NSString *)bucketName
  1619. withObjectKey:(NSString *)objectKey
  1620. withParameters:(NSDictionary *)parameters {
  1621. return [[OSSTask taskWithResult:nil] continueWithBlock:^id(OSSTask *task) {
  1622. NSURL * endpointURL = [NSURL URLWithString:self.endpoint];
  1623. NSString * host = endpointURL.host;
  1624. if ([OSSUtil isOssOriginBucketHost:host]) {
  1625. host = [NSString stringWithFormat:@"%@.%@", bucketName, host];
  1626. }
  1627. if ([parameters count] > 0) {
  1628. NSString * stringURL = [NSString stringWithFormat:@"%@://%@/%@?%@",
  1629. endpointURL.scheme,
  1630. host,
  1631. [OSSUtil encodeURL:objectKey],
  1632. [OSSUtil populateQueryStringFromParameter:parameters]];
  1633. return [OSSTask taskWithResult:stringURL];
  1634. } else {
  1635. NSString * stringURL = [NSString stringWithFormat:@"%@://%@/%@",
  1636. endpointURL.scheme,
  1637. host,
  1638. [OSSUtil encodeURL:objectKey]];
  1639. return [OSSTask taskWithResult:stringURL];
  1640. }
  1641. }];
  1642. }
  1643. @end
  1644. @implementation OSSClient (Utilities)
  1645. - (BOOL)doesObjectExistInBucket:(NSString *)bucketName
  1646. objectKey:(NSString *)objectKey
  1647. error:(const NSError **)error {
  1648. OSSHeadObjectRequest * headRequest = [OSSHeadObjectRequest new];
  1649. headRequest.bucketName = bucketName;
  1650. headRequest.objectKey = objectKey;
  1651. OSSTask * headTask = [self headObject:headRequest];
  1652. [headTask waitUntilFinished];
  1653. NSError *headError = headTask.error;
  1654. if (!headError) {
  1655. return YES;
  1656. } else {
  1657. if ([headError.domain isEqualToString: OSSServerErrorDomain] && labs(headError.code) == 404) {
  1658. return NO;
  1659. } else {
  1660. if (error != nil) {
  1661. *error = headError;
  1662. }
  1663. return NO;
  1664. }
  1665. }
  1666. }
  1667. @end
  1668. @implementation OSSClient (ImageService)
  1669. - (OSSTask *)imageActionPersist:(OSSImagePersistRequest *)request
  1670. {
  1671. if (![request.fromBucket oss_isNotEmpty]
  1672. || ![request.fromObject oss_isNotEmpty]
  1673. || ![request.toBucket oss_isNotEmpty]
  1674. || ![request.toObject oss_isNotEmpty]
  1675. || ![request.action oss_isNotEmpty]) {
  1676. NSError *error = [NSError errorWithDomain:OSSTaskErrorDomain
  1677. code:OSSClientErrorCodeInvalidArgument
  1678. userInfo:@{OSSErrorMessageTOKEN: @"imagePersist parameters not be empty!"}];
  1679. return [OSSTask taskWithError:error];
  1680. }
  1681. OSSNetworkingRequestDelegate *requestDelegate = request.requestDelegate;
  1682. NSMutableDictionary *params = [NSMutableDictionary dictionary];
  1683. [params oss_setObject:@"" forKey:OSSHttpQueryProcess];
  1684. requestDelegate.uploadingData = [OSSUtil constructHttpBodyForImagePersist:request.action toBucket:request.toBucket toObjectKey:request.toObject];
  1685. OSSHttpResponseParser *responseParser = [[OSSHttpResponseParser alloc] initForOperationType:OSSOperationTypeImagePersist];
  1686. requestDelegate.responseParser = responseParser;
  1687. OSSAllRequestNeededMessage *neededMsg = [[OSSAllRequestNeededMessage alloc] init];
  1688. neededMsg.endpoint = self.endpoint;
  1689. neededMsg.httpMethod = OSSHTTPMethodPOST;
  1690. neededMsg.bucketName = request.fromBucket;
  1691. neededMsg.objectKey = request.fromObject;
  1692. neededMsg.params = params;
  1693. requestDelegate.allNeededMessage = neededMsg;
  1694. requestDelegate.operType = OSSOperationTypeImagePersist;
  1695. return [self invokeRequest:requestDelegate requireAuthentication:request.isAuthenticationRequired];
  1696. }
  1697. @end
  1698. @implementation OSSClient (Callback)
  1699. - (OSSTask *)triggerCallBack:(OSSCallBackRequest *)request
  1700. {
  1701. OSSNetworkingRequestDelegate *requestDelegate = request.requestDelegate;
  1702. NSMutableDictionary *params = [NSMutableDictionary dictionary];
  1703. [params oss_setObject:@"" forKey:OSSHttpQueryProcess];
  1704. NSString *paramString = [request.callbackParam base64JsonString];
  1705. NSString *variblesString = [request.callbackVar base64JsonString];
  1706. requestDelegate.uploadingData = [OSSUtil constructHttpBodyForTriggerCallback:paramString callbackVaribles:variblesString];
  1707. NSString *md5String = [OSSUtil base64Md5ForData:requestDelegate.uploadingData];
  1708. OSSHttpResponseParser *responseParser = [[OSSHttpResponseParser alloc] initForOperationType:OSSOperationTypeTriggerCallBack];
  1709. requestDelegate.responseParser = responseParser;
  1710. OSSAllRequestNeededMessage *neededMsg = [[OSSAllRequestNeededMessage alloc] init];
  1711. neededMsg.endpoint = self.endpoint;
  1712. neededMsg.httpMethod = OSSHTTPMethodPOST;
  1713. neededMsg.bucketName = request.bucketName;
  1714. neededMsg.objectKey = request.objectName;
  1715. neededMsg.contentMd5 = md5String;
  1716. neededMsg.params = params;
  1717. requestDelegate.allNeededMessage = neededMsg;
  1718. requestDelegate.operType = OSSOperationTypeTriggerCallBack;
  1719. return [self invokeRequest:requestDelegate requireAuthentication:request.isAuthenticationRequired];
  1720. }
  1721. @end