mirror of
https://github.com/swisspol/GCDWebServer.git
synced 2024-10-07 16:48:03 +08:00
382 lines
11 KiB
Objective-C
382 lines
11 KiB
Objective-C
/*
|
|
Copyright (c) 2012-2014, Pierre-Olivier Latour
|
|
All rights reserved.
|
|
|
|
Redistribution and use in source and binary forms, with or without
|
|
modification, are permitted provided that the following conditions are met:
|
|
* Redistributions of source code must retain the above copyright
|
|
notice, this list of conditions and the following disclaimer.
|
|
* Redistributions in binary form must reproduce the above copyright
|
|
notice, this list of conditions and the following disclaimer in the
|
|
documentation and/or other materials provided with the distribution.
|
|
* The name of Pierre-Olivier Latour may not be used to endorse
|
|
or promote products derived from this software without specific
|
|
prior written permission.
|
|
|
|
THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND
|
|
ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
|
|
WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
|
|
DISCLAIMED. IN NO EVENT SHALL PIERRE-OLIVIER LATOUR BE LIABLE FOR ANY
|
|
DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
|
|
(INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
|
|
LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND
|
|
ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
|
|
(INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
|
|
SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
|
|
*/
|
|
|
|
#import <sys/stat.h>
|
|
|
|
#import "GCDWebServerPrivate.h"
|
|
|
|
@interface GCDWebServerResponse () {
|
|
@private
|
|
NSString* _type;
|
|
NSUInteger _length;
|
|
NSInteger _status;
|
|
NSUInteger _maxAge;
|
|
NSMutableDictionary* _headers;
|
|
}
|
|
@end
|
|
|
|
@interface GCDWebServerDataResponse () {
|
|
@private
|
|
NSData* _data;
|
|
NSInteger _offset;
|
|
}
|
|
@end
|
|
|
|
@interface GCDWebServerFileResponse () {
|
|
@private
|
|
NSString* _path;
|
|
NSUInteger _offset;
|
|
NSUInteger _size;
|
|
int _file;
|
|
}
|
|
@end
|
|
|
|
@implementation GCDWebServerResponse
|
|
|
|
@synthesize contentType=_type, contentLength=_length, statusCode=_status, cacheControlMaxAge=_maxAge, additionalHeaders=_headers;
|
|
|
|
+ (GCDWebServerResponse*)response {
|
|
return ARC_AUTORELEASE([[[self class] alloc] init]);
|
|
}
|
|
|
|
- (id)init {
|
|
return [self initWithContentType:nil contentLength:0];
|
|
}
|
|
|
|
- (id)initWithContentType:(NSString*)type contentLength:(NSUInteger)length {
|
|
if ((self = [super init])) {
|
|
_type = [type copy];
|
|
_length = length;
|
|
_status = 200;
|
|
_maxAge = 0;
|
|
_headers = [[NSMutableDictionary alloc] init];
|
|
|
|
if ((_length > 0) && (_type == nil)) {
|
|
_type = [kGCDWebServerDefaultMimeType copy];
|
|
}
|
|
}
|
|
return self;
|
|
}
|
|
|
|
- (void)dealloc {
|
|
ARC_RELEASE(_type);
|
|
ARC_RELEASE(_headers);
|
|
|
|
ARC_DEALLOC(super);
|
|
}
|
|
|
|
- (void)setValue:(NSString*)value forAdditionalHeader:(NSString*)header {
|
|
[_headers setValue:value forKey:header];
|
|
}
|
|
|
|
- (BOOL)hasBody {
|
|
return _type ? YES : NO;
|
|
}
|
|
|
|
@end
|
|
|
|
@implementation GCDWebServerResponse (Subclassing)
|
|
|
|
- (BOOL)open {
|
|
[self doesNotRecognizeSelector:_cmd];
|
|
return NO;
|
|
}
|
|
|
|
- (NSInteger)read:(void*)buffer maxLength:(NSUInteger)length {
|
|
[self doesNotRecognizeSelector:_cmd];
|
|
return -1;
|
|
}
|
|
|
|
- (BOOL)close {
|
|
[self doesNotRecognizeSelector:_cmd];
|
|
return NO;
|
|
}
|
|
|
|
@end
|
|
|
|
@implementation GCDWebServerResponse (Extensions)
|
|
|
|
+ (GCDWebServerResponse*)responseWithStatusCode:(NSInteger)statusCode {
|
|
return ARC_AUTORELEASE([[self alloc] initWithStatusCode:statusCode]);
|
|
}
|
|
|
|
+ (GCDWebServerResponse*)responseWithRedirect:(NSURL*)location permanent:(BOOL)permanent {
|
|
return ARC_AUTORELEASE([[self alloc] initWithRedirect:location permanent:permanent]);
|
|
}
|
|
|
|
- (id)initWithStatusCode:(NSInteger)statusCode {
|
|
if ((self = [self initWithContentType:nil contentLength:0])) {
|
|
self.statusCode = statusCode;
|
|
}
|
|
return self;
|
|
}
|
|
|
|
- (id)initWithRedirect:(NSURL*)location permanent:(BOOL)permanent {
|
|
if ((self = [self initWithContentType:nil contentLength:0])) {
|
|
self.statusCode = permanent ? 301 : 307;
|
|
[self setValue:[location absoluteString] forAdditionalHeader:@"Location"];
|
|
}
|
|
return self;
|
|
}
|
|
|
|
@end
|
|
|
|
@implementation GCDWebServerDataResponse
|
|
|
|
+ (GCDWebServerDataResponse*)responseWithData:(NSData*)data contentType:(NSString*)type {
|
|
return ARC_AUTORELEASE([[[self class] alloc] initWithData:data contentType:type]);
|
|
}
|
|
|
|
- (id)initWithData:(NSData*)data contentType:(NSString*)type {
|
|
if (data == nil) {
|
|
DNOT_REACHED();
|
|
ARC_RELEASE(self);
|
|
return nil;
|
|
}
|
|
|
|
if ((self = [super initWithContentType:type contentLength:data.length])) {
|
|
_data = ARC_RETAIN(data);
|
|
_offset = -1;
|
|
}
|
|
return self;
|
|
}
|
|
|
|
- (void)dealloc {
|
|
DCHECK(_offset < 0);
|
|
ARC_RELEASE(_data);
|
|
|
|
ARC_DEALLOC(super);
|
|
}
|
|
|
|
- (BOOL)open {
|
|
DCHECK(_offset < 0);
|
|
_offset = 0;
|
|
return YES;
|
|
}
|
|
|
|
- (NSInteger)read:(void*)buffer maxLength:(NSUInteger)length {
|
|
DCHECK(_offset >= 0);
|
|
NSInteger size = 0;
|
|
if (_offset < (NSInteger)_data.length) {
|
|
size = MIN(_data.length - _offset, length);
|
|
bcopy((char*)_data.bytes + _offset, buffer, size);
|
|
_offset += size;
|
|
}
|
|
return size;
|
|
}
|
|
|
|
- (BOOL)close {
|
|
DCHECK(_offset >= 0);
|
|
_offset = -1;
|
|
return YES;
|
|
}
|
|
|
|
@end
|
|
|
|
@implementation GCDWebServerDataResponse (Extensions)
|
|
|
|
+ (GCDWebServerDataResponse*)responseWithText:(NSString*)text {
|
|
return ARC_AUTORELEASE([[self alloc] initWithText:text]);
|
|
}
|
|
|
|
+ (GCDWebServerDataResponse*)responseWithHTML:(NSString*)html {
|
|
return ARC_AUTORELEASE([[self alloc] initWithHTML:html]);
|
|
}
|
|
|
|
+ (GCDWebServerDataResponse*)responseWithHTMLTemplate:(NSString*)path variables:(NSDictionary*)variables {
|
|
return ARC_AUTORELEASE([[self alloc] initWithHTMLTemplate:path variables:variables]);
|
|
}
|
|
|
|
+ (GCDWebServerDataResponse*)responseWithJSONObject:(id)object {
|
|
return ARC_AUTORELEASE([[self alloc] initWithJSONObject:object]);
|
|
}
|
|
|
|
+ (GCDWebServerDataResponse*)responseWithJSONObject:(id)object contentType:(NSString*)type {
|
|
return ARC_AUTORELEASE([[self alloc] initWithJSONObject:object contentType:type]);
|
|
}
|
|
|
|
- (id)initWithText:(NSString*)text {
|
|
NSData* data = [text dataUsingEncoding:NSUTF8StringEncoding];
|
|
if (data == nil) {
|
|
DNOT_REACHED();
|
|
ARC_RELEASE(self);
|
|
return nil;
|
|
}
|
|
return [self initWithData:data contentType:@"text/plain; charset=utf-8"];
|
|
}
|
|
|
|
- (id)initWithHTML:(NSString*)html {
|
|
NSData* data = [html dataUsingEncoding:NSUTF8StringEncoding];
|
|
if (data == nil) {
|
|
DNOT_REACHED();
|
|
ARC_RELEASE(self);
|
|
return nil;
|
|
}
|
|
return [self initWithData:data contentType:@"text/html; charset=utf-8"];
|
|
}
|
|
|
|
- (id)initWithHTMLTemplate:(NSString*)path variables:(NSDictionary*)variables {
|
|
NSMutableString* html = [[NSMutableString alloc] initWithContentsOfFile:path encoding:NSUTF8StringEncoding error:NULL];
|
|
[variables enumerateKeysAndObjectsUsingBlock:^(NSString* key, NSString* value, BOOL* stop) {
|
|
[html replaceOccurrencesOfString:[NSString stringWithFormat:@"%%%@%%", key] withString:value options:0 range:NSMakeRange(0, html.length)];
|
|
}];
|
|
id response = [self initWithHTML:html];
|
|
ARC_RELEASE(html);
|
|
return response;
|
|
}
|
|
|
|
- (id)initWithJSONObject:(id)object {
|
|
return [self initWithJSONObject:object contentType:@"application/json"];
|
|
}
|
|
|
|
- (id)initWithJSONObject:(id)object contentType:(NSString*)type {
|
|
NSData* data = [NSJSONSerialization dataWithJSONObject:object options:0 error:NULL];
|
|
if (data == nil) {
|
|
ARC_RELEASE(self);
|
|
return nil;
|
|
}
|
|
return [self initWithData:data contentType:type];
|
|
}
|
|
|
|
@end
|
|
|
|
@implementation GCDWebServerFileResponse
|
|
|
|
+ (GCDWebServerFileResponse*)responseWithFile:(NSString*)path {
|
|
return ARC_AUTORELEASE([[[self class] alloc] initWithFile:path]);
|
|
}
|
|
|
|
+ (GCDWebServerFileResponse*)responseWithFile:(NSString*)path isAttachment:(BOOL)attachment {
|
|
return ARC_AUTORELEASE([[[self class] alloc] initWithFile:path isAttachment:attachment]);
|
|
}
|
|
|
|
+ (GCDWebServerFileResponse*)responseWithFile:(NSString*)path byteRange:(NSRange)range {
|
|
return ARC_AUTORELEASE([[[self class] alloc] initWithFile:path byteRange:range]);
|
|
}
|
|
|
|
+ (GCDWebServerFileResponse*)responseWithFile:(NSString*)path byteRange:(NSRange)range isAttachment:(BOOL)attachment {
|
|
return ARC_AUTORELEASE([[[self class] alloc] initWithFile:path byteRange:range isAttachment:attachment]);
|
|
}
|
|
|
|
- (id)initWithFile:(NSString*)path {
|
|
return [self initWithFile:path byteRange:NSMakeRange(NSNotFound, 0) isAttachment:NO];
|
|
}
|
|
|
|
- (id)initWithFile:(NSString*)path isAttachment:(BOOL)attachment {
|
|
return [self initWithFile:path byteRange:NSMakeRange(NSNotFound, 0) isAttachment:attachment];
|
|
}
|
|
|
|
- (id)initWithFile:(NSString*)path byteRange:(NSRange)range {
|
|
return [self initWithFile:path byteRange:range isAttachment:NO];
|
|
}
|
|
|
|
- (id)initWithFile:(NSString*)path byteRange:(NSRange)range isAttachment:(BOOL)attachment {
|
|
struct stat info;
|
|
if (lstat([path fileSystemRepresentation], &info) || !(info.st_mode & S_IFREG)) {
|
|
DNOT_REACHED();
|
|
ARC_RELEASE(self);
|
|
return nil;
|
|
}
|
|
if ((range.location != NSNotFound) || (range.length > 0)) {
|
|
if (range.location != NSNotFound) {
|
|
range.location = MIN(range.location, (NSUInteger)info.st_size);
|
|
range.length = MIN(range.length, (NSUInteger)info.st_size - range.location);
|
|
} else {
|
|
range.length = MIN(range.length, (NSUInteger)info.st_size);
|
|
range.location = (NSUInteger)info.st_size - range.length;
|
|
}
|
|
if (range.length == 0) {
|
|
ARC_RELEASE(self);
|
|
return nil; // TODO: Return 416 status code and "Content-Range: bytes */{file length}" header
|
|
}
|
|
}
|
|
|
|
if ((self = [super initWithContentType:GCDWebServerGetMimeTypeForExtension([path pathExtension]) contentLength:(range.location != NSNotFound ? range.length : (NSUInteger)info.st_size)])) {
|
|
_path = [path copy];
|
|
if (range.location != NSNotFound) {
|
|
_offset = range.location;
|
|
_size = range.length;
|
|
[self setStatusCode:206];
|
|
[self setValue:[NSString stringWithFormat:@"bytes %i-%i/%i", (int)range.location, (int)(range.location + range.length - 1), (int)info.st_size] forAdditionalHeader:@"Content-Range"];
|
|
LOG_DEBUG(@"Using content bytes range [%i-%i] for file \"%@\"", (int)range.location, (int)(range.location + range.length - 1), path);
|
|
} else {
|
|
_offset = 0;
|
|
_size = (NSUInteger)info.st_size;
|
|
}
|
|
if (attachment) { // TODO: Use http://tools.ietf.org/html/rfc5987 to encode file names with special characters instead of using lossy conversion to ISO 8859-1
|
|
NSData* data = [[path lastPathComponent] dataUsingEncoding:NSISOLatin1StringEncoding allowLossyConversion:YES];
|
|
NSString* fileName = data ? [[NSString alloc] initWithData:data encoding:NSISOLatin1StringEncoding] : nil;
|
|
if (fileName) {
|
|
[self setValue:[NSString stringWithFormat:@"attachment; filename=\"%@\"", fileName] forAdditionalHeader:@"Content-Disposition"];
|
|
ARC_RELEASE(fileName);
|
|
} else {
|
|
DNOT_REACHED();
|
|
}
|
|
}
|
|
}
|
|
return self;
|
|
}
|
|
|
|
- (void)dealloc {
|
|
DCHECK(_file <= 0);
|
|
ARC_RELEASE(_path);
|
|
|
|
ARC_DEALLOC(super);
|
|
}
|
|
|
|
- (BOOL)open {
|
|
DCHECK(_file <= 0);
|
|
_file = open([_path fileSystemRepresentation], O_NOFOLLOW | O_RDONLY);
|
|
if (_file <= 0) {
|
|
return NO;
|
|
}
|
|
if (lseek(_file, _offset, SEEK_SET) != (off_t)_offset) {
|
|
close(_file);
|
|
_file = 0;
|
|
return NO;
|
|
}
|
|
return YES;
|
|
}
|
|
|
|
- (NSInteger)read:(void*)buffer maxLength:(NSUInteger)length {
|
|
DCHECK(_file > 0);
|
|
ssize_t result = read(_file, buffer, MIN(length, _size));
|
|
if (result > 0) {
|
|
_size -= result;
|
|
}
|
|
return result;
|
|
}
|
|
|
|
- (BOOL)close {
|
|
DCHECK(_file > 0);
|
|
int result = close(_file);
|
|
_file = 0;
|
|
return (result == 0 ? YES : NO);
|
|
}
|
|
|
|
@end
|