cdramey/wstationd
1
0
Fork 0
Code Releases Activity

Files are now written with the contents of the stream

Browse Source
This commit is contained in:
Christopher Ramey 2015-02-04 20:15:15 +00:00 committed by cdramey
parent b9d2f8b44f
commit 9e9f280de6
1 changed files with 52 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

60
src/wstationd.c
View File

@ -14,6 +14,8 @@
#include <getopt.h>
#include "wstationd.h"
#define BUFFERSIZE 4096
static char *exec_name;
@ -42,10 +44,12 @@ int main(int argc, char *argv[])
int port = 10800; // Default port
uint32_t address = INADDR_ANY; // Default listen address
int sockflag = 1, sockfd = 0, connfd = 0;
int connflag, sockflag = 1;
int sockfd = 0, connfd = 0;
struct sockaddr_in sock, client;
socklen_t client_sz = sizeof(client);
char timebuff[15], fnbuff[35];
char connbuff[BUFFERSIZE], timebuff[15], fnbuff[35];
ssize_t connbuff_sz;
time_t timeraw;
struct tm* timeptr;
@ -155,6 +159,25 @@ int main(int argc, char *argv[])
fprintf(stderr, "%s: accept failed - %s\n",
exec_name, strerror(errno)
);
continue;
}
// Get current flags before setting nonblock
if((connflag = fcntl(connfd, F_GETFL)) == -1){
fprintf(stderr, "%s: cannot get connection flags - %s\n",
exec_name, strerror(errno)
);
close(connfd);
continue;
}
// Set nonblocking on socket
if(fcntl(connfd, F_SETFL, connflag | O_NONBLOCK) == -1){
fprintf(stderr, "%s: cannot set connection flags - %s\n",
exec_name, strerror(errno)
);
close(connfd);
continue;
}
// Collect the time of the connection, process it into
@ -170,8 +193,9 @@ int main(int argc, char *argv[])
timebuff, inet_ntoa(client.sin_addr)
);
if((writefds[i] = open(fnbuff, O_WRONLY | O_CREAT, 0644)) == -1){
writefds[i] = 0;
// Open a file to dump our data to
if((writefds[connfd] = open(fnbuff, O_WRONLY | O_CREAT, 0644)) == -1){
writefds[connfd] = 0;
fprintf(stderr, "%s: cannot open - %s\n",
exec_name, strerror(errno)
);
@ -180,16 +204,36 @@ int main(int argc, char *argv[])
} else {
FD_SET(connfd, &masterfds);
}
connfd = 0;
// Read from existing connections
} else {
if(writefds[i] != 0){
// Do the read
connbuff_sz = recv(i, connbuff, BUFFERSIZE, 0);
if(connbuff_sz > 0){
// If the read fails, throw an error, and set the
// buffer size to 0, which forces a disconnect.
if(write(writefds[i], connbuff, connbuff_sz) == -1){
fprintf(stderr, "%s: file write error - %s\n",
exec_name, strerror(errno)
);
connbuff_sz = 0;
}
}
// If the buffer is empty (which with select(), will only happen
// on disconnect, disconnect and cleanup.
if(connbuff_sz == 0){
close(writefds[i]);
writefds[i] = 0;
FD_CLR(i, &masterfds);
close(i);
} else if(connbuff_sz == -1){
fprintf(stderr, "%s: recv error - %s\n",
exec_name, strerror(errno)
);
FD_CLR(i, &masterfds);
close(i);
}
FD_CLR(i, &masterfds);
close(i);
}
}
}