Method ProcessProxyHeader

phxsqlproxy/proxy_protocol_handler.cpp:144–183

}

int ProxyProtocolHandler::ProcessProxyHeader(int fd) {
    int ret = GetAddrFromFD(fd);
    if (ret < 0) {
        return ret;
    }

    ret = IsProxyHeaderNeed();
    if (ret <= 0) {
        return ret;
    }

    union {
        char     v1[108];
        ProxyHdr v2;
    } hdr;

    int hdr_size = RoutinePeekWithTimeout(fd, (char *) &hdr, sizeof(hdr), config_->ProxyProtocolTimeoutMs());
    if (hdr_size < 0) {
        LOG_ERR("read proxy header from fd [%d] failed, ret %d", fd, hdr_size);
        return -__LINE__;
    }

    if (hdr_size >= 16 && memcmp(&hdr.v2, PP2_SIGNATURE, 12) == 0 && (hdr.v2.ver_cmd & 0xF0) == PP2_VERSION) {
        hdr_size = ParseProxyHeaderV2(hdr.v2, hdr_size);
    } else if (hdr_size >= 8 && memcmp(hdr.v1, "PROXY ", 6) == 0) {
        hdr_size = ParseProxyHeaderV1(hdr.v1, hdr_size);
    } else {
        return -__LINE__;
    }

    if (hdr_size > 0) {
        ret = read(fd, &hdr, hdr_size);
        if (ret != hdr_size) {
            LOG_ERR("flush fd [%d] failed, ret %d hdr size %d", fd, ret, hdr_size);
            return -__LINE__;
        }
    }

    return hdr_size;
}

int ProxyProtocolHandler::ParseProxyHeaderV1(const char * hdr, int read_size) {
    const char * end = (const char *) memchr(hdr, '\r', read_size - 1);