重温Volley源码(三):添加Cookie或Https的能力

目录

一、Cookie设置与持久化

二、设置Https

参考资料

一、Cookie设置

方案一:通过Volley自定义Request对象进行设置

Request是Volley的一个抽象请求类,我们可以自定义实现里面的抽象方法来完成对Cookie的获取和保存,如:

public class CookieStringRequest extends Request<String> {

    private static final String SET_COOKIE_KEY = "Set-Cookie";
    private static final String COOKIE_KEY = "Cookie";

    private Listener<String> mListener;

    public CookieStringRequest(int method, String url, Listener<String> listener,
                               ErrorListener errorListener) {
        super(method, url, errorListener);
        mListener = listener;
    }

    public CookieStringRequest(String url, Listener<String> listener, ErrorListener errorListener) {
        this(Method.GET, url, listener, errorListener);
    }

    @Override
    protected void onFinish() {
        super.onFinish();
        mListener = null;
    }

    @Override
    protected void deliverResponse(String response) {
        if (mListener != null) {
            mListener.onResponse(response);
        }
    }

    @Override
    protected Response<String> parseNetworkResponse(NetworkResponse response) {
        String parsed;
        try {

            Map<String, String> headers=response.headers;

            if(headers!=null&&headers.containsKey(SET_COOKIE_KEY)){
                String cookie=headers.get(SET_COOKIE_KEY);

                if(!TextUtils.isEmpty(cookie)){
                    // TODO: 将cookie存到本地,如Sharepreference
                }
            }

            parsed = new String(response.data, HttpHeaderParser.parseCharset(response.headers));

        } catch (UnsupportedEncodingException e) {
            parsed = new String(response.data);
        }
        return Response.success(parsed, HttpHeaderParser.parseCacheHeaders(response));
    }

    @Override
    public Map<String, String> getHeaders() throws AuthFailureError {

        Map<String, String> headers=super.getHeaders();

        if(headers==null||headers.equals(Collections.emptyMap())){
            headers=new HashMap<>();
        }

        // TODO: 从本地获取到cookie,并把cookie添加到header中
        String value=getCookie();
        headers.put(COOKIE_KEY,value);

        return headers;
        //return super.getHeaders();
    }

    /**
     * 获取cookie
     * @return
     */
    private String getCookie() {
        return null;
    }
}

在parseNetworkResponse中获取Cookie并保存到本地,getHeaders方法从本地获取到Cookie后,请求时会添加到头部。

这种方案Volley默认会返回一个Cookie,如果返回多个Cookie的情况可能我们需要修改下源码了,先来看看原来Volley是怎样去拿header的,在HurlStack类的performRequest方法中可以看到:


public class HurlStack implements HttpStack {
    ......
    @Override
    public HttpResponse performRequest(Request<?> request, Map<String, String> additionalHeaders)
            throws IOException, AuthFailureError {
        ......

        for (Entry<String, List<String>> header : connection.getHeaderFields().entrySet()) {
            if (header.getKey() != null) {
                //get(0)即默认只获取第一条数据
                Header h = new BasicHeader(header.getKey(), header.getValue().get(0));
                response.addHeader(h);
            }
        }
        return response;
    }
}

找到了原因所在,只要对症下药,那么在遍历的时候把所有的key-value一并返回即可,修改后如下:


public class HurlStack implements HttpStack {
    ......
    @Override
    public HttpResponse performRequest(Request<?> request, Map<String, String> additionalHeaders)
            throws IOException, AuthFailureError {
        ......

        for (Entry<String, List<String>> header : connection.getHeaderFields().entrySet()) {
            if (header.getKey() != null) {
                StringBuilder builder = new StringBuilder();
                //获取一个key中的所有value
                for(int i=0;i<header.getValue().size();i++){
                    builder.append(header.getValue().get(i));
                }
                Header h = new BasicHeader(header.getKey(),builder.toString());
                response.addHeader(h);
            }
        }
        return response;
    }
}

修改完当我们使用StringRequest或其他Request时,只要重写parseNetworkResponse方法获取Cookie信息即可,同时还需要注意下此时返回的数据格式,是所有Cookie连在一起的字符串,使用的时候按规则解析一下即可。

方案二:使用Java提供的CookieManager和自定义的CookieStore进行设置

首先实现一个自定义的AppCookieStore类,它实现自CookieStore接口,在这个接口里进行我们App的本地持久化和获取操作,供Java自定调用,具体的实现网上有很多示例,下面给出伪代码:


public class AppCookieStore implements CookieStore{
    @Override
    public void add(URI uri, HttpCookie httpCookie) {

    }

    @Override
    public List<HttpCookie> get(URI uri) {
        return null;
    }

