Commit | Line | Data |
---|---|---|
aa846a75 RK |
1 | /* |
2 | * This file is part of DisOrder | |
3 | * Copyright (C) 2013 Richard Kettlewell | |
4 | * | |
5 | * This program is free software: you can redistribute it and/or modify | |
6 | * it under the terms of the GNU General Public License as published by | |
7 | * the Free Software Foundation, either version 3 of the License, or | |
8 | * (at your option) any later version. | |
9 | * | |
10 | * This program is distributed in the hope that it will be useful, | |
11 | * but WITHOUT ANY WARRANTY; without even the implied warranty of | |
12 | * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the | |
13 | * GNU General Public License for more details. | |
14 | * | |
15 | * You should have received a copy of the GNU General Public License | |
16 | * along with this program. If not, see <http://www.gnu.org/licenses/>. | |
17 | */ | |
18 | /** @file lib/socketio.c | |
19 | * @brief Buffered socket IO | |
20 | */ | |
21 | #include "common.h" | |
22 | #include "socketio.h" | |
23 | #include "mem.h" | |
24 | ||
25 | #include <errno.h> | |
26 | #include <sys/types.h> | |
27 | #if HAVE_SYS_SOCKET_H | |
28 | # include <sys/socket.h> | |
29 | #endif | |
30 | #if HAVE_UNISTD_H | |
31 | # include <unistd.h> | |
32 | #endif | |
33 | ||
34 | void socketio_init(struct socketio *sio, SOCKET sd) { | |
35 | sio->sd = sd; | |
36 | sio->inputptr = sio->inputlimit = sio->input; | |
37 | sio->outputused = 0; | |
38 | sio->error = 0; | |
39 | } | |
40 | ||
41 | int socketio_write(struct socketio *sio, const void *buffer, size_t n) { | |
42 | size_t chunk; | |
43 | while(n > 0) { | |
44 | chunk = n > sizeof sio->output ? sizeof sio->output : n; | |
45 | if(chunk) { | |
46 | memcpy(sio->output + sio->outputused, buffer, chunk); | |
47 | sio->outputused += chunk; | |
48 | buffer = (char *)buffer + chunk; | |
49 | n -= chunk; | |
50 | } | |
51 | if(sio->outputused == sizeof sio->output) | |
52 | if(socketio_flush(sio)) | |
53 | return -1; | |
54 | } | |
55 | return 0; | |
56 | } | |
57 | ||
58 | static int socketio_fill(struct socketio *sio) { | |
59 | int n = recv(sio->sd, sio->input, sizeof sio->input, 0); | |
60 | if(n <= 0) { | |
61 | sio->error = n < 0 ? socket_error() : -1; | |
62 | return -1; | |
63 | } | |
64 | sio->inputptr = sio->input; | |
65 | sio->inputlimit = sio->input + n; | |
66 | return 0; | |
67 | } | |
68 | ||
69 | int socketio_getc(struct socketio *sio) { | |
70 | if(sio->inputptr >= sio->inputlimit) { | |
71 | if(socketio_fill(sio)) | |
72 | return EOF; | |
73 | } | |
74 | return *sio->inputptr++; | |
75 | } | |
76 | ||
77 | int socketio_flush(struct socketio *sio) { | |
78 | size_t written = 0; | |
79 | while(written < sio->outputused) { | |
80 | int n = send(sio->sd, sio->output + written, sio->outputused - written, 0); | |
81 | if(n < 0) { | |
82 | sio->error = socket_error(); | |
83 | return -1; | |
84 | } | |
85 | written += n; | |
86 | } | |
87 | sio->outputused = 0; | |
88 | return 0; | |
89 | } | |
90 | ||
91 | void socketio_close(struct socketio *sio) { | |
92 | socketio_flush(sio); | |
93 | closesocket(sio->sd); | |
94 | } |