auto import from //depot/cupcake/@135843

1 files changed, 88 insertions, 0 deletions

diff --git a/WebKit/mac/Misc/WebNSURLRequestExtras.m b/WebKit/mac/Misc/WebNSURLRequestExtras.m
new file mode 100644
index 0000000..770ff8f
--- /dev/null
+++ b/WebKit/mac/Misc/WebNSURLRequestExtras.m
@@ -0,0 +1,88 @@
+/*
+ * Copyright (C) 2005 Apple Computer, Inc.  All rights reserved.
+ *
+ * Redistribution and use in source and binary forms, with or without
+ * modification, are permitted provided that the following conditions
+ * are met:
+ *
+ * 1.  Redistributions of source code must retain the above copyright
+ *     notice, this list of conditions and the following disclaimer. 
+ * 2.  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. 
+ * 3.  Neither the name of Apple Computer, Inc. ("Apple") nor the names of
+ *     its contributors may be used to endorse or promote products derived
+ *     from this software without specific prior written permission. 
+ *
+ * THIS SOFTWARE IS PROVIDED BY APPLE AND ITS 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 APPLE OR ITS CONTRIBUTORS 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 "WebNSURLRequestExtras.h"
+
+#import "WebNSURLExtras.h"
+
+#define WebContentType  (@"Content-Type")
+#define WebUserAgent    (@"User-Agent")
+#define WebReferrer     (@"Referer")
+
+@implementation NSURLRequest (WebNSURLRequestExtras)
+
+- (NSString *)_web_HTTPReferrer
+{
+    return [self valueForHTTPHeaderField:WebReferrer];
+}
+
+- (NSString *)_web_HTTPContentType
+{
+    return [self valueForHTTPHeaderField:WebContentType];
+}
+
+- (BOOL)_web_isConditionalRequest
+{
+    if ([self valueForHTTPHeaderField:@"If-Match"] ||
+        [self valueForHTTPHeaderField:@"If-Modified-Since"] ||
+        [self valueForHTTPHeaderField:@"If-None-Match"] ||
+        [self valueForHTTPHeaderField:@"If-Range"] ||
+        [self valueForHTTPHeaderField:@"If-Unmodified-Since"])
+        return YES;
+    return NO;
+}
+
+@end
+
+@implementation NSMutableURLRequest (WebNSURLRequestExtras)
+
+- (void)_web_setHTTPContentType:(NSString *)contentType
+{
+    [self setValue:contentType forHTTPHeaderField:WebContentType];
+}
+
+- (void)_web_setHTTPReferrer:(NSString *)referrer
+{
+    // Do not set the referrer to a string that refers to a file URL.
+    // That is a potential security hole.
+    if ([referrer _webkit_isFileURL])
+        return;
+
+    // Don't allow empty Referer: headers; some servers refuse them
+    if ([referrer length] == 0)
+        return;
+
+    [self setValue:referrer forHTTPHeaderField:WebReferrer];
+}
+
+- (void)_web_setHTTPUserAgent:(NSString *)userAgent
+{
+    [self setValue:userAgent forHTTPHeaderField:WebUserAgent];
+}
+
+@end