Annotation of parser3/src/main/pa_socks.C, revision 1.40
1.3 paf 1: /** @file
1.1 paf 2: Parser: socks initialization/finalization.
3:
1.35 moko 4: Copyright (c) 2001-2026 Art. Lebedev Studio (https://www.artlebedev.com)
1.32 moko 5: Authors: Konstantin Morshnev <moko@design.ru>, Alexandr Petrosian <paf@design.ru>
1.15 paf 6: */
1.1 paf 7:
1.40 ! moko 8: volatile const char * IDENT_PA_SOCKS_C="$Id: pa_socks.C,v 1.39 2026/10/17 23:11:25 moko Exp $";
1.2 paf 9:
10: #include "pa_config_includes.h"
1.36 moko 11: #include "pa_socks.h"
1.1 paf 12:
1.28 moko 13: #ifdef _MSC_VER
14:
1.9 paf 15: #include "pa_exception.h"
1.19 paf 16: #include "pa_string.h"
1.11 paf 17:
1.19 paf 18: #include <windows.h>
19:
1.1 paf 20: WSADATA wsaData;
21:
1.22 paf 22: void pa_socks_init() {
1.1 paf 23: WORD wVersionRequested;
24: int err;
25: wVersionRequested = MAKEWORD( 1, 1 );
26: err = WSAStartup( wVersionRequested, &wsaData );
27: if ( err != 0 ) {
1.28 moko 28: /* Tell the user that we could not find a usable */
1.33 moko 29: throw Exception(0, 0, "cannot WSAStartup, err=%d", err);
1.1 paf 30: }
31: }
32:
1.22 paf 33: void pa_socks_done() {
1.1 paf 34: /* Confirm that the WinSock DLL supports 2.2.*/
35: /* Note that if the DLL supports versions greater */
36: /* than 2.2 in addition to 2.2, it will still return */
37: /* 2.2 in wVersion since that is the version we */
38: /* requested. */
1.28 moko 39: if ( LOBYTE( wsaData.wVersion ) == 2 || HIBYTE( wsaData.wVersion ) == 2 ) {
40: WSACleanup();
41: return;
1.1 paf 42: }
43: }
44:
1.23 paf 45: const char* pa_socks_strerr(int no) {
46: char buf[MAX_STRING];
47: buf[0]=0;
48: size_t error_size=FormatMessage(
49: FORMAT_MESSAGE_FROM_SYSTEM |
50: FORMAT_MESSAGE_IGNORE_INSERTS,
51: NULL,
52: no,
53: 0, // Default language
54: (LPTSTR) &buf,
55: sizeof(buf),
56: NULL
57: );
58: if(error_size>3) // ".\r\n"
59: buf[error_size-3]=0;
60: return buf[0]? pa_strdup(buf): "unknown error";
61: }
62: int pa_socks_errno() {
63: return WSAGetLastError();
64: }
65:
1.37 moko 66: bool pa_socks_is_reset(int no) {
67: return no==WSAECONNRESET || no==WSAECONNABORTED;
68: }
69:
1.38 moko 70: bool pa_socks_is_interrupted(int no) {
71: return no==WSAEINTR;
72: }
73:
1.40 ! moko 74: // there are no signals, calls are not restarted for the time left
! 75: typedef int sock_time;
! 76:
! 77: static sock_time sock_now() {
! 78: return 0;
! 79: }
! 80:
! 81: static bool sock_ready_within_remaining_timeout(SOCKET, bool, sock_time) {
! 82: return false;
! 83: }
! 84:
1.36 moko 85: int pa_sock_ready(SOCKET sock, bool for_write, int timeout_ms) {
86: WSAPOLLFD fd={sock, (SHORT)(for_write ? POLLWRNORM : POLLRDNORM), 0};
87: return WSAPoll(&fd, 1, timeout_ms);
88: }
89:
1.1 paf 90: #else
1.10 paf 91:
1.36 moko 92: #include <poll.h>
93:
1.22 paf 94: void pa_socks_init() {}
95: void pa_socks_done() {}
1.23 paf 96:
97: const char* pa_socks_strerr(int no) {
98: return strerror(no);
99: }
100: int pa_socks_errno() {
101: return errno;
102: }
1.1 paf 103:
1.37 moko 104: bool pa_socks_is_reset(int no) {
105: return no==ECONNRESET;
106: }
107:
1.38 moko 108: bool pa_socks_is_interrupted(int no) {
109: return no==EINTR;
110: }
111:
1.40 ! moko 112: typedef struct timeval sock_time;
! 113:
! 114: static sock_time sock_now() {
! 115: sock_time now;
! 116: gettimeofday(&now, 0);
! 117: return now;
! 118: }
! 119:
! 120: static long long timeval_ms(const struct timeval &tv) {
! 121: return (long long)tv.tv_sec*1000+tv.tv_usec/1000;
! 122: }
! 123:
1.36 moko 124: int pa_sock_ready(SOCKET sock, bool for_write, int timeout_ms) {
125: struct pollfd fd={sock, (short)(for_write ? POLLOUT : POLLIN), 0};
1.40 ! moko 126: sock_time start=sock_now();
1.39 moko 127:
128: int left=timeout_ms;
129: int result;
1.40 ! moko 130: // GC stops the threads with signals: continue poll for the time left
1.39 moko 131: while((result=poll(&fd, 1, left))<0 && pa_socks_is_interrupted(errno)) {
1.40 ! moko 132: if(timeout_ms>0){
! 133: long long passed=timeval_ms(sock_now()) - timeval_ms(start);
! 134: if(passed>=timeout_ms)
! 135: return 0;
! 136: left=passed<0 ? timeout_ms : timeout_ms-(int)passed;
! 137: }
1.39 moko 138: }
139: return result;
1.36 moko 140: }
141:
1.40 ! moko 142: // the system restarts recv/send with the whole socket timeout again, under load it would never expire: wait for the remaining time
! 143: static bool sock_ready_within_remaining_timeout(SOCKET sock, bool for_write, sock_time start) {
! 144: struct timeval socket_timeout;
! 145: socklen_t timeout_size=sizeof(socket_timeout);
! 146: if(getsockopt(sock, SOL_SOCKET, for_write ? SO_SNDTIMEO : SO_RCVTIMEO, &socket_timeout, &timeout_size)<0 || !(socket_timeout.tv_sec || socket_timeout.tv_usec))
! 147: return true; // no timeout
! 148:
! 149: long long passed=timeval_ms(sock_now())-timeval_ms(start);
! 150: long long left=timeval_ms(socket_timeout)-passed;
! 151: int ready=left>0 ? pa_sock_ready(sock, for_write, left<INT_MAX ? (int)left : INT_MAX) : 0;
! 152: if(ready==0)
! 153: errno=EAGAIN;
! 154: return ready>0;
! 155: }
! 156:
1.1 paf 157: #endif
1.38 moko 158:
159: ssize_t pa_sock_recv(SOCKET sock, char *buf, size_t len) {
1.40 ! moko 160: sock_time start=sock_now();
1.38 moko 161: ssize_t result;
1.40 ! moko 162: while((result=recv(sock, buf, len, 0)) < 0 && pa_socks_is_interrupted(pa_socks_errno()) && sock_ready_within_remaining_timeout(sock, false, start));
1.38 moko 163: return result;
164: }
165:
166: ssize_t pa_sock_send(SOCKET sock, const char *buf, size_t len) {
1.40 ! moko 167: sock_time start=sock_now();
1.38 moko 168: ssize_t result;
1.40 ! moko 169: while((result=send(sock, buf, len, 0)) < 0 && pa_socks_is_interrupted(pa_socks_errno()) && sock_ready_within_remaining_timeout(sock, true, start));
1.38 moko 170: return result;
171: }
E-mail:
![[BACK]](/cvsweb/icons/back.gif){kind=icon}
Return to pa_socks.C CVS log ![[TXT]](/cvsweb/icons/text.gif){kind=icon}
![[DIR]](/cvsweb/icons/dir.gif){kind=icon}
Up to [parser3project] / parser3 / src / main