/* Copyright (C) 2002-2005 RealVNC Ltd. All Rights Reserved. * * This is free software; you can redistribute it and/or modify * it under the terms of the GNU General Public License as published by * the Free Software Foundation; either version 2 of the License, or * (at your option) any later version. * * This software is distributed in the hope that it will be useful, * but WITHOUT ANY WARRANTY; without even the implied warranty of * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the * GNU General Public License for more details. * * You should have received a copy of the GNU General Public License * along with this software; if not, write to the Free Software * Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, * USA. */ #ifdef HAVE_CONFIG_H #include #endif #include #include #include #include #ifdef _WIN32 #include #define errorNumber WSAGetLastError() #define close closesocket #include #else #include #include #include #define errorNumber errno #endif /* Old systems have select() in sys/time.h */ #ifdef HAVE_SYS_SELECT_H #include #endif #include #include using namespace rdr; FdInStream::FdInStream(int fd_, bool closeWhenDone_) : fd(fd_), closeWhenDone(closeWhenDone_) { } FdInStream::~FdInStream() { if (closeWhenDone) close(fd); } bool FdInStream::fillBuffer() { size_t n = readFd((uint8_t*)end, availSpace()); if (n == 0) return false; end += n; return true; } // // readFd() reads up to the given length in bytes from the // file descriptor into a buffer. Zero is // returned if no bytes can be read. Otherwise it returns the number of bytes read. It // never attempts to recv() unless select() indicates that the fd is readable - // this means it can be used on an fd which has been set non-blocking. It also // has to cope with the annoying possibility of both select() and recv() // returning EINTR. // size_t FdInStream::readFd(uint8_t* buf, size_t len) { int n; do { fd_set fds; struct timeval tv; tv.tv_sec = tv.tv_usec = 0; FD_ZERO(&fds); FD_SET(fd, &fds); n = select(fd+1, &fds, nullptr, nullptr, &tv); } while (n < 0 && errorNumber == EINTR); if (n < 0) throw SystemException("select", errorNumber); if (n == 0) return 0; do { n = ::recv(fd, (char*)buf, len, 0); } while (n < 0 && errorNumber == EINTR); if (n < 0) throw SystemException("read", errorNumber); if (n == 0) throw EndOfStream(); return n; } ption> Nextcloud server, a safe home for all your data: https://github.com/nextcloud/serverwww-data
aboutsummaryrefslogtreecommitdiffstats
blob: 0c3aa6ae29ce4c4a3c111deb820db54de462e234 (plain)
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