diff options
-rw-r--r-- | src/connection.cpp | 8 | ||||
-rw-r--r-- | src/test/connect/main.cpp | 38 |
2 files changed, 43 insertions, 3 deletions
diff --git a/src/connection.cpp b/src/connection.cpp index d66ff97..3d3c094 100644 --- a/src/connection.cpp +++ b/src/connection.cpp | |||
@@ -10,6 +10,7 @@ | |||
10 | #include <netinet/in.h> | 10 | #include <netinet/in.h> |
11 | #include <netdb.h> | 11 | #include <netdb.h> |
12 | #include <arpa/inet.h> | 12 | #include <arpa/inet.h> |
13 | #include <errno.h> | ||
13 | #include "exceptions.h" | 14 | #include "exceptions.h" |
14 | 15 | ||
15 | Connection::Connection() | 16 | Connection::Connection() |
@@ -208,15 +209,16 @@ int Connection::readInput() | |||
208 | //memset( buffer, 0, 2048 ); | 209 | //memset( buffer, 0, 2048 ); |
209 | 210 | ||
210 | nbytes = read( nSocket, buffer, 2048 ); | 211 | nbytes = read( nSocket, buffer, 2048 ); |
211 | if (nbytes < 0) | 212 | if( nbytes < 0 && errno != 0 && errno != EAGAIN ) |
212 | { | 213 | { |
214 | printf("errno: %d, %s\n", errno, strerror( errno ) ); | ||
213 | /* Read error. */ | 215 | /* Read error. */ |
214 | //perror("readInput"); | 216 | //perror("readInput"); |
215 | throw ConnectionException( excodeReadError, "Read error"); | 217 | throw ConnectionException( excodeReadError, "Read error: %s", strerror( errno ) ); |
216 | } | 218 | } |
217 | else | 219 | else |
218 | { | 220 | { |
219 | if( nbytes == 0 ) | 221 | if( nbytes <= 0 ) |
220 | break; | 222 | break; |
221 | nTotalRead += nbytes; | 223 | nTotalRead += nbytes; |
222 | appendInput( buffer, nbytes ); | 224 | appendInput( buffer, nbytes ); |
diff --git a/src/test/connect/main.cpp b/src/test/connect/main.cpp new file mode 100644 index 0000000..a9fca64 --- /dev/null +++ b/src/test/connect/main.cpp | |||
@@ -0,0 +1,38 @@ | |||
1 | #include <stdio.h> | ||
2 | #include <stdlib.h> | ||
3 | #include <unistd.h> | ||
4 | #include <fcntl.h> | ||
5 | #include "connection.h" | ||
6 | |||
7 | int main() | ||
8 | { | ||
9 | Connection c; | ||
10 | c.open("127.0.0.1", 12457 ); | ||
11 | |||
12 | { | ||
13 | int newSocket = c.getSocket(); | ||
14 | int flags; | ||
15 | |||
16 | flags = fcntl(newSocket, F_GETFL, 0); | ||
17 | flags |= O_NONBLOCK; | ||
18 | if (fcntl(newSocket, F_SETFL, flags) < 0) | ||
19 | { | ||
20 | return false; | ||
21 | } | ||
22 | } | ||
23 | |||
24 | for( int i = 0; i < 50; i++ ) | ||
25 | { | ||
26 | usleep( 100000 ); | ||
27 | int nbytes = c.readInput(); | ||
28 | if( nbytes == 0 ) | ||
29 | printf("0 bytes, EOF?\n"); | ||
30 | else | ||
31 | printf("Got %d bytes, whacky...\n", nbytes ); | ||
32 | } | ||
33 | |||
34 | c.close(); | ||
35 | |||
36 | return 0; | ||
37 | } | ||
38 | |||