113 lines
3.8 KiB
Objective-C
113 lines
3.8 KiB
Objective-C
/**
|
|
* Copyright (c) 2015-present, Facebook, Inc.
|
|
* All rights reserved.
|
|
*
|
|
* This source code is licensed under the BSD-style license found in the
|
|
* LICENSE file in the root directory of this source tree. An additional grant
|
|
* of patent rights can be found in the PATENTS file in the same directory.
|
|
*/
|
|
|
|
#import "RCTMultipartStreamReader.h"
|
|
|
|
#define CRLF @"\r\n"
|
|
|
|
@implementation RCTMultipartStreamReader {
|
|
__strong NSInputStream *_stream;
|
|
__strong NSString *_boundary;
|
|
}
|
|
|
|
- (instancetype)initWithInputStream:(NSInputStream *)stream boundary:(NSString *)boundary
|
|
{
|
|
if (self = [super init]) {
|
|
_stream = stream;
|
|
_boundary = boundary;
|
|
}
|
|
return self;
|
|
}
|
|
|
|
- (NSDictionary *)parseHeaders:(NSData *)data
|
|
{
|
|
NSMutableDictionary *headers = [NSMutableDictionary new];
|
|
NSString *text = [[NSString alloc] initWithData:data encoding:NSUTF8StringEncoding];
|
|
NSArray<NSString *> *lines = [text componentsSeparatedByString:CRLF];
|
|
for (NSString *line in lines) {
|
|
NSUInteger location = [line rangeOfString:@":"].location;
|
|
if (location == NSNotFound) {
|
|
continue;
|
|
}
|
|
NSString *key = [line substringToIndex:location];
|
|
NSString *value = [[line substringFromIndex:location + 1] stringByTrimmingCharactersInSet:[NSCharacterSet whitespaceAndNewlineCharacterSet]];
|
|
[headers setValue:value forKey:key];
|
|
}
|
|
return headers;
|
|
}
|
|
|
|
- (void)emitChunk:(NSData *)data callback:(RCTMultipartCallback)callback done:(BOOL)done
|
|
{
|
|
NSData *marker = [CRLF CRLF dataUsingEncoding:NSUTF8StringEncoding];
|
|
NSRange range = [data rangeOfData:marker options:0 range:NSMakeRange(0, data.length)];
|
|
if (range.location == NSNotFound) {
|
|
callback(nil, data, done);
|
|
} else {
|
|
NSData *headersData = [data subdataWithRange:NSMakeRange(0, range.location)];
|
|
NSInteger bodyStart = range.location + marker.length;
|
|
NSData *bodyData = [data subdataWithRange:NSMakeRange(bodyStart, data.length - bodyStart)];
|
|
callback([self parseHeaders:headersData], bodyData, done);
|
|
}
|
|
}
|
|
|
|
- (BOOL)readAllParts:(RCTMultipartCallback)callback
|
|
{
|
|
NSInteger chunkStart = 0;
|
|
NSInteger bytesSeen = 0;
|
|
|
|
NSData *delimiter = [[NSString stringWithFormat:@"%@--%@%@", CRLF, _boundary, CRLF] dataUsingEncoding:NSUTF8StringEncoding];
|
|
NSData *closeDelimiter = [[NSString stringWithFormat:@"%@--%@--%@", CRLF, _boundary, CRLF] dataUsingEncoding:NSUTF8StringEncoding];
|
|
NSMutableData *content = [[NSMutableData alloc] initWithCapacity:1];
|
|
|
|
const NSUInteger bufferLen = 4 * 1024;
|
|
uint8_t buffer[bufferLen];
|
|
|
|
[_stream open];
|
|
while (true) {
|
|
BOOL isCloseDelimiter = NO;
|
|
// Search only a subset of chunk that we haven't seen before + few bytes
|
|
// to allow for the edge case when the delimiter is cut by read call
|
|
NSInteger searchStart = MAX(bytesSeen - (NSInteger)closeDelimiter.length, chunkStart);
|
|
NSRange remainingBufferRange = NSMakeRange(searchStart, content.length - searchStart);
|
|
NSRange range = [content rangeOfData:delimiter options:0 range:remainingBufferRange];
|
|
if (range.location == NSNotFound) {
|
|
isCloseDelimiter = YES;
|
|
range = [content rangeOfData:closeDelimiter options:0 range:remainingBufferRange];
|
|
}
|
|
|
|
if (range.location == NSNotFound) {
|
|
bytesSeen = content.length;
|
|
NSInteger bytesRead = [_stream read:buffer maxLength:bufferLen];
|
|
if (bytesRead <= 0 || _stream.streamError) {
|
|
return NO;
|
|
}
|
|
[content appendBytes:buffer length:bytesRead];
|
|
continue;
|
|
}
|
|
|
|
NSInteger chunkEnd = range.location;
|
|
NSInteger length = chunkEnd - chunkStart;
|
|
bytesSeen = chunkEnd;
|
|
|
|
// Ignore preamble
|
|
if (chunkStart > 0) {
|
|
NSData *chunk = [content subdataWithRange:NSMakeRange(chunkStart, length)];
|
|
[self emitChunk:chunk callback:callback done:isCloseDelimiter];
|
|
}
|
|
|
|
if (isCloseDelimiter) {
|
|
return YES;
|
|
}
|
|
|
|
chunkStart = chunkEnd + delimiter.length;
|
|
}
|
|
}
|
|
|
|
@end
|