git.ucc.asn.au
/
tpg
/
opendispense2.git
/ commitdiff
commit
grep
author
committer
pickaxe
?
search:
re
summary
|
shortlog
|
log
|
commit
| commitdiff |
tree
raw
|
patch
|
inline
| side by side (parent:
7aab29a
)
Fixed not checking return value of recv()
author
John Hodge
<
[email protected]
>
Mon, 7 Mar 2011 01:13:45 +0000
(09:13 +0800)
committer
John Hodge
<
[email protected]
>
Mon, 7 Mar 2011 01:13:45 +0000
(09:13 +0800)
src/client/main.c
patch
|
blob
|
history
diff --git
a/src/client/main.c
b/src/client/main.c
index
fd5915d
..
0995d3b
100644
(file)
--- a/
src/client/main.c
+++ b/
src/client/main.c
@@
-287,11
+287,11
@@
int main(int argc, char *argv[])
gbDryRun = 1;
break;
default:
gbDryRun = 1;
break;
default:
-
if( !isdigit(argv[i][0]) ) {
-
fprintf(stderr, "%s: Unknown switch '%s'\n", argv[0], argv[i]);
-
ShowUsage();
-
return RV_ARGUMENTS;
-
}
+
//
if( !isdigit(argv[i][0]) ) {
+
//
fprintf(stderr, "%s: Unknown switch '%s'\n", argv[0], argv[i]);
+
//
ShowUsage();
+
//
return RV_ARGUMENTS;
+
//
}
if( text_argc + 1 == MAX_TXT_ARGS )
{
fprintf(stderr, "ERROR: Too many arguments\n");
if( text_argc + 1 == MAX_TXT_ARGS )
{
fprintf(stderr, "ERROR: Too many arguments\n");
@@
-2160,6
+2160,10
@@
char *ReadLine(int Socket)
}
else {
len = recv(Socket, buf+bufPos, BUFSIZ-1-bufPos, 0);
}
else {
len = recv(Socket, buf+bufPos, BUFSIZ-1-bufPos, 0);
+ if( len < 0 ) {
+ free(ret);
+ return strdup("499 Client Connection Error\n");
+ }
buf[bufPos+len] = '\0';
}
buf[bufPos+len] = '\0';
}
UCC
git Repository :: git.ucc.asn.au