    @Override
    public List<HttpCookie> getCookies() {
        return null;
    }

    @Override
    public List<URI> getURIs() {
        return null;
    }

    @Override
    public boolean remove(URI uri, HttpCookie httpCookie) {
        return false;
    }

    @Override
    public boolean removeAll() {
        return false;
    }
}


接着在我们的网络请求之前调用以下代码,就可以将自定义的CookieStore设置到CookieManager中:


CookieManager cookieManager=new java.net.CookieManager(new AppCookieStore(), CookiePolicy.ACCEPT_ALL);
        CookieHandler.setDefault(cookieManager);

CookieManager会在之后的Http请求中自动帮我们处理response中的Cookie,当然我们也可以把它写在Application的onCreate中。

在请求的时候和方案一类似,在自定义Request对象的getHeaders方法中传入我们需要的Coockies即可


    @Override
    public Map<String, String> getHeaders() throws AuthFailureError {

        Map<String, String> headers = super.getHeaders();

        if (headers == null || headers.equals(Collections.emptyMap())) {
            headers = new HashMap<>();
        }

        // 获取之前创建的CookieManager对象,调用getCookieStore方法拿到HttpCookie列表
        CookieManager cookieManager = getCookieManager();
        List<HttpCookie> httpCookies = cookieManager.getCookieStore().getCookies();
        StringBuilder cookieBuilder = new StringBuilder();

        String divider = "";
        for (HttpCookie cookie : httpCookies) {
            cookieBuilder.append(divider);
            divider = ";";
            cookieBuilder.append(cookie.getName());
            cookieBuilder.append("=");
            cookieBuilder.append(cookie.getValue());
        }

        String cookieString = cookieBuilder.toString();

        headers.put(COOKIE_KEY, cookieString);

        return headers;
        //return super.getHeaders();
    }

二、Https设置

Https简单的理解就是http+ssl,ssl即安全套接层,使用Https加密前我们需要准备ssl证书文件(crt、cet、pem格式等),其实Volley是可以支持Https的,但是源码中并未启用,我们可以看一下HurlStack这个类:


public class HurlStack implements HttpStack {
    ......
    public HurlStack() {
        this(null);
    }

    public HurlStack(UrlRewriter urlRewriter) {
        this(urlRewriter, null);
    }

    public HurlStack(UrlRewriter urlRewriter, SSLSocketFactory sslSocketFactory) {
        mUrlRewriter = urlRewriter;
        mSslSocketFactory = sslSocketFactory;
    }
    ......
}

HurlStack(UrlRewriter urlRewriter, SSLSocketFactory sslSocketFactory)这个构造方法仅为HurlStack(UrlRewriter urlRewriter)所调用,并传入了一个null的SSLSocketFactory,所以,知道了问题的根源,我们只要创建HurlStack对象的时候调用第三个构造函数,并传入相应的SSLSocketFactory对象即可。

这里给出一个网上写好的示例,替换掉原有源码里的Volley类:


public class Volley {

    /**
     * Default on-disk cache directory.
     */
    private static final String DEFAULT_CACHE_DIR = "volley";
    private static BasicNetwork network;
    private static RequestQueue queue;

    private Context mContext;

    /**
     * Creates a default instance of the worker pool and calls
     * {@link RequestQueue#start()} on it.
     *
     * @param context A {@link Context} to use for creating the cache dir.
     * @param stack   An {@link HttpStack} to use for the network, or null for
     *                default.
     * @return A started {@link RequestQueue} instance.
     */
    public static RequestQueue newRequestQueue(Context context,
                                               HttpStack stack, boolean selfSignedCertificate, int rawId) {
        File cacheDir = new File(context.getCacheDir(), DEFAULT_CACHE_DIR);

        String userAgent = "volley/0";
        try {
            String packageName = context.getPackageName();
            PackageInfo info = context.getPackageManager().getPackageInfo(
                    packageName, 0);
            userAgent = packageName + "/" + info.versionCode;
        } catch (NameNotFoundException e) {
        }

        if (stack == null) {
            if (Build.VERSION.SDK_INT >= 9) {
                if (selfSignedCertificate) {
                    stack = new HurlStack(null, buildSSLSocketFactory(context,
                            rawId));
                } else {
                    stack = new HurlStack();
                }
            } else {
                // Prior to Gingerbread, HttpUrlConnection was unreliable.
                // See:
                // http://android-developers.blogspot.com/2011/09/androids-http-clients.html
                if (selfSignedCertificate)
                    stack = new HttpClientStack(getHttpClient(context, rawId));
                else {
                    stack = new HttpClientStack(
                            AndroidHttpClient.newInstance(userAgent));
                }
            }
        }

        if (network == null) {
            network = new BasicNetwork(stack);
        }
        if (queue == null) {
            queue = new RequestQueue(new DiskBasedCache(cacheDir),network);
        }
        queue.start();

        return queue;
    }

