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
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
|
/* HSOCKET.C (c) Copyright Hercules development, 2003-2009 */
/* Socket read/write routines */
// $Id: hsocket.c 5125 2009-01-23 12:01:44Z bernard $
//
// $Log$
// Revision 1.8 2007/06/23 00:04:11 ivan
// Update copyright notices to include current year (2007)
//
// Revision 1.7 2006/12/08 09:43:27 jj
// Add CVS message log
//
#include "hstdinc.h"
#define _HSOCKET_C_
#define _HUTIL_DLL_
#include "hercules.h"
/************************************************************************
NAME: read_socket - read a passed number of bytes from a socket.
PURPOSE: This routine is used in place of read to read a passed number
of bytes from a socket. A read on a socket will return less
than the number of bytes requested if there are less currenly
available. Thus we read in a loop till we get all we want.
PARAMETERS:
1. fd - int (INPUT)
This is the file descriptor for the socket to be read.
2. ptr - pointer to void (OUTPUT)
This is a pointer to where the data is to be put.
3. nbytes - int (INPUT)
This is the number of bytes to read.
FUNCTIONS :
1. Read in a loop till an error occurs or the data is read.
OUTPUTS:
data into the buffer
*************************************************************************/
DLL_EXPORT int read_socket(int fd, void *_ptr, int nbytes)
{
int nleft, nread;
char *ptr;
nleft = nbytes;
ptr=_ptr;
while (nleft > 0)
{
#ifdef _MSVC_
nread = recv(fd, ptr, nleft, 0);
if ((nread == SOCKET_ERROR) || (nread < 0))
{
nread = -1;
break; /* error, return < 0 */
}
if (nread == 0)
break;
#else
nread = read(fd, ptr, nleft);
if (nread < 0)
return(nread); /* error, return < 0 */
else
if (nread == 0) /* eof */
break;
#endif
nleft -= nread;
ptr += nread;
} /* end of do while */
/* if (nleft != 0)
logmsg (_("BOB123 read_socket: Read of %d bytes requested, %d bytes actually read\n"),
nbytes, nbytes - nleft);*/
return (nbytes - nleft);
} /* end of read_socket */
/************************************************************************
NAME: write_socket - write a passed number of bytes into a socket.
PURPOSE: This routine is used in place of write to write a passed number
of bytes into a socket. A write on a socket may take less
than the number of bytes requested if there is currently insufficient
buffer space available. Thus we write in a loop till we get all we want.
PARAMETERS:
1. fd - int (OUTPUT)
This is the file descriptor for the socket to be written.
2. ptr - pointer to void (INPUT)
This is a pointer to where the data is to be gotten from.
3. nbytes - int (INPUT)
This is the number of bytes to write.
FUNCTIONS :
1. Write in a loop till an error occurs or the data is written.
OUTPUTS:
Data to the socket.
*************************************************************************/
/*
* Write "n" bytes to a descriptor.
* Use in place of write() when fd is a stream socket.
*/
DLL_EXPORT int write_socket(int fd, const void *_ptr, int nbytes)
{
int nleft, nwritten;
const char *ptr;
nleft = nbytes;
ptr=_ptr;
while (nleft > 0)
{
#ifdef _MSVC_
nwritten = send(fd, ptr, nleft, 0);
if (nwritten <= 0)
{
return(nwritten); /* error */
}
#else
nwritten = write(fd, ptr, nleft);
if (nwritten <= 0)
return(nwritten); /* error */
#endif
nleft -= nwritten;
ptr += nwritten;
} /* end of do while */
return(nbytes - nleft);
} /* end of write_socket */
|
