Method processHttpMessage

src/main/java/burp/CookieManager.java:95–123

(int toolFlag, boolean messageIsRequest, IHttpRequestResponse messageInfo)

    }

    @Override
    public void processHttpMessage(int toolFlag, boolean messageIsRequest, IHttpRequestResponse messageInfo) {
        if (messageIsRequest) {
            // 更新 Cookie, 目前只实现了增加, 没有做删除的逻辑

            IRequestInfo requestInfo = BurpExtender.getHelpers().analyzeRequest(messageInfo);
            String origin = this.urlToOrigin(requestInfo.getUrl());
            List<String> headers = requestInfo.getHeaders();

            // 找到所有的 Cookie Header
            List<String> cookies = headers.subList(1, headers.size()).stream().filter(
                    header -> "cookie".equalsIgnoreCase(header.split(":", 2)[0])
            ).collect(Collectors.toList());

            for (String cookieHeader : cookies) {
                String[] temp = cookieHeader.split(":", 2);
                if (temp.length == 2) {
                    HashMap<String, String> cookie = this.parseCookie(temp[1]);

                    // 合并 Cookie
                    if (this.cookies.get(origin) == null) {
                        this.cookies.put(origin, cookie);
                    } else {
                        this.cookies.get(origin).putAll(cookie);
                    }
                }
            }
        }
    }

    public String getCookieHeader(URL url) {
        HashMap<String, String> cookie = this.cookies.get(this.urlToOrigin(url));