/*
* Copyright (c) 2008, 2009, 2010, 2011, 2012, 2013, 2014, 2015, 2016, 2017
* Jonathan Schleifer <js@heap.zone>
*
* All rights reserved.
*
* This file is part of ObjFW. It may be distributed under the terms of the
* Q Public License 1.0, which can be found in the file LICENSE.QPL included in
* the packaging of this file.
*
* Alternatively, it may be distributed under the terms of the GNU General
* Public License, either version 2 or 3, which can be found in the file
* LICENSE.GPLv2 or LICENSE.GPLv3 respectively included in the packaging of this
* file.
*/
#include "config.h"
#import "OFHTTPResponse.h"
#import "OFString.h"
#import "OFDictionary.h"
#import "OFArray.h"
#import "OFDataArray.h"
#import "OFHTTPCookie.h"
#import "OFInvalidEncodingException.h"
#import "OFInvalidFormatException.h"
#import "OFOutOfRangeException.h"
#import "OFTruncatedDataException.h"
#import "OFUnsupportedVersionException.h"
static of_string_encoding_t
encodingForContentType(OFString *contentType)
{
const char *UTF8String = [contentType UTF8String];
size_t last, length = [contentType UTF8StringLength];
enum {
STATE_TYPE,
STATE_BEFORE_PARAM_NAME,
STATE_PARAM_NAME,
STATE_PARAM_VALUE_OR_QUOTE,
STATE_PARAM_VALUE,
STATE_PARAM_QUOTED_VALUE,
STATE_AFTER_PARAM_VALUE
} state = STATE_TYPE;
OFString *name = nil, *value = nil, *charset = nil;
of_string_encoding_t ret;
last = 0;
for (size_t i = 0; i < length; i++) {
switch (state) {
case STATE_TYPE:
if (UTF8String[i] == ';') {
state = STATE_BEFORE_PARAM_NAME;
last = i + 1;
}
break;
case STATE_BEFORE_PARAM_NAME:
if (UTF8String[i] == ' ')
last = i + 1;
else {
state = STATE_PARAM_NAME;
i--;
}
break;
case STATE_PARAM_NAME:
if (UTF8String[i] == '=') {
name = [OFString
stringWithUTF8String: UTF8String + last
length: i - last];
state = STATE_PARAM_VALUE_OR_QUOTE;
last = i + 1;
}
break;
case STATE_PARAM_VALUE_OR_QUOTE:
if (UTF8String[i] == '"') {
state = STATE_PARAM_QUOTED_VALUE;
last = i + 1;
} else {
state = STATE_PARAM_VALUE;
i--;
}
break;
case STATE_PARAM_VALUE:
if (UTF8String[i] == ';') {
value = [OFString
stringWithUTF8String: UTF8String + last
length: i - last];
value = [value
stringByDeletingTrailingWhitespaces];
if ([name isEqual: @"charset"])
charset = value;
state = STATE_BEFORE_PARAM_NAME;
last = i + 1;
}
break;
case STATE_PARAM_QUOTED_VALUE:
if (UTF8String[i] == '"') {
value = [OFString
stringWithUTF8String: UTF8String + last
length: i - last];
if ([name isEqual: @"charset"])
charset = value;
state = STATE_AFTER_PARAM_VALUE;
}
break;
case STATE_AFTER_PARAM_VALUE:
if (UTF8String[i] == ';') {
state = STATE_BEFORE_PARAM_NAME;
last = i + 1;
} else if (UTF8String[i] != ' ')
return OF_STRING_ENCODING_AUTODETECT;
break;
}
}
if (state == STATE_PARAM_VALUE) {
value = [OFString stringWithUTF8String: UTF8String + last
length: length - last];
value = [value stringByDeletingTrailingWhitespaces];
if ([name isEqual: @"charset"])
charset = value;
}
@try {
ret = of_string_parse_encoding(charset);
} @catch (OFInvalidEncodingException *e) {
ret = OF_STRING_ENCODING_AUTODETECT;
}
return ret;
}
@implementation OFHTTPResponse
@synthesize statusCode = _statusCode, headers = _headers, cookies = _cookies;
- init
{
self = [super init];
_protocolVersion.major = 1;
_protocolVersion.minor = 1;
return self;
}
- (void)dealloc
{
[_headers release];
[_cookies release];
[super dealloc];
}
- (void)setProtocolVersion: (of_http_request_protocol_version_t)protocolVersion
{
if (protocolVersion.major != 1 || protocolVersion.minor > 1)
@throw [OFUnsupportedVersionException exceptionWithVersion:
[OFString stringWithFormat: @"%u.%u",
protocolVersion.major,
protocolVersion.minor]];
_protocolVersion = protocolVersion;
}
- (of_http_request_protocol_version_t)protocolVersion
{
return _protocolVersion;
}
- (void)setProtocolVersionFromString: (OFString *)string
{
void *pool = objc_autoreleasePoolPush();
OFArray *components = [string componentsSeparatedByString: @"."];
intmax_t major, minor;
of_http_request_protocol_version_t protocolVersion;
if ([components count] != 2)
@throw [OFInvalidFormatException exception];
major = [[components firstObject] decimalValue];
minor = [[components lastObject] decimalValue];
if (major < 0 || major > UINT8_MAX || minor < 0 || minor > UINT8_MAX)
@throw [OFOutOfRangeException exception];
protocolVersion.major = (uint8_t)major;
protocolVersion.minor = (uint8_t)minor;
[self setProtocolVersion: protocolVersion];
objc_autoreleasePoolPop(pool);
}
- (OFString *)protocolVersionString
{
return [OFString stringWithFormat: @"%u.%u",
_protocolVersion.major,
_protocolVersion.minor];
}
- (OFString *)string
{
return [self stringWithEncoding: OF_STRING_ENCODING_AUTODETECT];
}
- (OFString *)stringWithEncoding: (of_string_encoding_t)encoding
{
void *pool = objc_autoreleasePoolPush();
OFString *contentType, *contentLength, *ret;
OFDataArray *data;
if (encoding == OF_STRING_ENCODING_AUTODETECT &&
(contentType = [_headers objectForKey: @"Content-Type"]) != nil)
encoding = encodingForContentType(contentType);
if (encoding == OF_STRING_ENCODING_AUTODETECT)
encoding = OF_STRING_ENCODING_UTF_8;
data = [self readDataArrayTillEndOfStream];
if ((contentLength = [_headers objectForKey: @"Content-Length"]) != nil)
if ([data count] != (size_t)[contentLength decimalValue])
@throw [OFTruncatedDataException exception];
ret = [[OFString alloc] initWithCString: (char *)[data items]
encoding: encoding
length: [data count]];
objc_autoreleasePoolPop(pool);
return [ret autorelease];
}
- (OFString *)description
{
void *pool = objc_autoreleasePoolPush();
OFString *indentedHeaders, *ret;
indentedHeaders = [[_headers description]
stringByReplacingOccurrencesOfString: @"\n"
withString: @"\n\t"];
ret = [[OFString alloc] initWithFormat:
@"<%@:\n"
@"\tStatus code = %d\n"
@"\tHeaders = %@\n"
@">",
[self class], _statusCode, indentedHeaders];
objc_autoreleasePoolPop(pool);
return [ret autorelease];
}
@end