    /**
     * Creates a default instance of the worker pool and calls
     * {@link RequestQueue#start()} on it.
     *
     * @param context A {@link Context} to use for creating the cache dir.
     * @return A started {@link RequestQueue} instance.
     */
    public static RequestQueue newRequestQueue(Context context) {
        // 如果你目前还没有证书,那么先用下面的这行代码,http可以照常使用.
        //       return newRequestQueue(context, null, false, 0);
        // 此处R.raw.certificateName 表示你的证书文件,替换为自己证书文件名字就好
        return newRequestQueue(context, null, true, R.raw.certificateName);
    }

    private static SSLSocketFactory buildSSLSocketFactory(Context context,
                                                          int certRawResId) {
        KeyStore keyStore = null;
        try {
            keyStore = buildKeyStore(context, certRawResId);
        } catch (KeyStoreException e) {
            e.printStackTrace();
        } catch (CertificateException e) {
            e.printStackTrace();
        } catch (NoSuchAlgorithmException e) {
            e.printStackTrace();
        } catch (IOException e) {
            e.printStackTrace();
        }

        String tmfAlgorithm = TrustManagerFactory.getDefaultAlgorithm();
        TrustManagerFactory tmf = null;
        try {
            tmf = TrustManagerFactory.getInstance(tmfAlgorithm);
            tmf.init(keyStore);

        } catch (NoSuchAlgorithmException e) {
            e.printStackTrace();
        } catch (KeyStoreException e) {
            e.printStackTrace();
        }

        SSLContext sslContext = null;
        try {
            sslContext = SSLContext.getInstance("TLS");
        } catch (NoSuchAlgorithmException e) {
            e.printStackTrace();
        }
        try {
            sslContext.init(null, tmf.getTrustManagers(), null);
        } catch (KeyManagementException e) {
            e.printStackTrace();
        }

        return sslContext.getSocketFactory();

    }

    private static HttpClient getHttpClient(Context context, int certRawResId) {
        KeyStore keyStore = null;
        try {
            keyStore = buildKeyStore(context, certRawResId);
        } catch (KeyStoreException e) {
            e.printStackTrace();
        } catch (CertificateException e) {
            e.printStackTrace();
        } catch (NoSuchAlgorithmException e) {
            e.printStackTrace();
        } catch (IOException e) {
            e.printStackTrace();
        }
        if (keyStore != null) {
        }
        org.apache.http.conn.ssl.SSLSocketFactory sslSocketFactory = null;
        try {
            sslSocketFactory = new org.apache.http.conn.ssl.SSLSocketFactory(
                    keyStore);
        } catch (NoSuchAlgorithmException e) {
            e.printStackTrace();
        } catch (KeyManagementException e) {
            e.printStackTrace();
        } catch (KeyStoreException e) {
            e.printStackTrace();
        } catch (UnrecoverableKeyException e) {
            e.printStackTrace();
        }

        HttpParams params = new BasicHttpParams();

        SchemeRegistry schemeRegistry = new SchemeRegistry();
        schemeRegistry.register(new Scheme("http", PlainSocketFactory
                .getSocketFactory(), 80));
        schemeRegistry.register(new Scheme("https", sslSocketFactory, 443));

        ThreadSafeClientConnManager cm = new ThreadSafeClientConnManager(
                params, schemeRegistry);

        return new DefaultHttpClient(cm, params);
    }

    private static KeyStore buildKeyStore(Context context, int certRawResId)
            throws KeyStoreException, CertificateException,
            NoSuchAlgorithmException, IOException {
        String keyStoreType = KeyStore.getDefaultType();
        KeyStore keyStore = KeyStore.getInstance(keyStoreType);
        keyStore.load(null, null);

        Certificate cert = readCert(context, certRawResId);
        keyStore.setCertificateEntry("ca", cert);

        return keyStore;
    }

    private static Certificate readCert(Context context, int certResourceID) {
        InputStream inputStream = context.getResources().openRawResource(
                certResourceID);
        Certificate ca = null;

        CertificateFactory cf = null;
        try {
            cf = CertificateFactory.getInstance("X.509");
            ca = cf.generateCertificate(inputStream);

        } catch (CertificateException e) {
            e.printStackTrace();
        }
        return ca;
    }
}

在项目新建raw文件夹,然后将SSL证书拷贝放到该目录,修改Volley的newRequestQueue方法即可。另外,由于在Android API 23中已经废弃了HttpClient,如果你的项目compileSdkVersion>=23,使用上述Volley源码时需降级编译。

参考资料

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

推荐阅读更多精彩内容