Moved variable initialization to when they're actually used, if you're
going to use c99 you might as well lean into it
This commit is contained in:
parent
9e9f280de6
commit
abec347ec2
@ -6,6 +6,7 @@
|
|||||||
#include <stdlib.h>
|
#include <stdlib.h>
|
||||||
#include <limits.h>
|
#include <limits.h>
|
||||||
#include <fcntl.h>
|
#include <fcntl.h>
|
||||||
|
#include <sys/select.h>
|
||||||
#include <sys/types.h>
|
#include <sys/types.h>
|
||||||
#include <sys/socket.h>
|
#include <sys/socket.h>
|
||||||
#include <netinet/in.h>
|
#include <netinet/in.h>
|
||||||
@ -40,23 +41,8 @@ int main(int argc, char *argv[])
|
|||||||
{ NULL, 0, NULL, 0 }
|
{ NULL, 0, NULL, 0 }
|
||||||
};
|
};
|
||||||
|
|
||||||
fd_set readfds, masterfds;
|
|
||||||
|
|
||||||
int port = 10800; // Default port
|
int port = 10800; // Default port
|
||||||
uint32_t address = INADDR_ANY; // Default listen address
|
uint32_t address = INADDR_ANY; // Default listen address
|
||||||
int connflag, sockflag = 1;
|
|
||||||
int sockfd = 0, connfd = 0;
|
|
||||||
struct sockaddr_in sock, client;
|
|
||||||
socklen_t client_sz = sizeof(client);
|
|
||||||
char connbuff[BUFFERSIZE], timebuff[15], fnbuff[35];
|
|
||||||
ssize_t connbuff_sz;
|
|
||||||
|
|
||||||
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]);
|
exec_name = basename(argv[0]);
|
||||||
|
|
||||||
@ -105,7 +91,7 @@ int main(int argc, char *argv[])
|
|||||||
}
|
}
|
||||||
|
|
||||||
// Establish the socket
|
// Establish the socket
|
||||||
sockfd = socket(AF_INET, SOCK_STREAM, 0);
|
int sockfd = socket(AF_INET, SOCK_STREAM, 0);
|
||||||
if(sockfd == -1){
|
if(sockfd == -1){
|
||||||
fprintf(stderr, "%s: cannot open socket - %s\n",
|
fprintf(stderr, "%s: cannot open socket - %s\n",
|
||||||
exec_name, strerror(errno)
|
exec_name, strerror(errno)
|
||||||
@ -115,8 +101,10 @@ int main(int argc, char *argv[])
|
|||||||
|
|
||||||
// This sets REUSE on the socket so it's easily reallocated if
|
// This sets REUSE on the socket so it's easily reallocated if
|
||||||
// this program dies
|
// this program dies
|
||||||
|
int sockflag = 1;
|
||||||
setsockopt(sockfd, SOL_SOCKET, SO_REUSEADDR, &sockflag, sizeof(sockflag));
|
setsockopt(sockfd, SOL_SOCKET, SO_REUSEADDR, &sockflag, sizeof(sockflag));
|
||||||
|
|
||||||
|
struct sockaddr_in sock;
|
||||||
memset(&sock, 0, sizeof(sock));
|
memset(&sock, 0, sizeof(sock));
|
||||||
sock.sin_family = AF_INET;
|
sock.sin_family = AF_INET;
|
||||||
sock.sin_addr.s_addr = address;
|
sock.sin_addr.s_addr = address;
|
||||||
@ -138,6 +126,12 @@ int main(int argc, char *argv[])
|
|||||||
goto shutdown_error;
|
goto shutdown_error;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// Allocate space for the output fds
|
||||||
|
int writefds[FD_SETSIZE];
|
||||||
|
memset(&writefds, 0, sizeof(int) * FD_SETSIZE);
|
||||||
|
|
||||||
|
// Setup select variables
|
||||||
|
fd_set readfds, masterfds;
|
||||||
FD_ZERO(&masterfds);
|
FD_ZERO(&masterfds);
|
||||||
FD_SET(sockfd, &masterfds);
|
FD_SET(sockfd, &masterfds);
|
||||||
|
|
||||||
@ -154,7 +148,10 @@ int main(int argc, char *argv[])
|
|||||||
if(FD_ISSET(i, &readfds)){
|
if(FD_ISSET(i, &readfds)){
|
||||||
// Deal with new connections
|
// Deal with new connections
|
||||||
if(i == sockfd){
|
if(i == sockfd){
|
||||||
connfd = accept(sockfd, (struct sockaddr*)&client, &client_sz);
|
struct sockaddr_in client;
|
||||||
|
socklen_t client_sz = sizeof(client);
|
||||||
|
|
||||||
|
int connfd = accept(sockfd, (struct sockaddr*)&client, &client_sz);
|
||||||
if(connfd == -1){
|
if(connfd == -1){
|
||||||
fprintf(stderr, "%s: accept failed - %s\n",
|
fprintf(stderr, "%s: accept failed - %s\n",
|
||||||
exec_name, strerror(errno)
|
exec_name, strerror(errno)
|
||||||
@ -163,6 +160,7 @@ int main(int argc, char *argv[])
|
|||||||
}
|
}
|
||||||
|
|
||||||
// Get current flags before setting nonblock
|
// Get current flags before setting nonblock
|
||||||
|
int connflag;
|
||||||
if((connflag = fcntl(connfd, F_GETFL)) == -1){
|
if((connflag = fcntl(connfd, F_GETFL)) == -1){
|
||||||
fprintf(stderr, "%s: cannot get connection flags - %s\n",
|
fprintf(stderr, "%s: cannot get connection flags - %s\n",
|
||||||
exec_name, strerror(errno)
|
exec_name, strerror(errno)
|
||||||
@ -182,13 +180,14 @@ int main(int argc, char *argv[])
|
|||||||
|
|
||||||
// Collect the time of the connection, process it into
|
// Collect the time of the connection, process it into
|
||||||
// a short string
|
// a short string
|
||||||
time(&timeraw);
|
time_t timeraw = time(NULL);
|
||||||
timeptr = localtime(&timeraw);
|
struct tm* timeptr = localtime(&timeraw);
|
||||||
|
char timebuff[15];
|
||||||
strftime(&timebuff[0], 15, "%Y%m%d%H%M%S", timeptr);
|
strftime(&timebuff[0], 15, "%Y%m%d%H%M%S", timeptr);
|
||||||
timeptr = NULL;
|
|
||||||
|
|
||||||
// Take the short date/time connection string and add the client address
|
// Take the short date/time connection string and add the client address
|
||||||
// to create a filename
|
// to create a filename
|
||||||
|
char fnbuff[35];
|
||||||
snprintf(&fnbuff[0], 34, "%s-%s.dat",
|
snprintf(&fnbuff[0], 34, "%s-%s.dat",
|
||||||
timebuff, inet_ntoa(client.sin_addr)
|
timebuff, inet_ntoa(client.sin_addr)
|
||||||
);
|
);
|
||||||
@ -208,7 +207,8 @@ int main(int argc, char *argv[])
|
|||||||
// Read from existing connections
|
// Read from existing connections
|
||||||
} else {
|
} else {
|
||||||
// Do the read
|
// Do the read
|
||||||
connbuff_sz = recv(i, connbuff, BUFFERSIZE, 0);
|
char connbuff[BUFFERSIZE];
|
||||||
|
ssize_t connbuff_sz = recv(i, connbuff, BUFFERSIZE, 0);
|
||||||
if(connbuff_sz > 0){
|
if(connbuff_sz > 0){
|
||||||
// If the read fails, throw an error, and set the
|
// If the read fails, throw an error, and set the
|
||||||
// buffer size to 0, which forces a disconnect.
|
// buffer size to 0, which forces a disconnect.
|
||||||
|
Loading…
Reference in New Issue
Block a user