X-Git-Url: https://git.sesse.net/?p=rdpsrv;a=blobdiff_plain;f=tcp.c;fp=tcp.c;h=fef986eafabd56c8d70ca9bfb5829ddfcbf872ba;hp=4e114223ef11e6a392544ff3e79cf6d85c1f3f0a;hb=4539b581a53654e64fb3c6d0de04ff31d035b81f;hpb=b3ae47054997c61d01087d17a3ab001b0a54324b diff --git a/tcp.c b/tcp.c index 4e11422..fef986e 100644 --- a/tcp.c +++ b/tcp.c @@ -134,85 +134,18 @@ tcp_recv(STREAM s, uint32 length) /* Establish a connection on the TCP layer */ BOOL -tcp_connect(char *server) +tcp_recv_connect(int server_sock) { - int true_value = 1; - -#ifdef IPv6 - - int n; - struct addrinfo hints, *res, *ressave; - char tcp_port_rdp_s[10]; - - snprintf(tcp_port_rdp_s, 10, "%d", tcp_port_rdp); - - memset(&hints, 0, sizeof(struct addrinfo)); - hints.ai_family = AF_UNSPEC; - hints.ai_socktype = SOCK_STREAM; - - n = getaddrinfo(server, tcp_port_rdp_s, &hints, &res); - - if (n < 0) - { - error("getaddrinfo: %s\n", gai_strerror(n)); - return False; - } - - ressave = res; - sock = -1; - while (res) - { - sock = socket(res->ai_family, res->ai_socktype, res->ai_protocol); - if (!(sock < 0)) - { - if (connect(sock, res->ai_addr, res->ai_addrlen) == 0) - break; - close(sock); - sock = -1; - } - res = res->ai_next; - } - freeaddrinfo(ressave); - - if (sock == -1) - { - error("%s: unable to connect\n", server); - return False; + struct sockaddr_in sin; + unsigned true_value = 1; + socklen_t len = sizeof(sin); + sock = accept(server_sock, (struct sockaddr *)&sin, &len); + + if (sock == -1) { + perror("accept()"); + exit(1); } -#else /* no IPv6 support */ - - struct hostent *nslookup; - struct sockaddr_in servaddr; - - if ((nslookup = gethostbyname(server)) != NULL) - { - memcpy(&servaddr.sin_addr, nslookup->h_addr, sizeof(servaddr.sin_addr)); - } - else if ((servaddr.sin_addr.s_addr = inet_addr(server)) == INADDR_NONE) - { - error("%s: unable to resolve host\n", server); - return False; - } - - if ((sock = socket(AF_INET, SOCK_STREAM, 0)) < 0) - { - error("socket: %s\n", strerror(errno)); - return False; - } - - servaddr.sin_family = AF_INET; - servaddr.sin_port = htons(tcp_port_rdp); - - if (connect(sock, (struct sockaddr *) &servaddr, sizeof(struct sockaddr)) < 0) - { - error("connect: %s\n", strerror(errno)); - close(sock); - return False; - } - -#endif /* IPv6 */ - setsockopt(sock, IPPROTO_TCP, TCP_NODELAY, (void *) &true_value, sizeof(true_value)); in.size = 4096;