Method ParseProxyHeaderV2

phxsqlproxy/proxy_protocol_handler.cpp:227–262

}

int ProxyProtocolHandler::ParseProxyHeaderV2(const ProxyHdr & hdr, int read_size) {
    int size = 16 + ntohs(hdr.len);
    if (read_size < size) { // truncated or too large header
        return -__LINE__;
    }
    switch (hdr.ver_cmd & 0xF) {
    case PP2_CMD_PROXY:
        switch (hdr.fam) {
        case PP2_FAM_TCP4:
            src_addr_.ss_family = AF_INET;
            ((struct sockaddr_in *) &src_addr_)->sin_addr.s_addr = hdr.addr.ip4.src_addr;
            ((struct sockaddr_in *) &src_addr_)->sin_port = hdr.addr.ip4.src_port;
            dst_addr_.ss_family = AF_INET;
            ((struct sockaddr_in *) &dst_addr_)->sin_addr.s_addr = hdr.addr.ip4.dst_addr;
            ((struct sockaddr_in *) &dst_addr_)->sin_port = hdr.addr.ip4.dst_port;
            return size;
        case PP2_FAM_TCP6:
            src_addr_.ss_family = AF_INET6;
            memcpy(&((struct sockaddr_in6 *) &src_addr_)->sin6_addr, hdr.addr.ip6.src_addr, 16);
            ((struct sockaddr_in6 *) &src_addr_)->sin6_port = hdr.addr.ip6.src_port;
            dst_addr_.ss_family = AF_INET6;
            memcpy(&((struct sockaddr_in6 *) &dst_addr_)->sin6_addr, hdr.addr.ip6.dst_addr, 16);
            ((struct sockaddr_in6 *) &dst_addr_)->sin6_port = hdr.addr.ip6.dst_port;
            return size;
        case PP2_FAM_UNSPEC:
            return size; // unknown protocol, keep local connection address
        default:
            return -__LINE__; // unsupport protocol
        }
    case PP2_CMD_LOCAL:
        return size; // keep local connection address for LOCAL
    default:
        return -__LINE__; // unsupport command
    }
    return -__LINE__;
}

int ProxyProtocolHandler::SendProxyHeader(int fd) {
    int ret = 0;