+++ /dev/null
-// Some macros to control options
-#define LISTENING_PORT 6667
-#define LISTENING_BACKLOG 10
-#define CLIENT_SOCKETS 30
-#define FD_SETSIZE 1024
-#define SUPPORT_IPV4
-#define SUPPORT_IPV6
-
-#include <stdio.h>
-#include <stdlib.h>
-#include <string.h>
-#include <errno.h>
-#if defined(_WIN32)
-#pragma message("Warning: Windows detected.")
-//#include <WinSock2.h>
-//#include <WS2tcpip.h>
-//#include <Windows.h>
-#include <winsock2.h>
-#include <ws2tcpip.h>
-#include <windows.h>
-#pragma message("IMPORTANT: COMPILE WITH -lws2_32")
-#pragma comment(lib, "ws2_32.lib")
-#define PRISocket "llu"
-#else
-#include <unistd.h>
-#include <netinet/in.h>
-#include <sys/socket.h>
-#include <sys/time.h>
-#include <sys/select.h>
-#include <arpa/inet.h>
-#include <netdb.h>
-#include <fcntl.h>
-//typedef unsigned long long SOCKET;
-typedef int SOCKET;
-#define PRISocket "d"
-#endif
-
-struct client_data {
- SOCKET FD;
- char type;
- char nick[32];
- size_t nickLen;
-};
-
-struct client_data client_socket[CLIENT_SOCKETS];
-
-#if defined(_WIN32)
-#define net_errno() (WSAGetLastError())
-const char *net_strerror(int e) {
- static char msg[256];
- if (!FormatMessageA(\
- FORMAT_MESSAGE_FROM_SYSTEM | FORMAT_MESSAGE_IGNORE_INSERTS,
- NULL, e, 0, msg, 256, NULL))
- sprintf(msg, "WSA Error %d", e);
- return msg;
-}
-int net_nonblock(SOCKET fd) {
- u_long v = 1;
- return ioctlsocket(fd, FIONBIO, &v);
-}
-#else
-#define net_errno() ((int)errno)
-#define net_strerror(e) strerror((e))
-#define closesocket close
-#define INVALID_SOCKET -1
-int net_nonblock(SOCKET fd) {
- int f = fcntl(fd, F_GETFL);
- if (f == -1)
- return -1;
- return fcntl(fd, F_SETFL, f | O_NONBLOCK);
-}
-#endif
-void net_perror(const char *p)
-{
- if (p && *p)
- fprintf(stderr, "%s: %s\n", p, net_strerror(net_errno()));
- else
- fprintf(stderr, "%s\n", net_strerror(net_errno()));
-}
-
-size_t send_log(int socket, const void *buffer, size_t length, int flags) {
- printf("SERVER: %s", (char*)buffer);
- fflush(stdout);
- return send(socket, buffer, length, flags);
-}
-
-void sendall_log(int socket, const void *buffer, size_t length, int flags) {
- int i;
- for(i=0;i<CLIENT_SOCKETS;i++) {
- if(client_socket[i].FD == socket || client_socket[i].FD == 0) {
- continue;
- } else {
- printf("SERVER: %"PRISocket" %s", client_socket[i].FD, (char*)buffer);
- fflush(stdout);
- send(client_socket[i].FD, buffer, length, flags);
- }
- }
-}
-
-// IRC STUFF COPY THIS WHEN WE REFINE ALL OF THIS CODE INTO THE NEW SERVER
-
-void processPacket(struct client_data *client, char *packet, size_t packetLen);
-void newUser(struct client_data *client);
-void sendMOTD(struct client_data *client);
-void setNick(struct client_data *client, char* nickname, size_t size);
-
-// TODO: check for newlines are process each one as a different packet
-void processPacket(struct client_data *client, char *packet, size_t packetLen) {
- if(strncmp(packet, "PING", 4) == 0) {
- packet[1] = 'O';
- send_log(client_socket->FD, packet, packetLen, 0);
- } else if(strncmp(packet, "NICK ", 5) == 0) {
- // User now registered
- setNick(client, packet+5, 32);
- printf("USER SET NICKNAME %lu %s\n", client->nickLen, client->nick);
- fflush(stdout);
- } else if(strncmp(packet, "USER ", 5) == 0) {
- // Send welcome
- newUser(client);
- } else if(strncmp(packet, "MOTD", 4) == 0) {
- // Send welcome
- sendMOTD(client);
- // Horrible join but it works and its funny
- } else if(strncmp(packet, "JOIN ", 5) == 0) {
- char data[256];
- data[0] = ':';
- strcpy(data+1, client->nick);
- strcpy(data+1+client->nickLen, " JOIN :");
- strcpy(data+8+client->nickLen, packet+5);
- send_log(client->FD, data, strlen(data), 0);
- sendall_log(client->FD, data, strlen(data), 0);
- } else if(strncmp(packet, "PRIVMSG ", 8) == 0) {
- char data[256];
- data[0] = ':';
- strcpy(data+1, client->nick);
- data[1+client->nickLen] = ' ';
- strcpy(data+2+client->nickLen, packet);
- sendall_log(client->FD, data, strlen(data), 0);
- }
-
-}
-
-void newUser(struct client_data *client) {
- char data[1024];
- // Copy in basics
- strcpy(data, ":irc.patchysicp.xyz 001 ");
- strcpy(data+24, client->nick);
- strcpy(data+24+client->nickLen, " :Welcome to PatchyIRCD!!!\n");
- //send_log(client->FD, data, (26+client->nickLen+27), 0);
- send_log(client->FD, data, strlen(data), 0);
- data[22] = '2';
- strcpy(data+26+client->nickLen, "Your host is PatchyIRCD V0.0.0\n");
- send_log(client->FD, data, strlen(data), 0);
- data[22] = '3';
- strcpy(data+26+client->nickLen, "This server was created when I had cybersex with your mother.\n");
- send_log(client->FD, data, strlen(data), 0);
- data[20] = '2';
- data[21] = '5';
- data[22] = '1';
- strcpy(data+26+client->nickLen, "This server exists\n");
- send_log(client->FD, data, strlen(data), 0);
- data[22] = '2';
- strcpy(data+26+client->nickLen, "Jannies don't exist\n");
- send_log(client->FD, data, strlen(data), 0);
- sendMOTD(client);
-}
-
-void sendMOTD(struct client_data *client) {
- char data[1024];
- // Copy in basics
- strcpy(data, ":irc.patchysicp.xyz 375 ");
- strcpy(data+24, client->nick);
- strcpy(data+24+client->nickLen, " :PatchySICP /MOTD\n");
- send_log(client->FD, data, strlen(data), 0);
- data[22] = '2';
- strcpy(data+26+client->nickLen, "Welcome to Montreal!\n");
- send_log(client->FD, data, strlen(data), 0);
- strcpy(data+26+client->nickLen, "You are retarded and I will have cybersex with your mom.\n");
- send_log(client->FD, data, strlen(data), 0);
- data[22] = '6';
- strcpy(data+26+client->nickLen, "/MOTD over\n");
- send_log(client->FD, data, strlen(data), 0);
-}
-
-void setNick(struct client_data *client, char* nickname, size_t size) {
- // TODO: perform basic sanitization
- strncpy(client->nick, nickname, size);
- // Null term
- client->nick[size-1] = 0;
- // Remove newline
- int i;
- for(i=0;i<size;i++){
- // CR/^M
- if(client->nick[i] == 13) {
- client->nick[i] = 0;
- break;
- }
- // Newline
- if(client->nick[i] == 10) {
- client->nick[i] = 0;
- break;
- }
- }
- // Set length
- client->nickLen = strlen(client->nick);
- return;
-}
-
-int main(){
- fd_set readfds;
- SOCKET highestFD;
- int i;
- SOCKET SD;
- SOCKET serverSocketV4 = 0;
- SOCKET serverSocketV6 = 0;
- char client_addr[100];
- char *welcomeMessage = ":irc.patchysicp.xyz NOTICE * :Hello welcome to PatchyIRCD!\n";
- char buffer[1024];
- size_t bytesRead;
-
-#if defined(_WIN32)
- struct WSAData wd;
- if (WSAStartup(MAKEWORD(2, 0), &wd)) printf("Winsock API startup failed.\n");
-#endif
-
- puts("Starting");
- fflush(stdout);
-
-#ifdef SUPPORT_IPV4
- struct sockaddr_in serverAddrV4;
- memset(&serverAddrV4, 0, sizeof(struct sockaddr_in));
- serverAddrV4.sin_addr.s_addr = INADDR_ANY;
- serverAddrV4.sin_port = htons(LISTENING_PORT);
- serverAddrV4.sin_family = AF_INET;
- size_t serverAddrV4len = sizeof(serverAddrV4);
-#endif
-#ifdef SUPPORT_IPV6
- struct sockaddr_in6 serverAddrV6;
- memset(&serverAddrV6, 0, sizeof(struct sockaddr_in6));
- serverAddrV6.sin6_addr = in6addr_any;
- serverAddrV6.sin6_flowinfo = 0;
- serverAddrV6.sin6_port = htons(LISTENING_PORT);
- serverAddrV6.sin6_family = AF_INET6;
- size_t serverAddrV6len = sizeof(serverAddrV6);
-#endif
-
- int opt = 1;
-#ifdef SUPPORT_IPV4
- // Create socket
- serverSocketV4 = socket(AF_INET, SOCK_STREAM, 0);
- if(serverSocketV4 == (SOCKET)INVALID_SOCKET) {
- net_perror("ERROR SOCKET IPV4");
- exit(EXIT_FAILURE);
- }
- // Set socket options
- if(setsockopt(serverSocketV4, SOL_SOCKET,\
- SO_REUSEADDR,\
- (void*)&opt, sizeof(opt)) == -1) {
- net_perror("ERROR SOCKET IPV4");
- exit(EXIT_FAILURE);
- }
- // Set nonblock
- net_nonblock(serverSocketV4);
- // Bind to socket
- if(bind(serverSocketV4, (struct sockaddr*)&serverAddrV4,\
- sizeof(struct sockaddr_in)) == -1) {
- net_perror("ERROR BIND IPV4");
- exit(EXIT_FAILURE);
- }
- // Listen to socket
- if(listen(serverSocketV4, LISTENING_BACKLOG) == -1) {
- net_perror("ERROR LISTEN IPV4");
- exit(EXIT_FAILURE);
- }
-#endif
-#ifdef SUPPORT_IPV6
- // Create socket
- serverSocketV6 = socket(AF_INET6, SOCK_STREAM, 0);
- if(serverSocketV6 == (SOCKET)INVALID_SOCKET) {
- net_perror("ERROR SOCKET IPV6");
- exit(EXIT_FAILURE);
- }
- // Set socket options
- if(setsockopt(serverSocketV6, SOL_SOCKET,\
- SO_REUSEADDR,\
- (void*)&opt, sizeof(opt)) == -1) {
- net_perror("ERROR SOCKET IPV6");
- exit(EXIT_FAILURE);
- }
- // IPV6 Only
- if(setsockopt(serverSocketV6, IPPROTO_IPV6,\
- IPV6_V6ONLY,\
- (void*)&opt, sizeof(opt)) == -1) {
- net_perror("ERROR SOCKET IPV6");
- exit(EXIT_FAILURE);
- }
- // Set nonblock
- net_nonblock(serverSocketV6);
- // Bind to socket
- if(bind(serverSocketV6, (struct sockaddr*)&serverAddrV6,\
- sizeof(struct sockaddr_in6)) == -1) {
- net_perror("ERROR BIND IPV6");
- exit(EXIT_FAILURE);
- }
- // Listen to socket
- if(listen(serverSocketV6, LISTENING_BACKLOG) == -1) {
- net_perror("ERROR LISTEN IPV6");
- exit(EXIT_FAILURE);
- }
-#endif
-
- memset(client_socket, 0, sizeof(client_socket));
-
- puts("Ready");
- fflush(stdout);
- while(1){
- // Clear all sockets
- FD_ZERO(&readfds);
- // Add main sockets
-#ifdef SUPPORT_IPV4
- FD_SET(serverSocketV4, &readfds);
-#endif
-#ifdef SUPPORT_IPV6
- FD_SET(serverSocketV6, &readfds);
-#endif
- if(serverSocketV4 < serverSocketV6) highestFD = serverSocketV6;
- else highestFD = serverSocketV4;
-
- // Check for new sockets
- for(i=0;i<CLIENT_SOCKETS;i++) {
- SD = client_socket[i].FD;
- // Valid FD
- if(SD>0){
- FD_SET(SD, &readfds);
- // Update highest FD
- if(SD>highestFD){
- highestFD = SD;
- }
- }
- }
-
- // Wait until activity on sockets
- if(select(highestFD+1,&readfds,NULL,NULL,NULL) == -1) {
- net_perror("ERROR SELECT");
- }
-
-#ifdef SUPPORT_IPV4
- if (FD_ISSET(serverSocketV4, &readfds)) {
- SOCKET newSocket = accept(serverSocketV4, NULL, NULL);
- if(newSocket == (SOCKET)INVALID_SOCKET){
- net_perror("ERROR newSocket IPV4");
- }
- getpeername(newSocket,(struct sockaddr*)&serverAddrV4,(socklen_t*)&serverAddrV4len);
- inet_ntop(AF_INET, &(serverAddrV4.sin_addr), client_addr, sizeof(client_addr));
- printf("Connection accepted from Socket: %"PRISocket", IP: %s PORT: %d\n", newSocket, client_addr, ntohs(serverAddrV4.sin_port));
- fflush(stdout);
- send_log(newSocket, welcomeMessage, strlen(welcomeMessage), 0);
- // Add to list of sockets
- for(i=0;i<CLIENT_SOCKETS;i++) {
- if(client_socket[i].FD==0){
- client_socket[i].FD = newSocket;
- client_socket[i].type = 4;
- // Set nonblock
- net_nonblock(client_socket[i].FD);
- break;
- }
- }
- }
-#endif
-#ifdef SUPPORT_IPV6
- if (FD_ISSET(serverSocketV6, &readfds)) {
- SOCKET newSocket = accept(serverSocketV6, NULL, NULL);
- if(newSocket == (SOCKET)INVALID_SOCKET){
- net_perror("ERROR newSocket IPV6");
- }
- getpeername(newSocket,(struct sockaddr*)&serverAddrV6,(socklen_t*)&serverAddrV6len);
- inet_ntop(AF_INET6, &(serverAddrV6.sin6_addr), client_addr, sizeof(client_addr));
- printf("Connection accepted from Socket: %"PRISocket", IP: %s PORT: %d\n", newSocket, client_addr, ntohs(serverAddrV6.sin6_port));
- fflush(stdout);
- send_log(newSocket, welcomeMessage, strlen(welcomeMessage), 0);
- // Add to list of sockets
- for(i=0;i<CLIENT_SOCKETS;i++) {
- if(client_socket[i].FD==0){
- client_socket[i].FD = newSocket;
- client_socket[i].type = 6;
- // Set nonblock
- net_nonblock(client_socket[i].FD);
- break;
- }
- }
- }
-#endif
- // Respond to already connected clients
- for(i=0;i<CLIENT_SOCKETS;i++) {
- SD = client_socket[i].FD;
- if(FD_ISSET(SD, &readfds)) {
- //bytesRead = read(SD, buffer, sizeof(buffer));
- bytesRead = recv(SD,buffer,sizeof(buffer), 0);
- // Disconnect
- if(bytesRead == 0) {
-#ifdef SUPPORT_IPV4
- if(client_socket[i].type == 4) {
- getpeername(SD,(struct sockaddr*)&serverAddrV4,(socklen_t*)&serverAddrV4len);
- inet_ntop(AF_INET, &(serverAddrV4.sin_addr), client_addr, sizeof(client_addr));
- printf("Client disconnected from Socket: %"PRISocket", IP: %s PORT: %d\n", SD, client_addr, ntohs(serverAddrV4.sin_port));
- fflush(stdout);
- }
-#endif
-#ifdef SUPPORT_IPV6
- if(client_socket[i].type == 6) {
- getpeername(SD,(struct sockaddr*)&serverAddrV6,(socklen_t*)&serverAddrV6len);
- inet_ntop(AF_INET6, &(serverAddrV6.sin6_addr), client_addr, sizeof(client_addr));
- printf("Client disconnected from Socket: %"PRISocket", IP: %s PORT: %d\n", SD, client_addr, ntohs(serverAddrV6.sin6_port));
- fflush(stdout);
- }
-#endif
- closesocket(SD);
- client_socket[i].FD = 0;
- } else {
- // Add null byte
- buffer[bytesRead] = '\0';
- // Send it back
- //send(SD,buffer,bytesRead,0);
- printf("CLIENT: %s", buffer);
- fflush(stdout);
-
- int j;
- char *buffer2 = buffer;
- char tmp;
- for(j=0;j<bytesRead;j++){
- if(buffer[j] == '\n'){
- char tmp = buffer[j+1];
- buffer[j+1] = 0;
- //Send to processing
- processPacket(&(client_socket[i]), buffer2, strlen(buffer2));
- buffer[j+1] = tmp;
- buffer2 = &(buffer[j+1]);
- }
- }
- }
- }
- }
- }
-
-#if defined(_WIN32)
- WSACleanup();
-#endif
- for(i=0;i<CLIENT_SOCKETS;i++) {
- if(client_socket[i].FD>0) {
- closesocket(client_socket[i].FD);
- }
- }
- return 0;
-}
-
+++ /dev/null
-// Some macros to control options
-#define LISTENING_PORT 4444
-#define LISTENING_BACKLOG 10
-#define CLIENT_SOCKETS 100
-#define FD_SETSIZE 1024
-#define SUPPORT_IPV4
-#define SUPPORT_IPV6
-
-#include <stdio.h>
-#include <stdlib.h>
-#include <string.h>
-#include <errno.h>
-#if defined(_WIN32)
-#pragma message("Warning: Windows detected.")
-//#include <WinSock2.h>
-//#include <WS2tcpip.h>
-//#include <Windows.h>
-#include <winsock2.h>
-#include <ws2tcpip.h>
-#include <windows.h>
-#pragma message("IMPORTANT: COMPILE WITH -lws2_32")
-#pragma comment(lib, "ws2_32.lib")
-#define PRISocket "llu"
-#else
-#include <unistd.h>
-#include <netinet/in.h>
-#include <sys/socket.h>
-#include <sys/time.h>
-#include <sys/select.h>
-#include <arpa/inet.h>
-#include <netdb.h>
-#include <fcntl.h>
-//typedef unsigned long long SOCKET;
-typedef int SOCKET;
-#define PRISocket "d"
-#endif
-
-struct client_data {
- SOCKET FD;
- char type;
-};
-
-struct client_data client_socket[CLIENT_SOCKETS];
-
-#if defined(_WIN32)
-#define net_errno() (WSAGetLastError())
-const char *net_strerror(int e) {
- static char msg[256];
- if (!FormatMessageA(\
- FORMAT_MESSAGE_FROM_SYSTEM | FORMAT_MESSAGE_IGNORE_INSERTS,
- NULL, e, 0, msg, 256, NULL))
- sprintf(msg, "WSA Error %d", e);
- return msg;
-}
-int net_nonblock(SOCKET fd) {
- u_long v = 1;
- return ioctlsocket(fd, FIONBIO, &v);
-}
-#else
-#define net_errno() ((int)errno)
-#define net_strerror(e) strerror((e))
-#define closesocket close
-#define INVALID_SOCKET -1
-int net_nonblock(SOCKET fd) {
- int f = fcntl(fd, F_GETFL);
- if (f == -1)
- return -1;
- return fcntl(fd, F_SETFL, f | O_NONBLOCK);
-}
-#endif
-void net_perror(const char *p)
-{
- if (p && *p)
- fprintf(stderr, "%s: %s\n", p, net_strerror(net_errno()));
- else
- fprintf(stderr, "%s\n", net_strerror(net_errno()));
-}
-
-int main(){
- fd_set readfds;
- SOCKET highestFD;
- int i;
- SOCKET SD;
- SOCKET serverSocketV4 = 0;
- SOCKET serverSocketV6 = 0;
- char client_addr[100];
- char *welcomeMessage = "Hello welcome to PatchyIRCD\n";
- char buffer[1024];
- size_t bytesRead;
-
-#if defined(_WIN32)
- struct WSAData wd;
- if (WSAStartup(MAKEWORD(2, 0), &wd)) printf("Winsock API startup failed.\n");
-#endif
-
- puts("Starting");
-
-#ifdef SUPPORT_IPV4
- struct sockaddr_in serverAddrV4;
- memset(&serverAddrV4, 0, sizeof(struct sockaddr_in));
- serverAddrV4.sin_addr.s_addr = INADDR_ANY;
- serverAddrV4.sin_port = htons(LISTENING_PORT);
- serverAddrV4.sin_family = AF_INET;
- size_t serverAddrV4len = sizeof(serverAddrV4);
-#endif
-#ifdef SUPPORT_IPV6
- struct sockaddr_in6 serverAddrV6;
- memset(&serverAddrV6, 0, sizeof(struct sockaddr_in6));
- serverAddrV6.sin6_addr = in6addr_any;
- serverAddrV6.sin6_flowinfo = 0;
- serverAddrV6.sin6_port = htons(LISTENING_PORT);
- serverAddrV6.sin6_family = AF_INET6;
- size_t serverAddrV6len = sizeof(serverAddrV6);
-#endif
-
- int opt = 1;
-#ifdef SUPPORT_IPV4
- // Create socket
- serverSocketV4 = socket(AF_INET, SOCK_STREAM, 0);
- if(serverSocketV4 == (SOCKET)INVALID_SOCKET) {
- net_perror("ERROR SOCKET IPV4");
- exit(EXIT_FAILURE);
- }
- // Set socket options
- if(setsockopt(serverSocketV4, SOL_SOCKET,\
- SO_REUSEADDR,\
- (void*)&opt, sizeof(opt)) == -1) {
- net_perror("ERROR SOCKET IPV4");
- exit(EXIT_FAILURE);
- }
- // Set nonblock
- net_nonblock(serverSocketV4);
- // Bind to socket
- if(bind(serverSocketV4, (struct sockaddr*)&serverAddrV4,\
- sizeof(struct sockaddr_in)) == -1) {
- net_perror("ERROR BIND IPV4");
- exit(EXIT_FAILURE);
- }
- // Listen to socket
- if(listen(serverSocketV4, LISTENING_BACKLOG) == -1) {
- net_perror("ERROR LISTEN IPV4");
- exit(EXIT_FAILURE);
- }
-#endif
-#ifdef SUPPORT_IPV6
- // Create socket
- serverSocketV6 = socket(AF_INET6, SOCK_STREAM, 0);
- if(serverSocketV6 == (SOCKET)INVALID_SOCKET) {
- net_perror("ERROR SOCKET IPV6");
- exit(EXIT_FAILURE);
- }
- // Set socket options
- if(setsockopt(serverSocketV6, SOL_SOCKET,\
- SO_REUSEADDR,\
- (void*)&opt, sizeof(opt)) == -1) {
- net_perror("ERROR SOCKET IPV6");
- exit(EXIT_FAILURE);
- }
- // IPV6 Only
- if(setsockopt(serverSocketV6, IPPROTO_IPV6,\
- IPV6_V6ONLY,\
- (void*)&opt, sizeof(opt)) == -1) {
- net_perror("ERROR SOCKET IPV6");
- exit(EXIT_FAILURE);
- }
- // Set nonblock
- net_nonblock(serverSocketV6);
- // Bind to socket
- if(bind(serverSocketV6, (struct sockaddr*)&serverAddrV6,\
- sizeof(struct sockaddr_in6)) == -1) {
- net_perror("ERROR BIND IPV6");
- exit(EXIT_FAILURE);
- }
- // Listen to socket
- if(listen(serverSocketV6, LISTENING_BACKLOG) == -1) {
- net_perror("ERROR LISTEN IPV6");
- exit(EXIT_FAILURE);
- }
-#endif
-
- memset(client_socket, 0, sizeof(client_socket));
-
- puts("Ready");
- while(1){
- // Clear all sockets
- FD_ZERO(&readfds);
- // Add main sockets
-#ifdef SUPPORT_IPV4
- FD_SET(serverSocketV4, &readfds);
-#endif
-#ifdef SUPPORT_IPV6
- FD_SET(serverSocketV6, &readfds);
-#endif
- if(serverSocketV4 < serverSocketV6) highestFD = serverSocketV6;
- else highestFD = serverSocketV4;
-
- // Check for new sockets
- for(i=0;i<CLIENT_SOCKETS;i++) {
- SD = client_socket[i].FD;
- // Valid FD
- if(SD>0){
- FD_SET(SD, &readfds);
- // Update highest FD
- if(SD>highestFD){
- highestFD = SD;
- }
- }
- }
-
- // Wait until activity on sockets
- if(select(highestFD+1,&readfds,NULL,NULL,NULL) == -1) {
- net_perror("ERROR SELECT");
- }
-
-#ifdef SUPPORT_IPV4
- if (FD_ISSET(serverSocketV4, &readfds)) {
- SOCKET newSocket = accept(serverSocketV4, NULL, NULL);
- if(newSocket == (SOCKET)INVALID_SOCKET){
- net_perror("ERROR newSocket IPV4");
- }
- getpeername(newSocket,(struct sockaddr*)&serverAddrV4,(socklen_t*)&serverAddrV4len);
- inet_ntop(AF_INET, &(serverAddrV4.sin_addr), client_addr, sizeof(client_addr));
- printf("Connection accepted from Socket: %"PRISocket", IP: %s PORT: %d\n", newSocket, client_addr, ntohs(serverAddrV4.sin_port));
- send(newSocket, welcomeMessage, strlen(welcomeMessage), 0);
- // Add to list of sockets
- for(i=0;i<CLIENT_SOCKETS;i++) {
- if(client_socket[i].FD==0){
- client_socket[i].FD = newSocket;
- client_socket[i].type = 4;
- // Set nonblock
- net_nonblock(client_socket[i].FD);
- break;
- }
- }
- }
-#endif
-#ifdef SUPPORT_IPV6
- if (FD_ISSET(serverSocketV6, &readfds)) {
- SOCKET newSocket = accept(serverSocketV6, NULL, NULL);
- if(newSocket == (SOCKET)INVALID_SOCKET){
- net_perror("ERROR newSocket IPV6");
- }
- getpeername(newSocket,(struct sockaddr*)&serverAddrV6,(socklen_t*)&serverAddrV6len);
- inet_ntop(AF_INET6, &(serverAddrV6.sin6_addr), client_addr, sizeof(client_addr));
- printf("Connection accepted from Socket: %"PRISocket", IP: %s PORT: %d\n", newSocket, client_addr, ntohs(serverAddrV6.sin6_port));
- send(newSocket, welcomeMessage, strlen(welcomeMessage), 0);
- // Add to list of sockets
- for(i=0;i<CLIENT_SOCKETS;i++) {
- if(client_socket[i].FD==0){
- client_socket[i].FD = newSocket;
- client_socket[i].type = 6;
- // Set nonblock
- net_nonblock(client_socket[i].FD);
- break;
- }
- }
- }
-#endif
- // Respond to already connected clients
- for(i=0;i<CLIENT_SOCKETS;i++) {
- SD = client_socket[i].FD;
- if(FD_ISSET(SD, &readfds)) {
- //bytesRead = read(SD, buffer, sizeof(buffer));
- bytesRead = recv(SD,buffer,sizeof(buffer), 0);
- // Disconnect
- if(bytesRead == 0) {
-#ifdef SUPPORT_IPV4
- if(client_socket[i].type == 4) {
- getpeername(SD,(struct sockaddr*)&serverAddrV4,(socklen_t*)&serverAddrV4len);
- inet_ntop(AF_INET, &(serverAddrV4.sin_addr), client_addr, sizeof(client_addr));
- printf("Client disconnected from Socket: %"PRISocket", IP: %s PORT: %d\n", SD, client_addr, ntohs(serverAddrV4.sin_port));
- }
-#endif
-#ifdef SUPPORT_IPV6
- if(client_socket[i].type == 6) {
- getpeername(SD,(struct sockaddr*)&serverAddrV6,(socklen_t*)&serverAddrV6len);
- inet_ntop(AF_INET6, &(serverAddrV6.sin6_addr), client_addr, sizeof(client_addr));
- printf("Client disconnected from Socket: %"PRISocket", IP: %s PORT: %d\n", SD, client_addr, ntohs(serverAddrV6.sin6_port));
- }
-#endif
- closesocket(SD);
- client_socket[i].FD = 0;
- } else {
- // Add null byte
- buffer[bytesRead] = '\0';
- // Send it back
- send(SD,buffer,bytesRead,0);
- }
- }
- }
- }
-
-#if defined(_WIN32)
- WSACleanup();
-#endif
- for(i=0;i<CLIENT_SOCKETS;i++) {
- if(client_socket[i].FD>0) {
- closesocket(client_socket[i].FD);
- }
- }
- return 0;
-}
-
projects / patchyircd.git / commitdiff