Added method to allow adding a file's contents directly. Fixes issue #105.

git-svn-id: http://google-breakpad.googlecode.com/svn/trunk@100 4c0a9323-5329-0410-9bdc-e9ce6186880e
This commit is contained in:
waylonis 2007-01-04 04:43:41 +00:00
parent d4e527b7ee
commit 0ec76c7fad
2 changed files with 31 additions and 12 deletions

View File

@ -51,6 +51,7 @@
- (NSDictionary *)parameters; - (NSDictionary *)parameters;
- (void)addFileAtPath:(NSString *)path name:(NSString *)name; - (void)addFileAtPath:(NSString *)path name:(NSString *)name;
- (void)addFileContents:(NSData *)data name:(NSString *)name;
- (NSDictionary *)files; - (NSDictionary *)files;
// Set the data and return the response // Set the data and return the response

View File

@ -32,6 +32,7 @@
@interface HTTPMultipartUpload(PrivateMethods) @interface HTTPMultipartUpload(PrivateMethods)
- (NSString *)multipartBoundary; - (NSString *)multipartBoundary;
- (NSData *)formDataForKey:(NSString *)key value:(NSString *)value; - (NSData *)formDataForKey:(NSString *)key value:(NSString *)value;
- (NSData *)formDataForFileContents:(NSData *)contents name:(NSString *)name;
- (NSData *)formDataForFile:(NSString *)file name:(NSString *)name; - (NSData *)formDataForFile:(NSString *)file name:(NSString *)name;
@end @end
@ -48,7 +49,7 @@
//============================================================================= //=============================================================================
- (NSData *)formDataForKey:(NSString *)key value:(NSString *)value { - (NSData *)formDataForKey:(NSString *)key value:(NSString *)value {
NSString *escaped = NSString *escaped =
[key stringByAddingPercentEscapesUsingEncoding:NSUTF8StringEncoding]; [key stringByAddingPercentEscapesUsingEncoding:NSUTF8StringEncoding];
NSString *fmt = NSString *fmt =
@"--%@\r\nContent-Disposition: form-data; name=\"%@\"\r\n\r\n%@\r\n"; @"--%@\r\nContent-Disposition: form-data; name=\"%@\"\r\n\r\n%@\r\n";
@ -58,23 +59,29 @@
} }
//============================================================================= //=============================================================================
- (NSData *)formDataForFile:(NSString *)file name:(NSString *)name { - (NSData *)formDataForFileContents:(NSData *)contents name:(NSString *)name {
NSMutableData *data = [NSMutableData data]; NSMutableData *data = [NSMutableData data];
NSString *escaped = NSString *escaped =
[name stringByAddingPercentEscapesUsingEncoding:NSUTF8StringEncoding]; [name stringByAddingPercentEscapesUsingEncoding:NSUTF8StringEncoding];
NSString *fmt = @"--%@\r\nContent-Disposition: form-data; name=\"%@\"; " NSString *fmt = @"--%@\r\nContent-Disposition: form-data; name=\"%@\"; "
"filename=\"%@\"\r\nContent-Type: application/octet-stream\r\n\r\n"; "filename=\"minidump.dmp\"\r\nContent-Type: application/octet-stream\r\n\r\n";
NSString *pre = [NSString stringWithFormat:fmt, boundary_, escaped, NSString *pre = [NSString stringWithFormat:fmt, boundary_, escaped];
[file lastPathComponent]];
NSString *post = [NSString stringWithFormat:@"\r\n--%@--\r\n", boundary_]; NSString *post = [NSString stringWithFormat:@"\r\n--%@--\r\n", boundary_];
[data appendData:[pre dataUsingEncoding:NSUTF8StringEncoding]]; [data appendData:[pre dataUsingEncoding:NSUTF8StringEncoding]];
[data appendData:[NSData dataWithContentsOfFile:file]]; [data appendData:contents];
[data appendData:[post dataUsingEncoding:NSUTF8StringEncoding]]; [data appendData:[post dataUsingEncoding:NSUTF8StringEncoding]];
return data; return data;
} }
//=============================================================================
- (NSData *)formDataForFile:(NSString *)file name:(NSString *)name {
NSData *contents = [NSData dataWithContentsOfFile:file];
return [self formDataForFileContents:contents name:name];
}
//============================================================================= //=============================================================================
#pragma mark - #pragma mark -
#pragma mark || Public || #pragma mark || Public ||
@ -83,6 +90,7 @@
if ((self = [super init])) { if ((self = [super init])) {
url_ = [url copy]; url_ = [url copy];
boundary_ = [[self multipartBoundary] retain]; boundary_ = [[self multipartBoundary] retain];
files_ = [[NSMutableDictionary alloc] init];
} }
return self; return self;
@ -119,12 +127,14 @@
//============================================================================= //=============================================================================
- (void)addFileAtPath:(NSString *)path name:(NSString *)name { - (void)addFileAtPath:(NSString *)path name:(NSString *)name {
if (!files_)
files_ = [[NSMutableDictionary alloc] init];
[files_ setObject:path forKey:name]; [files_ setObject:path forKey:name];
} }
//=============================================================================
- (void)addFileContents:(NSData *)data name:(NSString *)name {
[files_ setObject:data forKey:name];
}
//============================================================================= //=============================================================================
- (NSDictionary *)files { - (NSDictionary *)files {
return files_; return files_;
@ -155,9 +165,17 @@
count = [fileNames count]; count = [fileNames count];
for (i = 0; i < count; ++i) { for (i = 0; i < count; ++i) {
NSString *name = [fileNames objectAtIndex:i]; NSString *name = [fileNames objectAtIndex:i];
NSString *file = [files_ objectForKey:name]; id fileOrData = [files_ objectForKey:name];
NSData *fileData;
[postBody appendData:[self formDataForFile:file name:name]]; // The object can be either the path to a file (NSString) or the contents
// of the file (NSData).
if ([fileOrData isKindOfClass:[NSData class]])
fileData = [self formDataForFileContents:fileOrData name:name];
else
fileData = [self formDataForFile:fileOrData name:name];
[postBody appendData:fileData];
} }
[req setHTTPBody:postBody]; [req setHTTPBody:postBody];