AFNetWorking框架请求参数GBK编码解码不支持问题

这两天遇到一个问题是,当我用AFNetWorking进行POST请求,并且POST参数需要GBK编码时,请求会有问题。

跟踪代码后发现问题出在 AFPercentEscapedStringFromString 函数里:

    NSString * AFPercentEscapedStringFromString(NSString *string) {
        static NSString * const kAFCharactersGeneralDelimitersToEncode = @":#[]@"; // does not include "?" or "/" due to RFC 3986 - Section 3.4
        static NSString * const kAFCharactersSubDelimitersToEncode = @"!$&'()*+,;=";

        NSMutableCharacterSet * allowedCharacterSet = [[NSCharacterSet URLQueryAllowedCharacterSet] mutableCopy];
        [allowedCharacterSet removeCharactersInString:[kAFCharactersGeneralDelimitersToEncode stringByAppendingString:kAFCharactersSubDelimitersToEncode]];

        // FIXME: https://github.com/AFNetworking/AFNetworking/pull/3028
        // return [string stringByAddingPercentEncodingWithAllowedCharacters:allowedCharacterSet];

        static NSUInteger const batchSize = 50;

        NSUInteger index = 0;
        NSMutableString *escaped = @"".mutableCopy;

        while (index < string.length) {
            NSUInteger length = MIN(string.length - index, batchSize);
            NSRange range = NSMakeRange(index, length);

            // To avoid breaking up character sequences such as 👴🏻👮🏽
            range = [string rangeOfComposedCharacterSequencesForRange:range];

            NSString *substring = [string substringWithRange:range];
            NSString *encoded = [substring stringByAddingPercentEncodingWithAllowedCharacters:allowedCharacterSet];
            [escaped appendString:encoded];

            index += range.length;
        }

        return escaped;
    }

这个函数其实是做URL编码的,但是用的 stringByAddingPercentEncodingWithAllowedCharacters:allowedCharacterSet 编出来都是UTF8 格式的,有心的你会发现AF不是可以自定义请求编码格式的么,就是AFHTTPRequestSerializerstringEncoding 属性,这里稍微理解下AF请求的参数转变流程为如下:

  1. 你发送请求的入参 paramDic 字典
  2. 你的入参paramDic会被解析为 key1=value1&key2=value2 格式的字符串
  3. 被解析好的 key1=value1&key2=value2 字符串被转换为data格式传入请求的 httpbody 里

在这个过程中:

  • 1到2的解析过程用到的是上面说的有问题的方法 AFPercentEscapedStringFromString,所以任何请求这部都用的UTF8编码
  • 而2到3的时候的编码类型才用到AFHTTPRequestSerializerstringEncoding 属性

所以很明显,如果你需要GBK编码格式编码请求参数,并且也设定了AFHTTPRequestSerializerstringEncoding为GBK ,在这个过程中你的参数从字典变为拼接字符串的时候,会先进行UTF编码,然后字符串又转成了GBK格式的data类型,很显然不对。

所以最终导致,如果你的POSTBody需要GBK或者别的编码格式编码,当用AFNetworking时,会编码错误,导致你的服务器解码是也解不对,导致请求出问题。

这个问题再AFNetworking的Github官方主页的issue下也有记录,目前还是open状态,即未解决。


那怎么解决这个问题呢?

这有两种解决方法,一种不侵入AF源码,一种是改AF源码。


1.通过AFHTTPRequestSerializer 的 setQueryStringSerializationWithBlock: 方法自定义,那么整个请求流程代码就类似如下:
NSStringEncoding gbkEncoding = CFStringConvertEncodingToNSStringEncoding(kCFStringEncodingGB_18030_2000);
AFHTTPSessionManager *manager = [[AFHTTPSessionManager alloc] init];
[manager.requestSerializer setQueryStringSerializationWithBlock:^NSString * _Nonnull(NSURLRequest * _Nonnull request, id  _Nonnull parameters, NSError * _Nullable __autoreleasing * _Nullable error) {
    NSDictionary *dic = parameters;
    return [NSString stringWithFormat:@"%@=%@",[dic.allKeys[0] stringByAddingPercentEscapesUsingEncoding:gbkEncoding],[dic.allValues[0] stringByAddingPercentEscapesUsingEncoding:gbkEncoding]];

}];
manager.requestSerializer.stringEncoding = gbkEncoding;
[manager POST:searchHomeUrlStr parameters:@{@"key":@"GBK编码值"} progress:nil success:^(NSURLSessionDataTask * _Nonnull task, id  _Nullable responseObject) {

} failure:^(NSURLSessionDataTask * _Nullable task, NSError * _Nonnull error) {

}];

上面我是举例写,代码不很严瑾,你能明白我主要要表达什么意思就行。

