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
|
#include "socketstream.h"
#include "socket.h"
#include <assert.h>
#include <string.h>
static int SocketStreamClose(Stream *base)
{
int rv;
SocketStream *stream = (SocketStream *) base;
rv = SocketDisconnect(stream->sock);
stream->sock = -1;
return rv;
}
static int64_t SocketStreamRead(void *ptr, size_t size, Stream *stream)
{
SocketStream *ss = (SocketStream *) stream;
size_t received = 0;
if (ss->sock == -1)
return -1;
while (received < size)
{
char *p = ((char *) ptr) + received;
int64_t rv = SocketRecv(ss->sock, p, size - received, 0);
/* Error */
if (rv == -1)
return -1;
/* TODO: Closed? */
if (rv == 0)
break;
received += (size_t) rv;
}
return received;
}
static int64_t SocketStreamWrite(const void *ptr, size_t size, Stream *stream)
{
SocketStream *ss = (SocketStream *) stream;
size_t written = 0;
if (ss->sock == -1)
return -1;
while (written < size)
{
const char *p = ((char *) ptr) + written;
int64_t rv = SocketSend(ss->sock, p, size - written, 0);
if (rv == -1)
return -1;
written += (size_t) rv;
}
return written;
}
static const StreamOps SocketStreamOps =
{
SocketStreamRead,
SocketStreamWrite,
SocketStreamClose,
NULL,
NULL
};
int SocketStreamOpen(SocketStream *stream, int sock)
{
assert(stream != NULL);
assert(sock != -1);
memset(stream, 0, sizeof(*stream));
stream->sock = sock;
stream->base.ops = &SocketStreamOps;
return 0;
}
|
