GCDWebServer.m 22 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639
  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. #if TARGET_OS_IPHONE
  27. #import <MobileCoreServices/MobileCoreServices.h>
  28. #else
  29. #import <SystemConfiguration/SystemConfiguration.h>
  30. #endif
  31. #import <netinet/in.h>
  32. #import <ifaddrs.h>
  33. #import <net/if.h>
  34. #import <netdb.h>
  35. #import "GCDWebServerPrivate.h"
  36. #if TARGET_OS_IPHONE && !TARGET_IPHONE_SIMULATOR
  37. #define kDefaultPort 80
  38. #else
  39. #define kDefaultPort 8080
  40. #endif
  41. #define kMaxPendingConnections 16
  42. @interface GCDWebServer () {
  43. @private
  44. NSMutableArray* _handlers;
  45. NSUInteger _port;
  46. dispatch_source_t _source;
  47. CFNetServiceRef _service;
  48. }
  49. @end
  50. @interface GCDWebServerHandler () {
  51. @private
  52. GCDWebServerMatchBlock _matchBlock;
  53. GCDWebServerProcessBlock _processBlock;
  54. }
  55. @end
  56. #if !TARGET_OS_IPHONE
  57. static BOOL _run;
  58. #endif
  59. #ifndef __GCDWEBSERVER_LOGGING_HEADER__
  60. void GCDLogMessage(long level, NSString* format, ...) {
  61. static const char* levelNames[] = {"DEBUG", "VERBOSE", "INFO", "WARNING", "ERROR", "EXCEPTION"};
  62. static long minLevel = -1;
  63. if (minLevel < 0) {
  64. const char* logLevel = getenv("logLevel");
  65. minLevel = logLevel ? atoi(logLevel) : 0;
  66. }
  67. if (level >= minLevel) {
  68. va_list arguments;
  69. va_start(arguments, format);
  70. NSString* message = [[NSString alloc] initWithFormat:format arguments:arguments];
  71. va_end(arguments);
  72. fprintf(stderr, "[%s] %s\n", levelNames[level], [message UTF8String]);
  73. ARC_RELEASE(message);
  74. }
  75. }
  76. #endif
  77. NSString* GCDWebServerGetMimeTypeForExtension(NSString* extension) {
  78. static NSDictionary* _overrides = nil;
  79. if (_overrides == nil) {
  80. _overrides = [[NSDictionary alloc] initWithObjectsAndKeys:
  81. @"text/css", @"css",
  82. nil];
  83. }
  84. NSString* mimeType = nil;
  85. extension = [extension lowercaseString];
  86. if (extension.length) {
  87. mimeType = [_overrides objectForKey:extension];
  88. if (mimeType == nil) {
  89. CFStringRef uti = UTTypeCreatePreferredIdentifierForTag(kUTTagClassFilenameExtension, (ARC_BRIDGE CFStringRef)extension, NULL);
  90. if (uti) {
  91. mimeType = ARC_BRIDGE_RELEASE(UTTypeCopyPreferredTagWithClass(uti, kUTTagClassMIMEType));
  92. CFRelease(uti);
  93. }
  94. }
  95. }
  96. return mimeType ? mimeType : kGCDWebServerDefaultMimeType;
  97. }
  98. NSString* GCDWebServerEscapeURLString(NSString* string) {
  99. return ARC_BRIDGE_RELEASE(CFURLCreateStringByAddingPercentEscapes(kCFAllocatorDefault, (CFStringRef)string, NULL, CFSTR(":@/?&=+"), kCFStringEncodingUTF8));
  100. }
  101. NSString* GCDWebServerUnescapeURLString(NSString* string) {
  102. return ARC_BRIDGE_RELEASE(CFURLCreateStringByReplacingPercentEscapesUsingEncoding(kCFAllocatorDefault, (CFStringRef)string, CFSTR(""), kCFStringEncodingUTF8));
  103. }
  104. NSDictionary* GCDWebServerParseURLEncodedForm(NSString* form) {
  105. NSMutableDictionary* parameters = [NSMutableDictionary dictionary];
  106. NSScanner* scanner = [[NSScanner alloc] initWithString:form];
  107. [scanner setCharactersToBeSkipped:nil];
  108. while (1) {
  109. NSString* key = nil;
  110. if (![scanner scanUpToString:@"=" intoString:&key] || [scanner isAtEnd]) {
  111. break;
  112. }
  113. [scanner setScanLocation:([scanner scanLocation] + 1)];
  114. NSString* value = nil;
  115. if (![scanner scanUpToString:@"&" intoString:&value]) {
  116. break;
  117. }
  118. key = [key stringByReplacingOccurrencesOfString:@"+" withString:@" "];
  119. value = [value stringByReplacingOccurrencesOfString:@"+" withString:@" "];
  120. if (key && value) {
  121. [parameters setObject:GCDWebServerUnescapeURLString(value) forKey:GCDWebServerUnescapeURLString(key)];
  122. } else {
  123. DNOT_REACHED();
  124. }
  125. if ([scanner isAtEnd]) {
  126. break;
  127. }
  128. [scanner setScanLocation:([scanner scanLocation] + 1)];
  129. }
  130. ARC_RELEASE(scanner);
  131. return parameters;
  132. }
  133. NSString* GCDWebServerGetPrimaryIPv4Address() {
  134. NSString* address = nil;
  135. #if TARGET_OS_IPHONE
  136. #if !TARGET_IPHONE_SIMULATOR
  137. const char* primaryInterface = "en0"; // WiFi interface on iOS
  138. #endif
  139. #else
  140. const char* primaryInterface = NULL;
  141. SCDynamicStoreRef store = SCDynamicStoreCreate(kCFAllocatorDefault, CFSTR("GCDWebServer"), NULL, NULL);
  142. if (store) {
  143. CFPropertyListRef info = SCDynamicStoreCopyValue(store, CFSTR("State:/Network/Global/IPv4"));
  144. if (info) {
  145. primaryInterface = [[(ARC_BRIDGE NSDictionary*)info objectForKey:@"PrimaryInterface"] UTF8String];
  146. CFRelease(info);
  147. }
  148. CFRelease(store);
  149. }
  150. if (primaryInterface == NULL) {
  151. primaryInterface = "lo0";
  152. }
  153. #endif
  154. struct ifaddrs* list;
  155. if (getifaddrs(&list) >= 0) {
  156. for (struct ifaddrs* ifap = list; ifap; ifap = ifap->ifa_next) {
  157. #if TARGET_IPHONE_SIMULATOR
  158. if (strcmp(ifap->ifa_name, "en0") && strcmp(ifap->ifa_name, "en1")) // Assume en0 is Ethernet and en1 is WiFi since there is no way to use SystemConfiguration framework in iOS Simulator
  159. #else
  160. if (strcmp(ifap->ifa_name, primaryInterface))
  161. #endif
  162. {
  163. continue;
  164. }
  165. if ((ifap->ifa_flags & IFF_UP) && (ifap->ifa_addr->sa_family == AF_INET)) {
  166. char buffer[NI_MAXHOST];
  167. if (getnameinfo(ifap->ifa_addr, ifap->ifa_addr->sa_len, buffer, sizeof(buffer), NULL, 0, NI_NUMERICHOST | NI_NOFQDN) >= 0) {
  168. address = [NSString stringWithUTF8String:buffer];
  169. }
  170. break;
  171. }
  172. }
  173. freeifaddrs(list);
  174. }
  175. return address;
  176. }
  177. #if !TARGET_OS_IPHONE
  178. static void _SignalHandler(int signal) {
  179. _run = NO;
  180. printf("\n");
  181. }
  182. #endif
  183. @implementation GCDWebServerHandler
  184. @synthesize matchBlock=_matchBlock, processBlock=_processBlock;
  185. - (id)initWithMatchBlock:(GCDWebServerMatchBlock)matchBlock processBlock:(GCDWebServerProcessBlock)processBlock {
  186. if ((self = [super init])) {
  187. _matchBlock = [matchBlock copy];
  188. _processBlock = [processBlock copy];
  189. }
  190. return self;
  191. }
  192. - (void)dealloc {
  193. ARC_RELEASE(_matchBlock);
  194. ARC_RELEASE(_processBlock);
  195. ARC_DEALLOC(super);
  196. }
  197. @end
  198. @implementation GCDWebServer
  199. @synthesize handlers=_handlers, port=_port;
  200. + (void)initialize {
  201. [GCDWebServerConnection class]; // Initialize class immediately to make sure it happens on the main thread
  202. }
  203. - (id)init {
  204. if ((self = [super init])) {
  205. _handlers = [[NSMutableArray alloc] init];
  206. }
  207. return self;
  208. }
  209. - (void)dealloc {
  210. if (_source) {
  211. [self stop];
  212. }
  213. ARC_RELEASE(_handlers);
  214. ARC_DEALLOC(super);
  215. }
  216. - (NSString*)bonjourName {
  217. CFStringRef name = _service ? CFNetServiceGetName(_service) : NULL;
  218. return name && CFStringGetLength(name) ? ARC_BRIDGE_RELEASE(CFStringCreateCopy(kCFAllocatorDefault, name)) : nil;
  219. }
  220. - (void)addHandlerWithMatchBlock:(GCDWebServerMatchBlock)matchBlock processBlock:(GCDWebServerProcessBlock)handlerBlock {
  221. DCHECK(_source == NULL);
  222. GCDWebServerHandler* handler = [[GCDWebServerHandler alloc] initWithMatchBlock:matchBlock processBlock:handlerBlock];
  223. [_handlers insertObject:handler atIndex:0];
  224. ARC_RELEASE(handler);
  225. }
  226. - (void)removeAllHandlers {
  227. DCHECK(_source == NULL);
  228. [_handlers removeAllObjects];
  229. }
  230. - (BOOL)start {
  231. return [self startWithPort:kDefaultPort bonjourName:@""];
  232. }
  233. static void _NetServiceClientCallBack(CFNetServiceRef service, CFStreamError* error, void* info) {
  234. @autoreleasepool {
  235. if (error->error) {
  236. LOG_ERROR(@"Bonjour error %i (domain %i)", (int)error->error, (int)error->domain);
  237. } else {
  238. GCDWebServer* server = (ARC_BRIDGE GCDWebServer*)info;
  239. LOG_VERBOSE(@"%@ now reachable at %@", [server class], server.bonjourServerURL);
  240. }
  241. }
  242. }
  243. - (BOOL)startWithPort:(NSUInteger)port bonjourName:(NSString*)name {
  244. DCHECK(_source == NULL);
  245. int listeningSocket = socket(PF_INET, SOCK_STREAM, IPPROTO_TCP);
  246. if (listeningSocket > 0) {
  247. int yes = 1;
  248. setsockopt(listeningSocket, SOL_SOCKET, SO_REUSEADDR, &yes, sizeof(yes));
  249. setsockopt(listeningSocket, SOL_SOCKET, SO_REUSEPORT, &yes, sizeof(yes));
  250. struct sockaddr_in addr4;
  251. bzero(&addr4, sizeof(addr4));
  252. addr4.sin_len = sizeof(addr4);
  253. addr4.sin_family = AF_INET;
  254. addr4.sin_port = htons(port);
  255. addr4.sin_addr.s_addr = htonl(INADDR_ANY);
  256. if (bind(listeningSocket, (void*)&addr4, sizeof(addr4)) == 0) {
  257. if (listen(listeningSocket, kMaxPendingConnections) == 0) {
  258. _source = dispatch_source_create(DISPATCH_SOURCE_TYPE_READ, listeningSocket, 0, kGCDWebServerGCDQueue);
  259. dispatch_source_set_cancel_handler(_source, ^{
  260. @autoreleasepool {
  261. int result = close(listeningSocket);
  262. if (result != 0) {
  263. LOG_ERROR(@"Failed closing socket (%i): %s", errno, strerror(errno));
  264. } else {
  265. LOG_DEBUG(@"Closed listening socket");
  266. }
  267. }
  268. });
  269. dispatch_source_set_event_handler(_source, ^{
  270. @autoreleasepool {
  271. struct sockaddr remoteSockAddr;
  272. socklen_t remoteAddrLen = sizeof(remoteSockAddr);
  273. int socket = accept(listeningSocket, &remoteSockAddr, &remoteAddrLen);
  274. if (socket > 0) {
  275. NSData* remoteAddress = [NSData dataWithBytes:&remoteSockAddr length:remoteAddrLen];
  276. struct sockaddr localSockAddr;
  277. socklen_t localAddrLen = sizeof(localSockAddr);
  278. NSData* localAddress = nil;
  279. if (getsockname(socket, &localSockAddr, &localAddrLen) == 0) {
  280. localAddress = [NSData dataWithBytes:&localSockAddr length:localAddrLen];
  281. } else {
  282. DNOT_REACHED();
  283. }
  284. int yes = 1;
  285. setsockopt(socket, SOL_SOCKET, SO_NOSIGPIPE, &yes, sizeof(yes)); // Make sure this socket cannot generate SIG_PIPE
  286. Class connectionClass = [[self class] connectionClass];
  287. GCDWebServerConnection* connection = [[connectionClass alloc] initWithServer:self localAddress:localAddress remoteAddress:remoteAddress socket:socket]; // Connection will automatically retain itself while opened
  288. #if __has_feature(objc_arc)
  289. [connection self]; // Prevent compiler from complaining about unused variable / useless statement
  290. #else
  291. [connection release];
  292. #endif
  293. } else {
  294. LOG_ERROR(@"Failed accepting socket (%i): %s", errno, strerror(errno));
  295. }
  296. }
  297. });
  298. if (port == 0) { // Determine the actual port we are listening on
  299. struct sockaddr addr;
  300. socklen_t addrlen = sizeof(addr);
  301. if (getsockname(listeningSocket, &addr, &addrlen) == 0) {
  302. struct sockaddr_in* sockaddr = (struct sockaddr_in*)&addr;
  303. _port = ntohs(sockaddr->sin_port);
  304. } else {
  305. LOG_ERROR(@"Failed retrieving socket address (%i): %s", errno, strerror(errno));
  306. }
  307. } else {
  308. _port = port;
  309. }
  310. if (name) {
  311. _service = CFNetServiceCreate(kCFAllocatorDefault, CFSTR("local."), CFSTR("_http._tcp"), (ARC_BRIDGE CFStringRef)name, (SInt32)_port);
  312. if (_service) {
  313. CFNetServiceClientContext context = {0, (ARC_BRIDGE void*)self, NULL, NULL, NULL};
  314. CFNetServiceSetClient(_service, _NetServiceClientCallBack, &context);
  315. CFNetServiceScheduleWithRunLoop(_service, CFRunLoopGetMain(), kCFRunLoopCommonModes);
  316. CFStreamError error = {0};
  317. CFNetServiceRegisterWithOptions(_service, 0, &error);
  318. } else {
  319. LOG_ERROR(@"Failed creating CFNetService");
  320. }
  321. }
  322. dispatch_resume(_source);
  323. LOG_VERBOSE(@"%@ started on port %i and reachable at %@", [self class], (int)_port, self.serverURL);
  324. } else {
  325. LOG_ERROR(@"Failed listening on socket (%i): %s", errno, strerror(errno));
  326. close(listeningSocket);
  327. }
  328. } else {
  329. LOG_ERROR(@"Failed binding socket (%i): %s", errno, strerror(errno));
  330. close(listeningSocket);
  331. }
  332. } else {
  333. LOG_ERROR(@"Failed creating socket (%i): %s", errno, strerror(errno));
  334. }
  335. return (_source ? YES : NO);
  336. }
  337. - (BOOL)isRunning {
  338. return (_source ? YES : NO);
  339. }
  340. - (void)stop {
  341. DCHECK(_source != NULL);
  342. if (_source) {
  343. if (_service) {
  344. CFNetServiceUnscheduleFromRunLoop(_service, CFRunLoopGetMain(), kCFRunLoopCommonModes);
  345. CFNetServiceSetClient(_service, NULL, NULL);
  346. CFRelease(_service);
  347. _service = NULL;
  348. }
  349. dispatch_source_cancel(_source); // This will close the socket
  350. ARC_DISPATCH_RELEASE(_source);
  351. _source = NULL;
  352. LOG_VERBOSE(@"%@ stopped", [self class]);
  353. }
  354. _port = 0;
  355. }
  356. @end
  357. @implementation GCDWebServer (Subclassing)
  358. + (Class)connectionClass {
  359. return [GCDWebServerConnection class];
  360. }
  361. + (NSString*)serverName {
  362. return NSStringFromClass(self);
  363. }
  364. @end
  365. @implementation GCDWebServer (Extensions)
  366. - (NSURL*)serverURL {
  367. if (_source) {
  368. NSString* ipAddress = GCDWebServerGetPrimaryIPv4Address();
  369. if (ipAddress) {
  370. if (_port != 80) {
  371. return [NSURL URLWithString:[NSString stringWithFormat:@"http://%@:%i/", ipAddress, (int)_port]];
  372. } else {
  373. return [NSURL URLWithString:[NSString stringWithFormat:@"http://%@/", ipAddress]];
  374. }
  375. }
  376. }
  377. return nil;
  378. }
  379. - (NSURL*)bonjourServerURL {
  380. if (_source && _service) {
  381. CFStringRef name = CFNetServiceGetName(_service);
  382. if (name && CFStringGetLength(name)) {
  383. if (_port != 80) {
  384. return [NSURL URLWithString:[NSString stringWithFormat:@"http://%@.local:%i/", name, (int)_port]];
  385. } else {
  386. return [NSURL URLWithString:[NSString stringWithFormat:@"http://%@.local/", name]];
  387. }
  388. }
  389. }
  390. return nil;
  391. }
  392. #if !TARGET_OS_IPHONE
  393. - (BOOL)runWithPort:(NSUInteger)port {
  394. BOOL success = NO;
  395. _run = YES;
  396. void (*handler)(int) = signal(SIGINT, _SignalHandler);
  397. if (handler != SIG_ERR) {
  398. if ([self startWithPort:port bonjourName:@""]) {
  399. while (_run) {
  400. CFRunLoopRunInMode(kCFRunLoopDefaultMode, 1.0, true);
  401. }
  402. [self stop];
  403. success = YES;
  404. }
  405. signal(SIGINT, handler);
  406. }
  407. return success;
  408. }
  409. #endif
  410. @end
  411. @implementation GCDWebServer (Handlers)
  412. - (void)addDefaultHandlerForMethod:(NSString*)method requestClass:(Class)aClass processBlock:(GCDWebServerProcessBlock)block {
  413. [self addHandlerWithMatchBlock:^GCDWebServerRequest *(NSString* requestMethod, NSURL* requestURL, NSDictionary* requestHeaders, NSString* urlPath, NSDictionary* urlQuery) {
  414. return ARC_AUTORELEASE([[aClass alloc] initWithMethod:requestMethod url:requestURL headers:requestHeaders path:urlPath query:urlQuery]);
  415. } processBlock:block];
  416. }
  417. - (void)addHandlerForMethod:(NSString*)method path:(NSString*)path requestClass:(Class)aClass processBlock:(GCDWebServerProcessBlock)block {
  418. if ([path hasPrefix:@"/"] && [aClass isSubclassOfClass:[GCDWebServerRequest class]]) {
  419. [self addHandlerWithMatchBlock:^GCDWebServerRequest *(NSString* requestMethod, NSURL* requestURL, NSDictionary* requestHeaders, NSString* urlPath, NSDictionary* urlQuery) {
  420. if (![requestMethod isEqualToString:method]) {
  421. return nil;
  422. }
  423. if ([urlPath caseInsensitiveCompare:path] != NSOrderedSame) {
  424. return nil;
  425. }
  426. return ARC_AUTORELEASE([[aClass alloc] initWithMethod:requestMethod url:requestURL headers:requestHeaders path:urlPath query:urlQuery]);
  427. } processBlock:block];
  428. } else {
  429. DNOT_REACHED();
  430. }
  431. }
  432. - (void)addHandlerForMethod:(NSString*)method pathRegex:(NSString*)regex requestClass:(Class)aClass processBlock:(GCDWebServerProcessBlock)block {
  433. NSRegularExpression* expression = [NSRegularExpression regularExpressionWithPattern:regex options:NSRegularExpressionCaseInsensitive error:NULL];
  434. if (expression && [aClass isSubclassOfClass:[GCDWebServerRequest class]]) {
  435. [self addHandlerWithMatchBlock:^GCDWebServerRequest *(NSString* requestMethod, NSURL* requestURL, NSDictionary* requestHeaders, NSString* urlPath, NSDictionary* urlQuery) {
  436. if (![requestMethod isEqualToString:method]) {
  437. return nil;
  438. }
  439. if ([expression firstMatchInString:urlPath options:0 range:NSMakeRange(0, urlPath.length)] == nil) {
  440. return nil;
  441. }
  442. return ARC_AUTORELEASE([[aClass alloc] initWithMethod:requestMethod url:requestURL headers:requestHeaders path:urlPath query:urlQuery]);
  443. } processBlock:block];
  444. } else {
  445. DNOT_REACHED();
  446. }
  447. }
  448. @end
  449. @implementation GCDWebServer (GETHandlers)
  450. - (void)addGETHandlerForPath:(NSString*)path staticData:(NSData*)staticData contentType:(NSString*)contentType cacheAge:(NSUInteger)cacheAge {
  451. GCDWebServerResponse* response = [GCDWebServerDataResponse responseWithData:staticData contentType:contentType];
  452. response.cacheControlMaxAge = cacheAge;
  453. [self addHandlerForMethod:@"GET" path:path requestClass:[GCDWebServerRequest class] processBlock:^GCDWebServerResponse *(GCDWebServerRequest* request) {
  454. return response;
  455. }];
  456. }
  457. - (void)addGETHandlerForPath:(NSString*)path filePath:(NSString*)filePath isAttachment:(BOOL)isAttachment cacheAge:(NSUInteger)cacheAge allowRangeRequests:(BOOL)allowRangeRequests {
  458. [self addHandlerForMethod:@"GET" path:path requestClass:[GCDWebServerRequest class] processBlock:^GCDWebServerResponse *(GCDWebServerRequest* request) {
  459. GCDWebServerResponse* response = nil;
  460. if (allowRangeRequests) {
  461. response = [GCDWebServerFileResponse responseWithFile:filePath byteRange:request.byteRange isAttachment:isAttachment];
  462. [response setValue:@"bytes" forAdditionalHeader:@"Accept-Ranges"];
  463. } else {
  464. response = [GCDWebServerFileResponse responseWithFile:filePath isAttachment:isAttachment];
  465. }
  466. response.cacheControlMaxAge = cacheAge;
  467. return response;
  468. }];
  469. }
  470. - (GCDWebServerResponse*)_responseWithContentsOfDirectory:(NSString*)path {
  471. NSDirectoryEnumerator* enumerator = [[NSFileManager defaultManager] enumeratorAtPath:path];
  472. if (enumerator == nil) {
  473. return nil;
  474. }
  475. NSMutableString* html = [NSMutableString string];
  476. [html appendString:@"<!DOCTYPE html>\n"];
  477. [html appendString:@"<html><head><meta charset=\"utf-8\"></head><body>\n"];
  478. [html appendString:@"<ul>\n"];
  479. for (NSString* file in enumerator) {
  480. if (![file hasPrefix:@"."]) {
  481. NSString* type = [[enumerator fileAttributes] objectForKey:NSFileType];
  482. NSString* escapedFile = [file stringByAddingPercentEscapesUsingEncoding:NSUTF8StringEncoding];
  483. DCHECK(escapedFile);
  484. if ([type isEqualToString:NSFileTypeRegular]) {
  485. [html appendFormat:@"<li><a href=\"%@\">%@</a></li>\n", escapedFile, file];
  486. } else if ([type isEqualToString:NSFileTypeDirectory]) {
  487. [html appendFormat:@"<li><a href=\"%@/\">%@/</a></li>\n", escapedFile, file];
  488. }
  489. }
  490. [enumerator skipDescendents];
  491. }
  492. [html appendString:@"</ul>\n"];
  493. [html appendString:@"</body></html>\n"];
  494. return [GCDWebServerDataResponse responseWithHTML:html];
  495. }
  496. - (void)addGETHandlerForBasePath:(NSString*)basePath directoryPath:(NSString*)directoryPath indexFilename:(NSString*)indexFilename cacheAge:(NSUInteger)cacheAge allowRangeRequests:(BOOL)allowRangeRequests {
  497. if ([basePath hasPrefix:@"/"] && [basePath hasSuffix:@"/"]) {
  498. #if __has_feature(objc_arc)
  499. __unsafe_unretained GCDWebServer* server = self;
  500. #else
  501. __block GCDWebServer* server = self;
  502. #endif
  503. [self addHandlerWithMatchBlock:^GCDWebServerRequest *(NSString* requestMethod, NSURL* requestURL, NSDictionary* requestHeaders, NSString* urlPath, NSDictionary* urlQuery) {
  504. if (![requestMethod isEqualToString:@"GET"]) {
  505. return nil;
  506. }
  507. if (![urlPath hasPrefix:basePath]) {
  508. return nil;
  509. }
  510. return ARC_AUTORELEASE([[GCDWebServerRequest alloc] initWithMethod:requestMethod url:requestURL headers:requestHeaders path:urlPath query:urlQuery]);
  511. } processBlock:^GCDWebServerResponse *(GCDWebServerRequest* request) {
  512. GCDWebServerResponse* response = nil;
  513. NSString* filePath = [directoryPath stringByAppendingPathComponent:[request.path substringFromIndex:basePath.length]];
  514. BOOL isDirectory;
  515. if ([[NSFileManager defaultManager] fileExistsAtPath:filePath isDirectory:&isDirectory]) {
  516. if (isDirectory) {
  517. if (indexFilename) {
  518. NSString* indexPath = [filePath stringByAppendingPathComponent:indexFilename];
  519. if ([[NSFileManager defaultManager] fileExistsAtPath:indexPath isDirectory:&isDirectory] && !isDirectory) {
  520. return [GCDWebServerFileResponse responseWithFile:indexPath];
  521. }
  522. }
  523. response = [server _responseWithContentsOfDirectory:filePath];
  524. } else {
  525. if (allowRangeRequests) {
  526. response = [GCDWebServerFileResponse responseWithFile:filePath byteRange:request.byteRange];
  527. [response setValue:@"bytes" forAdditionalHeader:@"Accept-Ranges"];
  528. } else {
  529. response = [GCDWebServerFileResponse responseWithFile:filePath];
  530. }
  531. }
  532. }
  533. if (response) {
  534. response.cacheControlMaxAge = cacheAge;
  535. } else {
  536. response = [GCDWebServerResponse responseWithStatusCode:404];
  537. }
  538. return response;
  539. }];
  540. } else {
  541. DNOT_REACHED();
  542. }
  543. }
  544. @end