diff options
Diffstat (limited to 'src/net/network.cpp')
| -rw-r--r-- | src/net/network.cpp | 177 |
1 files changed, 177 insertions, 0 deletions
diff --git a/src/net/network.cpp b/src/net/network.cpp new file mode 100644 index 00000000..c8bff09a --- /dev/null +++ b/src/net/network.cpp @@ -0,0 +1,177 @@ +/** + + The Mana World + Copyright 2004 The Mana World Development Team + + This file is part of The Mana World. + + The Mana World is free software; you can redistribute it and/or modify + it under the terms of the GNU General Public License as published by + the Free Software Foundation; either version 2 of the License, or + any later version. + + The Mana World is distributed in the hope that it will be useful, + but WITHOUT ANY WARRANTY; without even the implied warranty of + MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the + GNU General Public License for more details. + + You should have received a copy of the GNU General Public License + along with The Mana World; if not, write to the Free Software + Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA + +*/ + +#include "network.h" +#ifndef WIN32 +#include <unistd.h> +#include <fcntl.h> +#include <errno.h> +#endif + +/** Warning: buffers and other variables are shared, + so there can be only one connection active at a time */ + +int buffer_size = 65536; +char *in, *out; +int in_size, out_size; + +SOCKET sock; +SOCKADDR_IN addr; +// File descriptors attached to socket +fd_set read_socket; +fd_set write_socket; + +/** Increase size of written data */ +void WFIFOSET(int len) { + if(out_size+len>=buffer_size) + warning("Output buffer full"); + else out_size+=len; +} + +/** Convert an address from int format to string */ +char *iptostring(int address) { + short temp1, temp2; + + char *temp = (char *)malloc(sizeof(char[20])); + temp1 = LOWORD(address); + temp2 = HIWORD(address); + sprintf(temp, "%i.%i.%i.%i", LOBYTE(temp1), HIBYTE(temp1), LOBYTE(temp2), HIBYTE(temp2)); + return temp; +} + +/** Open a session with a server */ +SOCKET open_session(const char* address, short port) { + #ifdef WIN32 + WSADATA wsda; + #endif + struct hostent *server; + int ret; + + // Init WinSock and connect the socket + #ifdef WIN32 + WSAStartup(MAKEWORD(2,0), &wsda); + #endif + + sock = socket(PF_INET, SOCK_STREAM, 0); // Create socket for current session + if(sock==SOCKET_ERROR)return SOCKET_ERROR; + + addr.sin_family = AF_INET; + addr.sin_port = htons(port); + addr.sin_addr.s_addr = inet_addr(address); + if(addr.sin_addr.s_addr == INADDR_NONE){ + server = NULL; + server = gethostbyname(address); + if(server == NULL)return SOCKET_ERROR; + memcpy(&addr.sin_addr, server->h_addr_list[0], server->h_length); + } + + ret = connect(sock, (struct sockaddr *) &addr, sizeof(addr)); + if(ret == SOCKET_ERROR)return SOCKET_ERROR; + + // Init buffers + in = (char *)malloc(buffer_size); + out = (char *)malloc(buffer_size); + memset(in, '\0', buffer_size); + memset(out, '\0', buffer_size); + in_size = 0; + out_size = 0; + FD_CLR(sock, &read_socket); + FD_CLR(sock, &write_socket); + + return sock; +} + +/** Close a session */ +void close_session() { + FD_CLR(sock,&read_socket); + FD_CLR(sock,&write_socket); + closesocket(sock); + if(in!=NULL)free(in); + if(out!=NULL)free(out); + in = NULL; + out = NULL; + in_size = 0; + out_size = 0; + WSACleanup(); +} + +/** Send and receive data waiting in the buffers */ +void flush() { + int ret = 0; + void *buf = out; //-kth5 + timeval time_out; + + // Init the time_out struct to 0s so it won't block + time_out.tv_sec=0; + time_out.tv_usec=0; + + // Clear file descriptors and set them to socket + FD_ZERO(&read_socket); + FD_ZERO(&write_socket); + FD_SET(sock, &read_socket); + FD_SET(sock, &write_socket); + + // Check if socket has available data by evaluating attached file descriptors + select(FD_SETSIZE, &read_socket, &write_socket, NULL, &time_out); + + // Send data if available + if(FD_ISSET(sock, &write_socket)) { + // While there wasn't a error or sent the whole data: handles partial packet send + while((ret!=SOCKET_ERROR)&&(out_size>0)) { + ret = send(sock, (char *)buf, out_size, 0); + /*FILE *file = fopen("log.log","wb"); + + fprintf(file, "%s", out[0]); + fclose(file);*/ + // If not the whole data has been sent, empty the buffer from already sent bytes + if(ret!=SOCKET_ERROR && ret>0) { + buf = (char*)buf+ret; //-kth5 + out_size -= ret; + } + } + if(ret==SOCKET_ERROR) { + error("Socket Error"); +#ifdef WIN32 + log_int("Error", "socket_error", WSAGetLastError()); +#else + log("Error", "socket_error", "Undefined socket error"); +#endif + } + } + + // Read data, if available + if(FD_ISSET(sock, &read_socket)) { + /* There's no check for partial received packets because at this level + the app doesn't know packet length, but it will done when parsing received data */ + ret = recv(sock, in+in_size, RFIFOSPACE, 0); + if(ret==SOCKET_ERROR) { +#ifdef WIN32 + log_int("Error", "socket_error", WSAGetLastError()); +#else + log("Error", "socket_error", "Undefined socket error"); +#endif + } else RFIFOSET(ret); // Set size of available data to read + } +} + + |