-/* ClientSocket.c - NeonServ v5.2
- * Copyright (C) 2011 Philipp Kreil (pk910)
+/* ClientSocket.c - NeonServ v5.3
+ * Copyright (C) 2011-2012 Philipp Kreil (pk910)
*
* This program is free software: you can redistribute it and/or modify
* it under the terms of the GNU General Public License as published by
#include "HandleInfoHandler.h"
#include "ssl.h"
#include "ConfigParser.h"
+#include "version.h"
struct socket_list {
struct ClientSocket *data;
return client;
}
+#ifndef WIN32
int connect_socket(struct ClientSocket *client) {
if((client->flags & SOCKET_FLAG_CONNECTED)) return 1;
int sock;
if(client->flags & SOCKET_FLAG_SSL) {
ssl_connect(client);
- }
+ client->flags |= SOCKET_FLAG_HAVE_SSL;
+ } else
+ client->flags &= ~SOCKET_FLAG_HAVE_SSL;
//send the IRC Headers
char sendBuf[512];
return 1;
}
+#else
+int connect_socket(struct ClientSocket *client) {
+ if((client->flags & SOCKET_FLAG_CONNECTED)) return 1;
+ struct hostent *host;
+ struct sockaddr_in addr;
+ int sock;
+ addr.sin_addr.s_addr = inet_addr(client->host);
+ if (addr.sin_addr.s_addr == INADDR_NONE) {
+ host = gethostbyname(client->host);
+ if(!host) {
+ return SOCKET_ERROR;
+ }
+ memcpy(&(addr.sin_addr), host->h_addr_list[0], 4);
+ }
+ sock = socket(PF_INET, SOCK_STREAM, 0);
+ if (sock == -1)
+ {
+ perror("socket() failed");
+ return 0;
+ }
+
+ addr.sin_port = htons(client->port);
+ addr.sin_family = AF_INET;
+
+ if (connect(sock, (struct sockaddr*)&addr, sizeof(addr)) == -1)
+ {
+ perror("connect() failed");
+ return 0;
+ }
+
+ client->sock = sock;
+ client->flags |= SOCKET_FLAG_CONNECTED | SOCKET_FLAG_RECONNECT;
+ client->connection_time = time(0);
+
+
+ if(client->flags & SOCKET_FLAG_SSL) {
+ ssl_connect(client);
+ client->flags |= SOCKET_FLAG_HAVE_SSL;
+ } else
+ client->flags &= ~SOCKET_FLAG_HAVE_SSL;
+
+ //send the IRC Headers
+ char sendBuf[512];
+ int len;
+
+ if(client->pass && strcmp(client->pass, "")) {
+ len = sprintf(sendBuf, "PASS :%s\n", client->pass);
+ write_socket(client, sendBuf, len);
+ }
+ len = sprintf(sendBuf, "USER %s 0 0 :%s\n", client->ident, client->realname);
+ write_socket(client, sendBuf, len);
+ len = sprintf(sendBuf, "NICK %s\n", client->nick);
+ write_socket(client, sendBuf, len);
+
+ return 1;
+}
+#endif
int close_socket(struct ClientSocket *client) {
if(client == NULL) return 0;
- if((client->flags & SOCKET_FLAG_CONNECTED))
+ if((client->flags & SOCKET_FLAG_CONNECTED)) {
+ char quitbuf[MAXLEN];
+ int quitlen = sprintf(quitbuf, "QUIT :[NeonServ %s.%d] disconnect requested.\n", NEONSERV_VERSION, patchlevel);
+ write_socket_force(client, quitbuf, quitlen);
+ }
+ client->flags &= ~(SOCKET_FLAG_READY | SOCKET_FLAG_RECONNECT);
+ client->flags |= SOCKET_FLAG_QUITTED | SOCKET_FLAG_DEAD;
+ return 1;
+}
+
+int disconnect_socket(struct ClientSocket *client) {
+ if(client == NULL) return 0;
+ if((client->flags & SOCKET_FLAG_CONNECTED)) {
+ char quitbuf[MAXLEN];
+ int quitlen = sprintf(quitbuf, "QUIT :[NeonServ %s.%d] disconnect requested.\n", NEONSERV_VERSION, patchlevel);
+ write_socket_force(client, quitbuf, quitlen);
+ }
+ client->flags &= ~(SOCKET_FLAG_READY | SOCKET_FLAG_RECONNECT);
+ client->flags |= SOCKET_FLAG_QUITTED;
+ return 1;
+}
+
+static void destroy_socket(struct ClientSocket *client, int free_socket) {
+ if((client->flags & SOCKET_FLAG_CONNECTED)) {
close(client->sock);
- if(client->flags & SOCKET_FLAG_SSL)
- ssl_disconnect(client);
- struct ClientSocket *sock, *last_sock = NULL;
- for (sock = sockets->data; sock; sock = sock->next) {
- if(sock == client) {
- if(last_sock)
- last_sock->next = sock->next;
- else
- sockets->data = sock->next;
- sockets->count--;
- } else
- last_sock = sock;
+ bot_disconnect(client);
}
+ if(client->flags & SOCKET_FLAG_HAVE_SSL)
+ ssl_disconnect(client);
if(client->queue)
queue_destroy(client);
if(client->whoqueue_first)
clear_whoqueue(client);
if(client->handleinfo_first)
clear_handleinfoqueue(client);
- free(client->host);
- if(client->bind)
- free(client->bind);
- if(client->pass)
- free(client->pass);
- free(client);
- return 1;
+ client->flags &= ~(SOCKET_FLAG_CONNECTED | SOCKET_FLAG_READY | SOCKET_FLAG_HAVE_SSL);
+ if(free_socket) {
+ struct ClientSocket *sock, *last_sock = NULL;
+ for (sock = sockets->data; sock; sock = sock->next) {
+ if(sock == client) {
+ if(last_sock)
+ last_sock->next = sock->next;
+ else
+ sockets->data = sock->next;
+ sockets->count--;
+ break;
+ } else
+ last_sock = sock;
+ }
+ free(client->host);
+ if(client->bind)
+ free(client->bind);
+ if(client->pass)
+ free(client->pass);
+ free(client);
+ } else if(client->flags & SOCKET_FLAG_FAST_JUMP) {
+ client->flags &= ~SOCKET_FLAG_FAST_JUMP;
+ connect_socket(client);
+ }
}
int write_socket_force(struct ClientSocket *client, char* msg, int len) {
printf("[send %d] %s", len, msg);
- if(!(client->flags & SOCKET_FLAG_SSL) || ssl_write(client, msg, len) == -2) {
+ if(!(client->flags & SOCKET_FLAG_HAVE_SSL) || ssl_write(client, msg, len) == -2) {
#ifdef WIN32
send(client->sock, msg, len, 0);
#else
if(sockets == NULL) return;
fd_set fds;
struct timeval timeout;
- struct ClientSocket *sock;
+ struct ClientSocket *sock, *next;
int ret = 0, bytes, i;
FD_ZERO(&fds);
timeout.tv_usec = 0;
ret = select(ret + 1, &fds, NULL, NULL, &timeout);
if(ret == 0) return;
- for (sock = sockets->data; sock; sock = sock->next) {
- if((sock->flags & SOCKET_FLAG_CONNECTED) && FD_ISSET(sock->sock, &fds)) {
+ for (sock = sockets->data; sock; sock = next) {
+ next = sock->next;
+ if((sock->flags & (SOCKET_FLAG_CONNECTED | SOCKET_FLAG_QUITTED)) == SOCKET_FLAG_CONNECTED && FD_ISSET(sock->sock, &fds)) {
if(sock->bufferpos != 0) {
- if(!(sock->flags & SOCKET_FLAG_SSL) || (bytes = ssl_read(sock, buffer, sizeof(buffer))) == -2) {
+ if(!(sock->flags & SOCKET_FLAG_HAVE_SSL) || (bytes = ssl_read(sock, buffer, sizeof(buffer))) == -2) {
#ifdef WIN32
bytes = recv(sock->sock, buffer, sizeof(buffer), 0);
#else
sock->bufferpos += i;
}
} else {
- if(!(sock->flags & SOCKET_FLAG_SSL) || (bytes = ssl_read(sock, sock->buffer, sizeof(sock->buffer))) == -2) {
+ if(!(sock->flags & SOCKET_FLAG_HAVE_SSL) || (bytes = ssl_read(sock, sock->buffer, sizeof(sock->buffer))) == -2) {
#ifdef WIN32
bytes = recv(sock->sock, sock->buffer, sizeof(sock->buffer), 0);
#else
}
if(bytes <= 0) {
//error
- sock->flags &= ~(SOCKET_FLAG_CONNECTED | SOCKET_FLAG_READY);
- bot_disconnect(sock);
- if(sock->queue)
- queue_destroy(sock);
- close(sock->sock);
- if(sock->flags & SOCKET_FLAG_SSL)
- ssl_disconnect(sock);
+ sock->flags |= SOCKET_FLAG_QUITTED;
} else {
sock->traffic_in += bytes;
int used = parse_lines(sock, sock->buffer, sock->bufferpos);
sock->bufferpos -= used;
}
}
- } else if(!(sock->flags & SOCKET_FLAG_CONNECTED) && (sock->flags & SOCKET_FLAG_RECONNECT)) {
+ } else if((sock->flags & (SOCKET_FLAG_CONNECTED | SOCKET_FLAG_RECONNECT)) == SOCKET_FLAG_RECONNECT) {
if(time(0) - sock->connection_time >= SOCKET_RECONNECT_TIME) {
connect_socket(sock);
}
}
+ if((sock->flags & SOCKET_FLAG_QUITTED)) {
+ sock->flags &= ~SOCKET_FLAG_QUITTED;
+ destroy_socket(sock, (sock->flags & SOCKET_FLAG_DEAD));
+ }
}
}
next = client->next;
if((client->flags & SOCKET_FLAG_CONNECTED))
close(client->sock);
- if(client->flags & SOCKET_FLAG_SSL)
+ if(client->flags & SOCKET_FLAG_HAVE_SSL)
ssl_disconnect(client);
if(client->queue)
queue_destroy(client);
projects / NeonServV5.git / blobdiff