GCDWebServer.m 28 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772
  1. /*
  2. Copyright (c) 2012-2014, Pierre-Olivier Latour
  3. All rights reserved.
  4. Redistribution and use in source and binary forms, with or without
  5. modification, are permitted provided that the following conditions are met:
  6. * Redistributions of source code must retain the above copyright
  7. notice, this list of conditions and the following disclaimer.
  8. * Redistributions in binary form must reproduce the above copyright
  9. notice, this list of conditions and the following disclaimer in the
  10. documentation and/or other materials provided with the distribution.
  11. * The name of Pierre-Olivier Latour may not be used to endorse
  12. or promote products derived from this software without specific
  13. prior written permission.
  14. THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND
  15. ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
  16. WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
  17. DISCLAIMED. IN NO EVENT SHALL PIERRE-OLIVIER LATOUR BE LIABLE FOR ANY
  18. DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
  19. (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
  20. LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND
  21. ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
  22. (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
  23. SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  24. */
  25. #import <TargetConditionals.h>
  26. #import <netinet/in.h>
  27. #import "GCDWebServerPrivate.h"
  28. #if TARGET_OS_IPHONE && !TARGET_IPHONE_SIMULATOR
  29. #define kDefaultPort 80
  30. #else
  31. #define kDefaultPort 8080
  32. #endif
  33. #define kMaxPendingConnections 16
  34. @interface GCDWebServer () {
  35. @private
  36. NSMutableArray* _handlers;
  37. NSUInteger _port;
  38. dispatch_source_t _source;
  39. CFNetServiceRef _service;
  40. #if !TARGET_OS_IPHONE
  41. BOOL _recording;
  42. #endif
  43. }
  44. @end
  45. @interface GCDWebServerHandler () {
  46. @private
  47. GCDWebServerMatchBlock _matchBlock;
  48. GCDWebServerProcessBlock _processBlock;
  49. }
  50. @end
  51. #ifndef __GCDWEBSERVER_LOGGING_HEADER__
  52. #ifdef NDEBUG
  53. GCDWebServerLogLevel GCDLogLevel = kGCDWebServerLogLevel_Info;
  54. #else
  55. GCDWebServerLogLevel GCDLogLevel = kGCDWebServerLogLevel_Debug;
  56. #endif
  57. #endif
  58. #if !TARGET_OS_IPHONE
  59. static BOOL _run;
  60. #endif
  61. #ifndef __GCDWEBSERVER_LOGGING_HEADER__
  62. void GCDLogMessage(GCDWebServerLogLevel level, NSString* format, ...) {
  63. static const char* levelNames[] = {"DEBUG", "VERBOSE", "INFO", "WARNING", "ERROR", "EXCEPTION"};
  64. va_list arguments;
  65. va_start(arguments, format);
  66. NSString* message = [[NSString alloc] initWithFormat:format arguments:arguments];
  67. va_end(arguments);
  68. fprintf(stderr, "[%s] %s\n", levelNames[level], [message UTF8String]);
  69. ARC_RELEASE(message);
  70. }
  71. #endif
  72. #if !TARGET_OS_IPHONE
  73. static void _SignalHandler(int signal) {
  74. _run = NO;
  75. printf("\n");
  76. }
  77. #endif
  78. @implementation GCDWebServerHandler
  79. @synthesize matchBlock=_matchBlock, processBlock=_processBlock;
  80. - (id)initWithMatchBlock:(GCDWebServerMatchBlock)matchBlock processBlock:(GCDWebServerProcessBlock)processBlock {
  81. if ((self = [super init])) {
  82. _matchBlock = [matchBlock copy];
  83. _processBlock = [processBlock copy];
  84. }
  85. return self;
  86. }
  87. - (void)dealloc {
  88. ARC_RELEASE(_matchBlock);
  89. ARC_RELEASE(_processBlock);
  90. ARC_DEALLOC(super);
  91. }
  92. @end
  93. @implementation GCDWebServer
  94. @synthesize handlers=_handlers, port=_port;
  95. #ifndef __GCDWEBSERVER_LOGGING_HEADER__
  96. + (void)load {
  97. const char* logLevel = getenv("logLevel");
  98. if (logLevel) {
  99. GCDLogLevel = atoi(logLevel);
  100. }
  101. }
  102. #endif
  103. + (void)initialize {
  104. GCDWebServerInitializeFunctions();
  105. }
  106. - (instancetype)init {
  107. if ((self = [super init])) {
  108. _handlers = [[NSMutableArray alloc] init];
  109. }
  110. return self;
  111. }
  112. - (void)dealloc {
  113. if (_source) {
  114. [self stop];
  115. }
  116. ARC_RELEASE(_handlers);
  117. ARC_DEALLOC(super);
  118. }
  119. - (NSString*)bonjourName {
  120. CFStringRef name = _service ? CFNetServiceGetName(_service) : NULL;
  121. return name && CFStringGetLength(name) ? ARC_BRIDGE_RELEASE(CFStringCreateCopy(kCFAllocatorDefault, name)) : nil;
  122. }
  123. - (void)addHandlerWithMatchBlock:(GCDWebServerMatchBlock)matchBlock processBlock:(GCDWebServerProcessBlock)handlerBlock {
  124. DCHECK(_source == NULL);
  125. GCDWebServerHandler* handler = [[GCDWebServerHandler alloc] initWithMatchBlock:matchBlock processBlock:handlerBlock];
  126. [_handlers insertObject:handler atIndex:0];
  127. ARC_RELEASE(handler);
  128. }
  129. - (void)removeAllHandlers {
  130. DCHECK(_source == NULL);
  131. [_handlers removeAllObjects];
  132. }
  133. - (BOOL)start {
  134. return [self startWithPort:kDefaultPort bonjourName:@""];
  135. }
  136. static void _NetServiceClientCallBack(CFNetServiceRef service, CFStreamError* error, void* info) {
  137. @autoreleasepool {
  138. if (error->error) {
  139. LOG_ERROR(@"Bonjour error %i (domain %i)", (int)error->error, (int)error->domain);
  140. } else {
  141. GCDWebServer* server = (ARC_BRIDGE GCDWebServer*)info;
  142. LOG_INFO(@"%@ now reachable at %@", [server class], server.bonjourServerURL);
  143. }
  144. }
  145. }
  146. - (BOOL)startWithPort:(NSUInteger)port bonjourName:(NSString*)name {
  147. DCHECK(_source == NULL);
  148. int listeningSocket = socket(PF_INET, SOCK_STREAM, IPPROTO_TCP);
  149. if (listeningSocket > 0) {
  150. int yes = 1;
  151. setsockopt(listeningSocket, SOL_SOCKET, SO_REUSEADDR, &yes, sizeof(yes));
  152. struct sockaddr_in addr4;
  153. bzero(&addr4, sizeof(addr4));
  154. addr4.sin_len = sizeof(addr4);
  155. addr4.sin_family = AF_INET;
  156. addr4.sin_port = htons(port);
  157. addr4.sin_addr.s_addr = htonl(INADDR_ANY);
  158. if (bind(listeningSocket, (void*)&addr4, sizeof(addr4)) == 0) {
  159. if (listen(listeningSocket, kMaxPendingConnections) == 0) {
  160. _source = dispatch_source_create(DISPATCH_SOURCE_TYPE_READ, listeningSocket, 0, kGCDWebServerGCDQueue);
  161. dispatch_source_set_cancel_handler(_source, ^{
  162. @autoreleasepool {
  163. int result = close(listeningSocket);
  164. if (result != 0) {
  165. LOG_ERROR(@"Failed closing socket (%i): %s", errno, strerror(errno));
  166. } else {
  167. LOG_DEBUG(@"Closed listening socket");
  168. }
  169. }
  170. });
  171. dispatch_source_set_event_handler(_source, ^{
  172. @autoreleasepool {
  173. struct sockaddr remoteSockAddr;
  174. socklen_t remoteAddrLen = sizeof(remoteSockAddr);
  175. int socket = accept(listeningSocket, &remoteSockAddr, &remoteAddrLen);
  176. if (socket > 0) {
  177. NSData* remoteAddress = [NSData dataWithBytes:&remoteSockAddr length:remoteAddrLen];
  178. struct sockaddr localSockAddr;
  179. socklen_t localAddrLen = sizeof(localSockAddr);
  180. NSData* localAddress = nil;
  181. if (getsockname(socket, &localSockAddr, &localAddrLen) == 0) {
  182. localAddress = [NSData dataWithBytes:&localSockAddr length:localAddrLen];
  183. } else {
  184. DNOT_REACHED();
  185. }
  186. int noSigPipe = 1;
  187. setsockopt(socket, SOL_SOCKET, SO_NOSIGPIPE, &noSigPipe, sizeof(noSigPipe)); // Make sure this socket cannot generate SIG_PIPE
  188. Class connectionClass = [[self class] connectionClass];
  189. GCDWebServerConnection* connection = [[connectionClass alloc] initWithServer:self localAddress:localAddress remoteAddress:remoteAddress socket:socket]; // Connection will automatically retain itself while opened
  190. #if __has_feature(objc_arc)
  191. [connection self]; // Prevent compiler from complaining about unused variable / useless statement
  192. #else
  193. [connection release];
  194. #endif
  195. } else {
  196. LOG_ERROR(@"Failed accepting socket (%i): %s", errno, strerror(errno));
  197. }
  198. }
  199. });
  200. if (port == 0) { // Determine the actual port we are listening on
  201. struct sockaddr addr;
  202. socklen_t addrlen = sizeof(addr);
  203. if (getsockname(listeningSocket, &addr, &addrlen) == 0) {
  204. struct sockaddr_in* sockaddr = (struct sockaddr_in*)&addr;
  205. _port = ntohs(sockaddr->sin_port);
  206. } else {
  207. LOG_ERROR(@"Failed retrieving socket address (%i): %s", errno, strerror(errno));
  208. }
  209. } else {
  210. _port = port;
  211. }
  212. if (name) {
  213. _service = CFNetServiceCreate(kCFAllocatorDefault, CFSTR("local."), CFSTR("_http._tcp"), (ARC_BRIDGE CFStringRef)name, (SInt32)_port);
  214. if (_service) {
  215. CFNetServiceClientContext context = {0, (ARC_BRIDGE void*)self, NULL, NULL, NULL};
  216. CFNetServiceSetClient(_service, _NetServiceClientCallBack, &context);
  217. CFNetServiceScheduleWithRunLoop(_service, CFRunLoopGetMain(), kCFRunLoopCommonModes);
  218. CFStreamError error = {0};
  219. CFNetServiceRegisterWithOptions(_service, 0, &error);
  220. } else {
  221. LOG_ERROR(@"Failed creating CFNetService");
  222. }
  223. }
  224. dispatch_resume(_source);
  225. LOG_INFO(@"%@ started on port %i and reachable at %@", [self class], (int)_port, self.serverURL);
  226. } else {
  227. LOG_ERROR(@"Failed listening on socket (%i): %s", errno, strerror(errno));
  228. close(listeningSocket);
  229. }
  230. } else {
  231. LOG_ERROR(@"Failed binding socket (%i): %s", errno, strerror(errno));
  232. close(listeningSocket);
  233. }
  234. } else {
  235. LOG_ERROR(@"Failed creating socket (%i): %s", errno, strerror(errno));
  236. }
  237. return (_source ? YES : NO);
  238. }
  239. - (BOOL)isRunning {
  240. return (_source ? YES : NO);
  241. }
  242. - (void)stop {
  243. DCHECK(_source != NULL);
  244. if (_source) {
  245. if (_service) {
  246. CFNetServiceUnscheduleFromRunLoop(_service, CFRunLoopGetMain(), kCFRunLoopCommonModes);
  247. CFNetServiceSetClient(_service, NULL, NULL);
  248. CFRelease(_service);
  249. _service = NULL;
  250. }
  251. dispatch_source_cancel(_source); // This will close the socket
  252. ARC_DISPATCH_RELEASE(_source);
  253. _source = NULL;
  254. LOG_INFO(@"%@ stopped", [self class]);
  255. }
  256. _port = 0;
  257. }
  258. @end
  259. @implementation GCDWebServer (Subclassing)
  260. + (Class)connectionClass {
  261. return [GCDWebServerConnection class];
  262. }
  263. + (NSString*)serverName {
  264. return NSStringFromClass(self);
  265. }
  266. + (BOOL)shouldAutomaticallyMapHEADToGET {
  267. return YES;
  268. }
  269. @end
  270. @implementation GCDWebServer (Extensions)
  271. #if !TARGET_OS_IPHONE
  272. - (void)setRecordingEnabled:(BOOL)flag {
  273. _recording = flag;
  274. }
  275. - (BOOL)isRecordingEnabled {
  276. return _recording;
  277. }
  278. #endif
  279. - (NSURL*)serverURL {
  280. if (_source) {
  281. NSString* ipAddress = GCDWebServerGetPrimaryIPv4Address();
  282. if (ipAddress) {
  283. if (_port != 80) {
  284. return [NSURL URLWithString:[NSString stringWithFormat:@"http://%@:%i/", ipAddress, (int)_port]];
  285. } else {
  286. return [NSURL URLWithString:[NSString stringWithFormat:@"http://%@/", ipAddress]];
  287. }
  288. }
  289. }
  290. return nil;
  291. }
  292. - (NSURL*)bonjourServerURL {
  293. if (_source && _service) {
  294. CFStringRef name = CFNetServiceGetName(_service);
  295. if (name && CFStringGetLength(name)) {
  296. if (_port != 80) {
  297. return [NSURL URLWithString:[NSString stringWithFormat:@"http://%@.local:%i/", name, (int)_port]];
  298. } else {
  299. return [NSURL URLWithString:[NSString stringWithFormat:@"http://%@.local/", name]];
  300. }
  301. }
  302. }
  303. return nil;
  304. }
  305. #if !TARGET_OS_IPHONE
  306. - (BOOL)runWithPort:(NSUInteger)port {
  307. BOOL success = NO;
  308. _run = YES;
  309. void (*handler)(int) = signal(SIGINT, _SignalHandler);
  310. if (handler != SIG_ERR) {
  311. if ([self startWithPort:port bonjourName:@""]) {
  312. while (_run) {
  313. CFRunLoopRunInMode(kCFRunLoopDefaultMode, 1.0, true);
  314. }
  315. [self stop];
  316. success = YES;
  317. }
  318. signal(SIGINT, handler);
  319. }
  320. return success;
  321. }
  322. #endif
  323. #ifdef __GCDWEBSERVER_ENABLE_TESTING__
  324. static CFHTTPMessageRef _CreateHTTPMessageFromData(NSData* data, BOOL isRequest) {
  325. CFHTTPMessageRef message = CFHTTPMessageCreateEmpty(kCFAllocatorDefault, isRequest);
  326. if (CFHTTPMessageAppendBytes(message, data.bytes, data.length)) {
  327. return message;
  328. }
  329. CFRelease(message);
  330. return NULL;
  331. }
  332. static CFHTTPMessageRef _CreateHTTPMessageFromPerformingRequest(NSData* inData, NSUInteger port) {
  333. CFHTTPMessageRef response = NULL;
  334. int httpSocket = socket(PF_INET, SOCK_STREAM, IPPROTO_TCP);
  335. if (httpSocket > 0) {
  336. struct sockaddr_in addr4;
  337. bzero(&addr4, sizeof(addr4));
  338. addr4.sin_len = sizeof(port);
  339. addr4.sin_family = AF_INET;
  340. addr4.sin_port = htons(8080);
  341. addr4.sin_addr.s_addr = htonl(INADDR_ANY);
  342. if (connect(httpSocket, (void*)&addr4, sizeof(addr4)) == 0) {
  343. if (write(httpSocket, inData.bytes, inData.length) == (ssize_t)inData.length) {
  344. NSMutableData* outData = [[NSMutableData alloc] initWithLength:(256 * 1024)];
  345. NSUInteger length = 0;
  346. while (1) {
  347. ssize_t result = read(httpSocket, (char*)outData.mutableBytes + length, outData.length - length);
  348. if (result < 0) {
  349. length = NSNotFound;
  350. break;
  351. } else if (result == 0) {
  352. break;
  353. }
  354. length += result;
  355. if (length >= outData.length) {
  356. outData.length = 2 * outData.length;
  357. }
  358. }
  359. if (length != NSNotFound) {
  360. outData.length = length;
  361. response = _CreateHTTPMessageFromData(outData, NO);
  362. } else {
  363. DNOT_REACHED();
  364. }
  365. ARC_RELEASE(outData);
  366. }
  367. }
  368. close(httpSocket);
  369. }
  370. return response;
  371. }
  372. static void _LogResult(NSString* format, ...) {
  373. va_list arguments;
  374. va_start(arguments, format);
  375. NSString* message = [[NSString alloc] initWithFormat:format arguments:arguments];
  376. va_end(arguments);
  377. fprintf(stdout, "%s\n", [message UTF8String]);
  378. ARC_RELEASE(message);
  379. }
  380. - (NSInteger)runTestsInDirectory:(NSString*)path withPort:(NSUInteger)port {
  381. NSArray* ignoredHeaders = @[@"Date", @"Etag"]; // Dates are always different by definition and ETags depend on file system node IDs
  382. NSInteger result = -1;
  383. if ([self startWithPort:port bonjourName:nil]) {
  384. result = 0;
  385. NSArray* files = [[NSFileManager defaultManager] contentsOfDirectoryAtPath:path error:NULL];
  386. for (NSString* requestFile in files) {
  387. if (![requestFile hasSuffix:@".request"]) {
  388. continue;
  389. }
  390. @autoreleasepool {
  391. NSString* index = [[requestFile componentsSeparatedByString:@"-"] firstObject];
  392. BOOL success = NO;
  393. NSData* requestData = [NSData dataWithContentsOfFile:[path stringByAppendingPathComponent:requestFile]];
  394. if (requestData) {
  395. CFHTTPMessageRef request = _CreateHTTPMessageFromData(requestData, YES);
  396. if (request) {
  397. NSString* requestMethod = ARC_BRIDGE_RELEASE(CFHTTPMessageCopyRequestMethod(request));
  398. NSURL* requestURL = ARC_BRIDGE_RELEASE(CFHTTPMessageCopyRequestURL(request));
  399. _LogResult(@"[%i] %@ %@", (int)[index integerValue], requestMethod, requestURL.path);
  400. NSString* prefix = [index stringByAppendingString:@"-"];
  401. for (NSString* responseFile in files) {
  402. if ([responseFile hasPrefix:prefix] && [responseFile hasSuffix:@".response"]) {
  403. NSData* responseData = [NSData dataWithContentsOfFile:[path stringByAppendingPathComponent:responseFile]];
  404. if (responseData) {
  405. CFHTTPMessageRef expectedResponse = _CreateHTTPMessageFromData(responseData, NO);
  406. if (expectedResponse) {
  407. CFHTTPMessageRef actualResponse = _CreateHTTPMessageFromPerformingRequest(requestData, port);
  408. if (actualResponse) {
  409. success = YES;
  410. CFIndex expectedStatusCode = CFHTTPMessageGetResponseStatusCode(expectedResponse);
  411. CFIndex actualStatusCode = CFHTTPMessageGetResponseStatusCode(actualResponse);
  412. if (actualStatusCode != expectedStatusCode) {
  413. _LogResult(@" Status code not matching:\n Expected: %i\n Actual: %i", (int)expectedStatusCode, (int)actualStatusCode);
  414. success = NO;
  415. }
  416. NSDictionary* expectedHeaders = ARC_BRIDGE_RELEASE(CFHTTPMessageCopyAllHeaderFields(expectedResponse));
  417. NSDictionary* actualHeaders = ARC_BRIDGE_RELEASE(CFHTTPMessageCopyAllHeaderFields(actualResponse));
  418. for (NSString* expectedHeader in expectedHeaders) {
  419. if ([ignoredHeaders containsObject:expectedHeader]) {
  420. continue;
  421. }
  422. NSString* expectedValue = [expectedHeaders objectForKey:expectedHeader];
  423. NSString* actualValue = [actualHeaders objectForKey:expectedHeader];
  424. if (![actualValue isEqualToString:expectedValue]) {
  425. _LogResult(@" Header '%@' not matching:\n Expected: \"%@\"\n Actual: \"%@\"", expectedHeader, expectedValue, actualValue);
  426. success = NO;
  427. }
  428. }
  429. for (NSString* actualHeader in actualHeaders) {
  430. if (![expectedHeaders objectForKey:actualHeader]) {
  431. _LogResult(@" Header '%@' not matching:\n Expected: \"%@\"\n Actual: \"%@\"", actualHeader, nil, [actualHeaders objectForKey:actualHeader]);
  432. success = NO;
  433. }
  434. }
  435. NSData* expectedBody = ARC_BRIDGE_RELEASE(CFHTTPMessageCopyBody(expectedResponse));
  436. NSData* actualBody = ARC_BRIDGE_RELEASE(CFHTTPMessageCopyBody(actualResponse));
  437. if (![actualBody isEqualToData:expectedBody]) {
  438. _LogResult(@" Bodies not matching:\n Expected: %lu bytes\n Actual: %lu bytes", (unsigned long)expectedBody.length, (unsigned long)actualBody.length);
  439. success = NO;
  440. #ifndef NDEBUG
  441. if (GCDWebServerIsTextContentType([expectedHeaders objectForKey:@"Content-Type"])) {
  442. NSString* expectedPath = [NSTemporaryDirectory() stringByAppendingPathComponent:[[[NSProcessInfo processInfo] globallyUniqueString] stringByAppendingPathExtension:@"txt"]];
  443. NSString* actualPath = [NSTemporaryDirectory() stringByAppendingPathComponent:[[[NSProcessInfo processInfo] globallyUniqueString] stringByAppendingPathExtension:@"txt"]];
  444. if ([expectedBody writeToFile:expectedPath atomically:YES] && [actualBody writeToFile:actualPath atomically:YES]) {
  445. NSTask* task = [[NSTask alloc] init];
  446. [task setLaunchPath:@"/usr/bin/opendiff"];
  447. [task setArguments:@[expectedPath, actualPath]];
  448. [task launch];
  449. ARC_RELEASE(task);
  450. }
  451. }
  452. #endif
  453. }
  454. CFRelease(actualResponse);
  455. }
  456. CFRelease(expectedResponse);
  457. }
  458. } else {
  459. DNOT_REACHED();
  460. }
  461. break;
  462. }
  463. }
  464. CFRelease(request);
  465. }
  466. } else {
  467. DNOT_REACHED();
  468. }
  469. _LogResult(@"");
  470. if (!success) {
  471. ++result;
  472. }
  473. }
  474. }
  475. [self stop];
  476. }
  477. return result;
  478. }
  479. #endif
  480. @end
  481. @implementation GCDWebServer (Handlers)
  482. - (void)addDefaultHandlerForMethod:(NSString*)method requestClass:(Class)aClass processBlock:(GCDWebServerProcessBlock)block {
  483. [self addHandlerWithMatchBlock:^GCDWebServerRequest *(NSString* requestMethod, NSURL* requestURL, NSDictionary* requestHeaders, NSString* urlPath, NSDictionary* urlQuery) {
  484. if (![requestMethod isEqualToString:method]) {
  485. return nil;
  486. }
  487. return ARC_AUTORELEASE([[aClass alloc] initWithMethod:requestMethod url:requestURL headers:requestHeaders path:urlPath query:urlQuery]);
  488. } processBlock:block];
  489. }
  490. - (void)addHandlerForMethod:(NSString*)method path:(NSString*)path requestClass:(Class)aClass processBlock:(GCDWebServerProcessBlock)block {
  491. if ([path hasPrefix:@"/"] && [aClass isSubclassOfClass:[GCDWebServerRequest class]]) {
  492. [self addHandlerWithMatchBlock:^GCDWebServerRequest *(NSString* requestMethod, NSURL* requestURL, NSDictionary* requestHeaders, NSString* urlPath, NSDictionary* urlQuery) {
  493. if (![requestMethod isEqualToString:method]) {
  494. return nil;
  495. }
  496. if ([urlPath caseInsensitiveCompare:path] != NSOrderedSame) {
  497. return nil;
  498. }
  499. return ARC_AUTORELEASE([[aClass alloc] initWithMethod:requestMethod url:requestURL headers:requestHeaders path:urlPath query:urlQuery]);
  500. } processBlock:block];
  501. } else {
  502. DNOT_REACHED();
  503. }
  504. }
  505. - (void)addHandlerForMethod:(NSString*)method pathRegex:(NSString*)regex requestClass:(Class)aClass processBlock:(GCDWebServerProcessBlock)block {
  506. NSRegularExpression* expression = [NSRegularExpression regularExpressionWithPattern:regex options:NSRegularExpressionCaseInsensitive error:NULL];
  507. if (expression && [aClass isSubclassOfClass:[GCDWebServerRequest class]]) {
  508. [self addHandlerWithMatchBlock:^GCDWebServerRequest *(NSString* requestMethod, NSURL* requestURL, NSDictionary* requestHeaders, NSString* urlPath, NSDictionary* urlQuery) {
  509. if (![requestMethod isEqualToString:method]) {
  510. return nil;
  511. }
  512. if ([expression firstMatchInString:urlPath options:0 range:NSMakeRange(0, urlPath.length)] == nil) {
  513. return nil;
  514. }
  515. return ARC_AUTORELEASE([[aClass alloc] initWithMethod:requestMethod url:requestURL headers:requestHeaders path:urlPath query:urlQuery]);
  516. } processBlock:block];
  517. } else {
  518. DNOT_REACHED();
  519. }
  520. }
  521. @end
  522. @implementation GCDWebServer (GETHandlers)
  523. - (void)addGETHandlerForPath:(NSString*)path staticData:(NSData*)staticData contentType:(NSString*)contentType cacheAge:(NSUInteger)cacheAge {
  524. GCDWebServerResponse* response = [GCDWebServerDataResponse responseWithData:staticData contentType:contentType];
  525. response.cacheControlMaxAge = cacheAge;
  526. [self addHandlerForMethod:@"GET" path:path requestClass:[GCDWebServerRequest class] processBlock:^GCDWebServerResponse *(GCDWebServerRequest* request) {
  527. return response;
  528. }];
  529. }
  530. - (void)addGETHandlerForPath:(NSString*)path filePath:(NSString*)filePath isAttachment:(BOOL)isAttachment cacheAge:(NSUInteger)cacheAge allowRangeRequests:(BOOL)allowRangeRequests {
  531. [self addHandlerForMethod:@"GET" path:path requestClass:[GCDWebServerRequest class] processBlock:^GCDWebServerResponse *(GCDWebServerRequest* request) {
  532. GCDWebServerResponse* response = nil;
  533. if (allowRangeRequests) {
  534. response = [GCDWebServerFileResponse responseWithFile:filePath byteRange:request.byteRange isAttachment:isAttachment];
  535. [response setValue:@"bytes" forAdditionalHeader:@"Accept-Ranges"];
  536. } else {
  537. response = [GCDWebServerFileResponse responseWithFile:filePath isAttachment:isAttachment];
  538. }
  539. response.cacheControlMaxAge = cacheAge;
  540. return response;
  541. }];
  542. }
  543. - (GCDWebServerResponse*)_responseWithContentsOfDirectory:(NSString*)path {
  544. NSDirectoryEnumerator* enumerator = [[NSFileManager defaultManager] enumeratorAtPath:path];
  545. if (enumerator == nil) {
  546. return nil;
  547. }
  548. NSMutableString* html = [NSMutableString string];
  549. [html appendString:@"<!DOCTYPE html>\n"];
  550. [html appendString:@"<html><head><meta charset=\"utf-8\"></head><body>\n"];
  551. [html appendString:@"<ul>\n"];
  552. for (NSString* file in enumerator) {
  553. if (![file hasPrefix:@"."]) {
  554. NSString* type = [[enumerator fileAttributes] objectForKey:NSFileType];
  555. NSString* escapedFile = [file stringByAddingPercentEscapesUsingEncoding:NSUTF8StringEncoding];
  556. DCHECK(escapedFile);
  557. if ([type isEqualToString:NSFileTypeRegular]) {
  558. [html appendFormat:@"<li><a href=\"%@\">%@</a></li>\n", escapedFile, file];
  559. } else if ([type isEqualToString:NSFileTypeDirectory]) {
  560. [html appendFormat:@"<li><a href=\"%@/\">%@/</a></li>\n", escapedFile, file];
  561. }
  562. }
  563. [enumerator skipDescendents];
  564. }
  565. [html appendString:@"</ul>\n"];
  566. [html appendString:@"</body></html>\n"];
  567. return [GCDWebServerDataResponse responseWithHTML:html];
  568. }
  569. - (void)addGETHandlerForBasePath:(NSString*)basePath directoryPath:(NSString*)directoryPath indexFilename:(NSString*)indexFilename cacheAge:(NSUInteger)cacheAge allowRangeRequests:(BOOL)allowRangeRequests {
  570. if ([basePath hasPrefix:@"/"] && [basePath hasSuffix:@"/"]) {
  571. #if __has_feature(objc_arc)
  572. GCDWebServer* __unsafe_unretained server = self;
  573. #else
  574. __block GCDWebServer* server = self;
  575. #endif
  576. [self addHandlerWithMatchBlock:^GCDWebServerRequest *(NSString* requestMethod, NSURL* requestURL, NSDictionary* requestHeaders, NSString* urlPath, NSDictionary* urlQuery) {
  577. if (![requestMethod isEqualToString:@"GET"]) {
  578. return nil;
  579. }
  580. if (![urlPath hasPrefix:basePath]) {
  581. return nil;
  582. }
  583. return ARC_AUTORELEASE([[GCDWebServerRequest alloc] initWithMethod:requestMethod url:requestURL headers:requestHeaders path:urlPath query:urlQuery]);
  584. } processBlock:^GCDWebServerResponse *(GCDWebServerRequest* request) {
  585. GCDWebServerResponse* response = nil;
  586. NSString* filePath = [directoryPath stringByAppendingPathComponent:[request.path substringFromIndex:basePath.length]];
  587. BOOL isDirectory;
  588. if ([[NSFileManager defaultManager] fileExistsAtPath:filePath isDirectory:&isDirectory]) {
  589. if (isDirectory) {
  590. if (indexFilename) {
  591. NSString* indexPath = [filePath stringByAppendingPathComponent:indexFilename];
  592. if ([[NSFileManager defaultManager] fileExistsAtPath:indexPath isDirectory:&isDirectory] && !isDirectory) {
  593. return [GCDWebServerFileResponse responseWithFile:indexPath];
  594. }
  595. }
  596. response = [server _responseWithContentsOfDirectory:filePath];
  597. } else {
  598. if (allowRangeRequests) {
  599. response = [GCDWebServerFileResponse responseWithFile:filePath byteRange:request.byteRange];
  600. [response setValue:@"bytes" forAdditionalHeader:@"Accept-Ranges"];
  601. } else {
  602. response = [GCDWebServerFileResponse responseWithFile:filePath];
  603. }
  604. }
  605. }
  606. if (response) {
  607. response.cacheControlMaxAge = cacheAge;
  608. } else {
  609. response = [GCDWebServerResponse responseWithStatusCode:kGCDWebServerHTTPStatusCode_NotFound];
  610. }
  611. return response;
  612. }];
  613. } else {
  614. DNOT_REACHED();
  615. }
  616. }
  617. @end
  618. @implementation GCDWebServer (Logging)
  619. #ifndef __GCDWEBSERVER_LOGGING_HEADER__
  620. + (void)setLogLevel:(GCDWebServerLogLevel)level {
  621. GCDLogLevel = level;
  622. }
  623. #endif
  624. - (void)logVerbose:(NSString*)format, ... {
  625. va_list arguments;
  626. va_start(arguments, format);
  627. NSString* message = [[NSString alloc] initWithFormat:format arguments:arguments];
  628. va_end(arguments);
  629. LOG_VERBOSE(@"%@", message);
  630. ARC_RELEASE(message);
  631. }
  632. - (void)logInfo:(NSString*)format, ... {
  633. va_list arguments;
  634. va_start(arguments, format);
  635. NSString* message = [[NSString alloc] initWithFormat:format arguments:arguments];
  636. va_end(arguments);
  637. LOG_INFO(@"%@", message);
  638. ARC_RELEASE(message);
  639. }
  640. - (void)logWarning:(NSString*)format, ... {
  641. va_list arguments;
  642. va_start(arguments, format);
  643. NSString* message = [[NSString alloc] initWithFormat:format arguments:arguments];
  644. va_end(arguments);
  645. LOG_WARNING(@"%@", message);
  646. ARC_RELEASE(message);
  647. }
  648. - (void)logError:(NSString*)format, ... {
  649. va_list arguments;
  650. va_start(arguments, format);
  651. NSString* message = [[NSString alloc] initWithFormat:format arguments:arguments];
  652. va_end(arguments);
  653. LOG_ERROR(@"%@", message);
  654. ARC_RELEASE(message);
  655. }
  656. @end