每个需要 非UTF8 的请求都需要定义下 setQueryStringSerializationWithBlock: 方法,在这个方法中你需要把你的参数字典解析成 key1=value1&key2=value2 格式,并且用你需要的编码方式编码。

2.直接改源码

改源码要改3处地方,都在AFHTTPRequestSerializer里,第1处和第2处新增两个方法需要能接收编码方式,第3处改动,需要 requestBySerializingRequest: withParameters: error:方法里判断编码方式是不是UTF8,如果不是,走你新增的那两个方法来达到用你自己编码方式编码。

新增1:旧AFQueryStringFromParametersWithEncoding下新增一个下面这个函数,多一个编码方式入参

NSString * AFQueryStringFromParametersWithEncoding(NSDictionary *parameters, NSStringEncoding encode) {
    NSMutableArray *mutablePairs = [NSMutableArray array];
    for (AFQueryStringPair *pair in AFQueryStringPairsFromDictionary(parameters)) {
        [mutablePairs addObject:[pair URLEncodedStringValue:encode]];
    }

    return [mutablePairs componentsJoinedByString:@"&"];
}

新增2: 旧URLEncodedStringValue 方法下新增一个下面这个方法,也是多一个编码方式入参

- (NSString *)URLEncodedStringValue:(NSStringEncoding)encode {
        if (!self.value || [self.value isEqual:[NSNull null]]) {
            return AFPercentEscapedStringFromString([self.field description]);
        } else {
            if (encode == NSUTF8StringEncoding) {
                return [NSString stringWithFormat:@"%@=%@", AFPercentEscapedStringFromString([self.field description]), AFPercentEscapedStringFromString([self.value description])];
            } else {
                
                return [NSString stringWithFormat:@"%@=%@", [self.field stringByAddingPercentEscapesUsingEncoding:encode], [self.value stringByAddingPercentEscapesUsingEncoding:encode]];
            }
        }
    }

改动3,下面这个方法里,改动地方看下面代码块下面的图:

- (NSURLRequest *)requestBySerializingRequest:(NSURLRequest *)request
                                   withParameters:(id)parameters
                                            error:(NSError *__autoreleasing *)error
    {
        NSParameterAssert(request);

        NSMutableURLRequest *mutableRequest = [request mutableCopy];

        [self.HTTPRequestHeaders enumerateKeysAndObjectsUsingBlock:^(id field, id value, BOOL * __unused stop) {
            if (![request valueForHTTPHeaderField:field]) {
                [mutableRequest setValue:value forHTTPHeaderField:field];
            }
        }];

        NSString *query = nil;
        if (parameters) {
            if (self.queryStringSerialization) {
                NSError *serializationError;
                query = self.queryStringSerialization(request, parameters, &serializationError);

                if (serializationError) {
                    if (error) {
                        *error = serializationError;
                    }

                    return nil;
                }
            } else {
                switch (self.queryStringSerializationStyle) {
                    case AFHTTPRequestQueryStringDefaultStyle:
                        if (self.stringEncoding == NSUTF8StringEncoding) {
                            query = AFQueryStringFromParameters(parameters);
                        } else {
                            query = AFQueryStringFromParametersWithEncoding(parameters, self.stringEncoding);
                        }
                        
                        break;
                }
            }
        }

        if ([self.HTTPMethodsEncodingParametersInURI containsObject:[[request HTTPMethod] uppercaseString]]) {
            if (query && query.length > 0) {
                mutableRequest.URL = [NSURL URLWithString:[[mutableRequest.URL absoluteString] stringByAppendingFormat:mutableRequest.URL.query ? @"&%@" : @"?%@", query]];
            }
        } else {
            // #2864: an empty string is a valid x-www-form-urlencoded payload
            if (!query) {
                query = @"";
            }
            if (![mutableRequest valueForHTTPHeaderField:@"Content-Type"]) {
                [mutableRequest setValue:@"application/x-www-form-urlencoded" forHTTPHeaderField:@"Content-Type"];
            }
            [mutableRequest setHTTPBody:[query dataUsingEncoding:self.stringEncoding]];
        }

        return mutableRequest;
    }

改动地方:这里做一个是否UTF8的判断,不是UTF8类型走你新增1里的那个方法,把编码方式作为入参传入即可:


image.png

做个记录,如果有更好的改进方法,请留言。谢谢。

主要还是要等官方修这个bug啊~

©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 203,456评论 5 477
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 85,370评论 2 381
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 150,337评论 0 337
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 54,583评论 1 273
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 63,596评论 5 365
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 48,572评论 1 281
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 37,936评论 3 395
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 36,595评论 0 258
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 40,850评论 1 297
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 35,601评论 2 321
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 37,685评论 1 329
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 33,371评论 4 318
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 38,951评论 3 307
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 29,934评论 0 19
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 31,167评论 1 259
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 43,636评论 2 349
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 42,411评论 2 342

推荐阅读更多精彩内容