iOS 根据二进制数据流来判断文件类型

原理:二进制的前2位的ascii码用来进行判断文件类型

网上搜索得来的:
    // 255216 jpg;
    // 7173 gif;
    // 6677 bmp,
    // 13780 png;
    // 6787 swf
    // 7790 exe dll,
    // 8297 rar
    // 8075 zip
    // 55122 7z
    // 6063 xml
    // 6033 html
    // 239187 aspx
    // 117115 cs
    // 119105 js
    // 102100 txt
    // 255254 sql

自己尝试校验出来的:
    // 3780 PDF


我的使用场景:

在WKWebView 加载的网页中,点击下载按钮(会有一个下载链接,但链接无数据类型后缀),如果是PDF类型的文件做特殊处理。

代码:

- (void)webView:(WKWebView *)webView decidePolicyForNavigationAction:(WKNavigationAction *)navigationAction decisionHandler:(void (^)(WKNavigationActionPolicy))decisionHandler
{
    // 无数据类型后缀
    // FIXME:通过二进制数据流,进行数据类型判断
    NSData *data = [NSData dataWithContentsOfURL:navigationAction.request.URL];
    if (data.length < 2) {
        NSLog(@"NOT FILE");
    }
    else
    {
        int char1 = 0 ,char2 = 0 ; //必须这样初始化
        [data getBytes:&char1 range:NSMakeRange(0, 1)];
        [data getBytes:&char2 range:NSMakeRange(1, 1)];
        NSString *numStr = [NSString stringWithFormat:@"%i%i",char1,char2];
        
        NSLog(@"%@",numStr);
        if ([numStr intValue] == 3780) {
            // 自己的特殊处理(我这里是将 PDF数据加载至WebView中)
            [self.webView loadData:data MIMEType:@"application/pdf" characterEncodingName:@"GBK" baseURL:[NSURL URLWithString:@""]];
        }
    }
    
    // 有数据类型后缀
//     NSString *_webUrlStr = navigationAction.request.URL.absoluteString;
//     NSString *lastName =[[_webUrlStr lastPathComponent] lowercaseString];
//
//     if ([lastName containsString:@".pdf"]) {
//         NSData *data = [NSData dataWithContentsOfURL:navigationAction.request.URL];
//         [self.webView loadData:data MIMEType:@"application/pdf" characterEncodingName:@"GBK" baseURL:[NSURL URLWithString:@""]];
//     }
    
     decisionHandler(WKNavigationActionPolicyAllow);
}

你可能感兴趣的:(iOS 根据二进制数据流来判断文件类型)