cdramey/wstationd
1
0
Fork 0
Code Releases Activity

connections now create files

Browse Source
This commit is contained in:
Christopher Ramey
2015-02-04 16:38:40 +00:00
committed by cdramey
parent d5bbdd6c44
commit b9d2f8b44f
1 changed files with 62 additions and 26 deletions
Download Patch File Download Diff File Expand all files Collapse all files

88
src/wstationd.c
View File

@ -5,6 +5,7 @@
#include <string.h>
#include <stdlib.h>
#include <limits.h>
#include <fcntl.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <netinet/in.h>
@ -37,19 +38,22 @@ int main(int argc, char *argv[])
{ NULL, 0, NULL, 0 }
};
int flag = 1;
int sockfd = 0, connfd = 0;
int port = 10800;
uint32_t address = INADDR_ANY;
struct sockaddr_in sock;
fd_set readfds, masterfds;
struct sockaddr_in client;
int port = 10800; // Default port
uint32_t address = INADDR_ANY; // Default listen address
int sockflag = 1, sockfd = 0, connfd = 0;
struct sockaddr_in sock, client;
socklen_t client_sz = sizeof(client);
char timebuff[15], fnbuff[35];
time_t timeraw;
struct tm* timeptr;
// Allocate space for the output fds
int writefds[FD_SETSIZE];
memset(&writefds, 0, sizeof(int) * FD_SETSIZE);
exec_name = basename(argv[0]);
for(int ch; (ch = getopt_long(argc, argv, "b:p:vh0", longopts, NULL)) != -1;){
@ -107,7 +111,7 @@ int main(int argc, char *argv[])
// This sets REUSE on the socket so it's easily reallocated if
// this program dies
setsockopt(sockfd, SOL_SOCKET, SO_REUSEADDR, &flag, sizeof(flag));
setsockopt(sockfd, SOL_SOCKET, SO_REUSEADDR, &sockflag, sizeof(sockflag));
memset(&sock, 0, sizeof(sock));
sock.sin_family = AF_INET;
@ -123,47 +127,79 @@ int main(int argc, char *argv[])
}
// Listen on port
if(listen(sockfd, 5) != 0){
if(listen(sockfd, 10) != 0){
fprintf(stderr, "%s: cannot listen - %s\n",
exec_name, strerror(errno)
);
goto shutdown_error;
}
FD_ZERO(&masterfds);
FD_SET(sockfd, &masterfds);
while(1){
// Wait for a connection on our bound socket
connfd = accept(sockfd, (struct sockaddr*)&client, &client_sz);
if(connfd == -1){
fprintf(stderr, "%s: accept failed - %s\n",
readfds = masterfds;
if(select(FD_SETSIZE, &readfds, NULL, NULL, NULL) == -1){
fprintf(stderr, "%s: socket select error - %s\n",
exec_name, strerror(errno)
);
goto shutdown_error;
}
// Collect the time of the connection, process it into
// a short string
time(&timeraw);
timeptr = localtime(&timeraw);
strftime(&timebuff[0], 15, "%Y%m%d%H%M%S", timeptr);
timeptr = NULL;
for(int i = 0; i < FD_SETSIZE; i++){
if(FD_ISSET(i, &readfds)){
// Deal with new connections
if(i == sockfd){
connfd = accept(sockfd, (struct sockaddr*)&client, &client_sz);
if(connfd == -1){
fprintf(stderr, "%s: accept failed - %s\n",
exec_name, strerror(errno)
);
}
// Take the short date/time connection string and add the client address
// to create a filename
snprintf(&fnbuff[0], 34, "%s-%s.dat", timebuff, inet_ntoa(client.sin_addr));
// Collect the time of the connection, process it into
// a short string
time(&timeraw);
timeptr = localtime(&timeraw);
strftime(&timebuff[0], 15, "%Y%m%d%H%M%S", timeptr);
timeptr = NULL;
fprintf(stdout, "filename: %s\n", fnbuff);
// Take the short date/time connection string and add the client address
// to create a filename
snprintf(&fnbuff[0], 34, "%s-%s.dat",
timebuff, inet_ntoa(client.sin_addr)
);
// We're done, close the connection
close(connfd); connfd = 0;
if((writefds[i] = open(fnbuff, O_WRONLY | O_CREAT, 0644)) == -1){
writefds[i] = 0;
fprintf(stderr, "%s: cannot open - %s\n",
exec_name, strerror(errno)
);
close(connfd);
} else {
FD_SET(connfd, &masterfds);
}
connfd = 0;
// Read from existing connections
} else {
if(writefds[i] != 0){
close(writefds[i]);
writefds[i] = 0;
}
FD_CLR(i, &masterfds);
close(i);
}
}
}
}
shutdown_clean:
if(connfd > 0) close(connfd);
if(sockfd > 0) close(sockfd);
return 0;
shutdown_error:
if(connfd > 0) close(connfd);
if(sockfd > 0) close(sockfd);
return 1;
}