Reject non-ASCII text header values too.

pull/3216/head
Jorge Canizales 9 years ago
parent 2529735d3d
commit bbba491b7c
  1. 13
      src/objective-c/GRPCClient/private/GRPCRequestHeaders.m

@ -39,13 +39,13 @@
#import "NSDictionary+GRPC.h" #import "NSDictionary+GRPC.h"
// Used by the setter. // Used by the setter.
static void CheckKeyIsValid(NSString* key) { static void CheckIsNonNilASCII(NSString *name, NSString* value) {
if (!key) { if (!value) {
[NSException raise:NSInvalidArgumentException format:@"Key cannot be nil"]; [NSException raise:NSInvalidArgumentException format:@"%@ cannot be nil", name];
} }
if (![key canBeConvertedToEncoding:NSASCIIStringEncoding]) { if (![value canBeConvertedToEncoding:NSASCIIStringEncoding]) {
[NSException raise:NSInvalidArgumentException [NSException raise:NSInvalidArgumentException
format:@"Key %@ contains non-ASCII characters", key]; format:@"%@ %@ contains non-ASCII characters", name, value];
} }
} }
@ -63,6 +63,7 @@ static void CheckKeyValuePairIsValid(NSString *key, id value) {
format:@"Expected NSString value for header %@ not ending in \"-bin\", " format:@"Expected NSString value for header %@ not ending in \"-bin\", "
@"instead got %@", key, value]; @"instead got %@", key, value];
} }
CheckIsNonNilASCII(@"Text header value", (NSString *)value);
} }
} }
@ -92,7 +93,7 @@ static void CheckKeyValuePairIsValid(NSString *key, id value) {
- (void)setObject:(id)obj forKeyedSubscript:(NSString *)key { - (void)setObject:(id)obj forKeyedSubscript:(NSString *)key {
[self checkCallIsNotStarted]; [self checkCallIsNotStarted];
CheckKeyIsValid(key); CheckIsNonNilASCII(@"Header name", key);
key = key.lowercaseString; key = key.lowercaseString;
CheckKeyValuePairIsValid(key, obj); CheckKeyValuePairIsValid(key, obj);
_delegate[key] = obj; _delegate[key] = obj;

Loading…
Cancel
Save