Home | History | Annotate | Download | only in Misc 
      1 /*
      2  * Copyright (C) 2005 Apple Computer, Inc.  All rights reserved.
      3  *
      4  * Redistribution and use in source and binary forms, with or without
      5  * modification, are permitted provided that the following conditions
      6  * are met:
      7  *
      8  * 1.  Redistributions of source code must retain the above copyright
      9  *     notice, this list of conditions and the following disclaimer.
     10  * 2.  Redistributions in binary form must reproduce the above copyright
     11  *     notice, this list of conditions and the following disclaimer in the
     12  *     documentation and/or other materials provided with the distribution.
     13  * 3.  Neither the name of Apple Computer, Inc. ("Apple") nor the names of
     14  *     its contributors may be used to endorse or promote products derived
     15  *     from this software without specific prior written permission.
     16  *
     17  * THIS SOFTWARE IS PROVIDED BY APPLE AND ITS CONTRIBUTORS "AS IS" AND ANY
     18  * EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
     19  * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
     20  * DISCLAIMED. IN NO EVENT SHALL APPLE OR ITS CONTRIBUTORS BE LIABLE FOR ANY
     21  * DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
     22  * (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
     23  * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND
     24  * ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
     25  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
     26  * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
     27  */
     28 
     29 #import "WebNSURLRequestExtras.h"
     30 
     31 #import "WebNSURLExtras.h"
     32 
     33 #define WebContentType  (@"Content-Type")
     34 #define WebUserAgent    (@"User-Agent")
     35 #define WebReferrer     (@"Referer")
     36 
     37 @implementation NSURLRequest (WebNSURLRequestExtras)
     38 
     39 - (NSString *)_web_HTTPReferrer
     40 {
     41     return [self valueForHTTPHeaderField:WebReferrer];
     42 }
     43 
     44 - (NSString *)_web_HTTPContentType
     45 {
     46     return [self valueForHTTPHeaderField:WebContentType];
     47 }
     48 
     49 - (BOOL)_web_isConditionalRequest
     50 {
     51     if ([self valueForHTTPHeaderField:@"If-Match"] ||
     52         [self valueForHTTPHeaderField:@"If-Modified-Since"] ||
     53         [self valueForHTTPHeaderField:@"If-None-Match"] ||
     54         [self valueForHTTPHeaderField:@"If-Range"] ||
     55         [self valueForHTTPHeaderField:@"If-Unmodified-Since"])
     56         return YES;
     57     return NO;
     58 }
     59 
     60 @end
     61 
     62 @implementation NSMutableURLRequest (WebNSURLRequestExtras)
     63 
     64 - (void)_web_setHTTPContentType:(NSString *)contentType
     65 {
     66     [self setValue:contentType forHTTPHeaderField:WebContentType];
     67 }
     68 
     69 - (void)_web_setHTTPReferrer:(NSString *)referrer
     70 {
     71     // Do not set the referrer to a string that refers to a file URL.
     72     // That is a potential security hole.
     73     if ([referrer _webkit_isFileURL])
     74         return;
     75 
     76     // Don't allow empty Referer: headers; some servers refuse them
     77     if ([referrer length] == 0)
     78         return;
     79 
     80     [self setValue:referrer forHTTPHeaderField:WebReferrer];
     81 }
     82 
     83 - (void)_web_setHTTPUserAgent:(NSString *)userAgent
     84 {
     85     [self setValue:userAgent forHTTPHeaderField:WebUserAgent];
     86 }
     87 
     88 @end
     89