1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
|
/* Stest.c: this file contains code which allows one to test if data
* is present via the socket and not be blocked (not held, not made to sleep)
*
* Returns: number of bytes of data awaiting perusal (which may be 0)
* or EOF if unable to "select" the socket
* or select returns 1 but nothing is on socket
*
* Authors: Charles E. Campbell, Jr.
* Terry McRoberts
* Copyright: Copyright (C) 1999-2005 Charles E. Campbell, Jr. {{{1
* Permission is hereby granted to use and distribute this code,
* with or without modifications, provided that this copyright
* notice is copied with it. Like anything else that's free,
* Stest.c is provided *as is* and comes with no warranty
* of any kind, either expressed or implied. By using this
* software, you agree that in no event will the copyright
* holder be liable for any damages resulting from the use
* of this software.
* Date: Aug 22, 2005
*/
#include <stdio.h>
#define SSLNEEDTIME
#include "sockets.h"
/* ---------------------------------------------------------------------
* Source Code:
*/
#ifdef __PROTOTYPE__
int Stest(Socket *skt)
#else
int Stest(skt)
Socket *skt;
#endif
{
fd_set emask;
fd_set rmask;
fd_set wmask;
int ret;
short result;
static char buf[PM_BIGBUF];
struct timeval timeout;
/* sanity check */
if(!skt) {
return -1;
}
FD_ZERO(&rmask);
FD_SET(skt->skt,&rmask);
FD_ZERO(&wmask);
FD_ZERO(&emask);
timeout.tv_sec = 0;
timeout.tv_usec= 0;
#ifdef SSLNOPEEK
result = select(skt->skt+1,rmask.fds_bits,wmask.fds_bits,emask.fds_bits,&timeout);
#else
result = select(skt->skt+1,&rmask,&wmask,&emask,&timeout);
#endif /* SSLNOPEEK */
if(result < 0) {
return EOF;
}
/* server sockets return the select result (client awaiting connection) */
if(skt->type == PM_SERVER) {
return result;
}
#ifdef SSLNOPEEK
if(FD_ISSET(skt->skt,&rmask)) {
return 1;
}
#else /* #ifdef SSLNOPEEK ... #else ... #endif */
/* test if message available from socket, return qty bytes avail */
if(FD_ISSET(skt->skt,&rmask)) {
ret= recv(skt->skt,buf,PM_BIGBUF-1,MSG_PEEK);
/* return error indication when select returned a result of 1
* (indicating that *something* is on the socket)
* and recv indicates that *nothing* is on the socket
*/
if(result == 1 && ret == 0) ret= EOF;
return ret;
}
#endif /* #ifdef SSLNOPEEK ... #else ... #endif */
/* socket is empty */
return 0;
}
/* ---------------------------------------------------------------------
* vim: ts=4
*/
|
