mirror of
https://github.com/swisspol/GCDWebServer.git
synced 2024-10-07 08:32:16 +08:00
288 lines
8.2 KiB
Objective-C
288 lines
8.2 KiB
Objective-C
/*
|
|
Copyright (c) 2012-2013, 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"
|
|
|
|
@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 < _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]);
|
|
}
|
|
|
|
- (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;
|
|
}
|
|
|
|
@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]);
|
|
}
|
|
|
|
- (id)initWithFile:(NSString*)path {
|
|
return [self initWithFile:path isAttachment:NO];
|
|
}
|
|
|
|
- (id)initWithFile:(NSString*)path isAttachment:(BOOL)attachment {
|
|
struct stat info;
|
|
if (lstat([path fileSystemRepresentation], &info) || !(info.st_mode & S_IFREG)) {
|
|
DNOT_REACHED();
|
|
ARC_RELEASE(self);
|
|
return nil;
|
|
}
|
|
NSString* type = GCDWebServerGetMimeTypeForExtension([path pathExtension]);
|
|
if (type == nil) {
|
|
type = kGCDWebServerDefaultMimeType;
|
|
}
|
|
|
|
if ((self = [super initWithContentType:type contentLength:info.st_size])) {
|
|
_path = [path copy];
|
|
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);
|
|
return (_file > 0 ? YES : NO);
|
|
}
|
|
|
|
- (NSInteger)read:(void*)buffer maxLength:(NSUInteger)length {
|
|
DCHECK(_file > 0);
|
|
return read(_file, buffer, length);
|
|
}
|
|
|
|
- (BOOL)close {
|
|
DCHECK(_file > 0);
|
|
int result = close(_file);
|
|
_file = 0;
|
|
return (result == 0 ? YES : NO);
|
|
}
|
|
|
|
@end
|