diff options
| author | Paul Buetow <paul@buetow.org> | 2013-04-06 13:14:41 +0200 |
|---|---|---|
| committer | Paul Buetow <paul@buetow.org> | 2013-04-06 13:14:41 +0200 |
| commit | 9cd3ccffd5372dfde3af478e3f832f18db4be3f1 (patch) | |
| tree | 631c295a4a4a16b57502b847626763a279bf6df7 /ychat-0.7.8/src/sock | |
| parent | 13aaf70af703748fe096e0664c305cd202637ad2 (diff) | |
tagging tags
Diffstat (limited to 'ychat-0.7.8/src/sock')
| -rw-r--r-- | ychat-0.7.8/src/sock/CVS/Entries | 3 | ||||
| -rw-r--r-- | ychat-0.7.8/src/sock/CVS/Repository | 1 | ||||
| -rw-r--r-- | ychat-0.7.8/src/sock/CVS/Root | 1 | ||||
| -rwxr-xr-x | ychat-0.7.8/src/sock/sock.cpp | 329 | ||||
| -rwxr-xr-x | ychat-0.7.8/src/sock/sock.h | 74 |
5 files changed, 408 insertions, 0 deletions
diff --git a/ychat-0.7.8/src/sock/CVS/Entries b/ychat-0.7.8/src/sock/CVS/Entries new file mode 100644 index 0000000..0d7b592 --- /dev/null +++ b/ychat-0.7.8/src/sock/CVS/Entries @@ -0,0 +1,3 @@ +/sock.cpp/1.28/Tue Mar 1 09:54:29 2005// +/sock.h/1.14/Sun Feb 27 06:28:53 2005// +D diff --git a/ychat-0.7.8/src/sock/CVS/Repository b/ychat-0.7.8/src/sock/CVS/Repository new file mode 100644 index 0000000..bbe6e49 --- /dev/null +++ b/ychat-0.7.8/src/sock/CVS/Repository @@ -0,0 +1 @@ +ychat/src/sock diff --git a/ychat-0.7.8/src/sock/CVS/Root b/ychat-0.7.8/src/sock/CVS/Root new file mode 100644 index 0000000..745de68 --- /dev/null +++ b/ychat-0.7.8/src/sock/CVS/Root @@ -0,0 +1 @@ +:pserver:anonymous@buetow.org:/usr/home/cvs/cvsroot diff --git a/ychat-0.7.8/src/sock/sock.cpp b/ychat-0.7.8/src/sock/sock.cpp new file mode 100755 index 0000000..5f6ede3 --- /dev/null +++ b/ychat-0.7.8/src/sock/sock.cpp @@ -0,0 +1,329 @@ +#ifndef SOCK_CPP +#define SOCK_CPP + +#include <arpa/inet.h> +#include <errno.h> +#include <sys/types.h> +#include <unistd.h> +#include "sock.h" +#include "../chat/chat.h" +#include "../chat/user.h" + +using namespace std; + +sock::sock() +{ + this->b_run = true; + this->i_req = 0; + this->req_parser = new reqp(); +#ifdef LOGGING + + this->log_daemon = new logd( wrap::CONF->get_elem( "httpd.logging.accessfile" ), + + wrap::CONF->get_elem( "httpd.logging.accesslines" ) ); +#endif +} + +sock::~sock() +{} + +//<<* +void +sock::chat_stream( int i_sock, user *p_user, map<string,string> &map_params ) +{ + string s_msg( "\n" ); + + for ( int i = 0; i < PUSHSTR; i++ ) + send( i_sock, s_msg.c_str(), s_msg.size(), 0 ); + + pthread_mutex_t mutex; + pthread_mutex_init( &mutex, NULL ); + pthread_mutex_lock( &mutex ); + + do + { + s_msg = p_user->get_mess( ); + + if ( 0 > send( i_sock, s_msg.c_str(), s_msg.size(), 0 ) ) + { + p_user->set_online( false ); + break; + } + + pthread_cond_wait( &(p_user->cond_message), &mutex ); + } + while( p_user->get_online() ); + + pthread_mutex_destroy( &mutex ); + + // if there is still a message to send: + s_msg = p_user->get_mess( ); + + if ( ! s_msg.empty() ) + send( i_sock, s_msg.c_str(), s_msg.size(), 0 ); + + // remove the user from its room. + string s_user( p_user->get_name() ); + string s_user_lowercase( p_user->get_lowercase_name() ); + + p_user->get_room()->del_elem( s_user_lowercase ); + + // post the room that the user has left the chat. + s_msg = wrap::TIMR->get_time() + " " + + p_user->get_colored_bold_name() + + wrap::CONF->get_elem( "chat.msgs.userleaveschat" ) + + "<br>\n"; + + p_user->get_room()->msg_post( &s_msg ); + p_user->get_room()->reload_onlineframe(); + +#ifdef VERBOSE + + cout << REMUSER << s_user << endl; +#endif + + wrap::GCOL->add_user_to_garbage( p_user ); +} +//*>> + +int +sock::make_server_socket( int i_port ) +{ + size_t i_sock; + struct sockaddr_in name; + + // create the server socket. + i_sock = socket (PF_INET, SOCK_STREAM, 0); + if (i_sock < 0) + { + wrap::system_message( SOCKERR ); + + if ( ++i_port > MAXPORT ) + exit(1); + + wrap::system_message( SOCKERR ); + + return make_server_socket( i_port ); + } + + // give the server socket a name. + name.sin_family = AF_INET; + name.sin_port = htons(i_port); + name.sin_addr.s_addr = htonl(INADDR_ANY); + int i_optval = 1; + + setsockopt( i_sock, SOL_SOCKET, SO_REUSEADDR, (char*)&i_optval, sizeof(int) ); + + if ( bind(i_sock, (struct sockaddr *) &name, sizeof (name)) < 0 ) + { + + wrap::system_message( BINDERR ); + + if ( ++i_port > MAXPORT ) + exit(1); + + wrap::system_message( string(SOCKERR) + tool::int2string(i_port) ); + + // Rerun recursive. + return make_server_socket( i_port ); + } + + wrap::system_message( SOCKCRT + string("localhost:") + tool::int2string(i_port) ); + +#ifdef NCURSES + + mvprintw( NCUR_PORT_X,NCUR_PORT_Y, "Port: %d ", i_port); + refresh(); +#endif + + return i_sock; +} + +int +sock::read_write( int* p_sock ) +{ + int i_sock = *p_sock; + char c_req[READSOCK]; + int i_bytes = read(i_sock, c_req, READSOCK); + + if (i_bytes <= 0) + { + wrap::system_message( READERR ); + } + + else + { + // stores the request params. + map<string,string> map_params; + + // get the s_rep ( s_html response which will be send imediatly to the client + struct sockaddr_in client; + size_t size = sizeof(client); + +#ifdef CYGWIN + + getpeername( i_sock, (struct sockaddr *)&client, (int*)&size); +#else + getpeername( i_sock, (struct sockaddr *)&client, &size); +#endif + + uint32_t &s_addr = client.sin_addr.s_addr; + if ( (map_params["REMOTE_ADDR"] = get_elem(s_addr)) == "" ) + { + map_params["REMOTE_ADDR"] = string(inet_ntoa(client.sin_addr)); + set_elem(map_params["REMOTE_ADDR"], s_addr); + wrap::system_message(SOCKCAC+map_params["REMOTE_ADDR"]); + } + + //map_params["REMOTE_ADDR"] = inet_ntoa_callback(&client.sin_addr); + //map_params["REMOTE_PORT"] = ntohs(client.sin_port); + + string s_rep = req_parser->parse(i_sock, string(c_req), map_params); + +#ifdef LOGGING + + log_daemon->log_access(map_params); +#endif + + // send s_rep to the client. + send(i_sock, s_rep.c_str(), s_rep.size(), 0); + + // dont need those vals anymore. + map_params.clear(); + + shutdown( i_sock, 2 ); + close ( i_sock ); + + return 0; + } + + shutdown( i_sock, 2 ); + close ( i_sock ); + + return 1; +} + +int +sock::start() +{ + wrap::system_message( SOCKSRV ); + pool* p_pool = wrap::POOL; + +#ifdef NCURSES + print_hits(); + p_pool->print_pool_size(); +#endif + + int i_port = tool::string2int( wrap::CONF->get_elem( "httpd.serverport" ) ); + int i_sock, i; + fd_set active_fd_set, read_fd_set; + struct sockaddr_in clientname; + size_t size; + + + // create the server socket and set it up to accept connections. + i_sock = make_server_socket ( i_port ); + + if (listen (i_sock, 1) < 0) + { + wrap::system_message( LISTERR ); + exit( EXIT_FAILURE ); + } + + wrap::system_message( SOCKRDY ); + + // initialize the set of active sockets. + FD_ZERO (&active_fd_set); + FD_SET (i_sock, &active_fd_set); + + while( b_run ) + { + // block until input arrives on one or more active sockets. + read_fd_set = active_fd_set; + if (select (FD_SETSIZE, &read_fd_set, NULL, NULL, NULL) < 0) + { + wrap::system_message( SELCERR ); + + exit( EXIT_FAILURE ); + } + + // service all the sockets with input pending. + for ( i = 0; i < FD_SETSIZE; i++ ) + if ( FD_ISSET (i, &read_fd_set) ) + { + if ( i == i_sock ) + { + // connection request on original socket. + ++i_req; + +#ifdef NCURSES + + print_hits(); +#endif + + int i_new_sock; + size = sizeof(clientname); +#ifdef CYGWIN + + i_new_sock = accept (i_sock, (struct sockaddr *) &clientname, (int*)&size); +#else + + i_new_sock = accept (i_sock, (struct sockaddr *) &clientname, &size); +#endif + + if (i_new_sock < 0) + { + wrap::system_message( ACCPERR ); + close(i_new_sock); + } + else + { +#ifdef VERBOSE + wrap::system_message(NEWREQU + + tool::int2string(i_req) + " " + + string(inet_ntoa( clientname.sin_addr )) + ":" + + tool::int2string(ntohs ( clientname.sin_port )) + ); +#endif + + FD_SET (i_new_sock, &active_fd_set); + } + } + else + { + int *p_sock = new int; + *p_sock = i; + p_pool->run( (void*) p_sock ); + FD_CLR( i, &active_fd_set ); + } + } + } +} + +void +sock::clean_ipcache() +{ + int i_ipcachesize = wrap::CONF->get_int("httpd.ipcachesize"); + int i_currentsize = size(); + + if ( i_currentsize > 0 && (i_ipcachesize == 0 || i_ipcachesize >= i_currentsize) ) + { + wrap::system_message( + SOCKCA2+tool::int2string(i_currentsize)+","+tool::int2string(i_ipcachesize)+")"); + clear(); + } +} + +#ifdef NCURSES +void +sock::print_hits() +{ + if ( wrap::NCUR->is_ready() ) + { + mvprintw( NCUR_HITS_X,NCUR_HITS_Y, "Hits: %d ", i_req); + refresh(); + } +} +#endif + +#endif diff --git a/ychat-0.7.8/src/sock/sock.h b/ychat-0.7.8/src/sock/sock.h new file mode 100755 index 0000000..62f32ae --- /dev/null +++ b/ychat-0.7.8/src/sock/sock.h @@ -0,0 +1,74 @@ +#include "../incl.h" + +#ifndef SOCK_H +#define SOCK_H + +#include <queue> +#include <stdio.h> +#include <stdlib.h> +#include <sys/socket.h> +#include <netinet/in.h> +#include <netdb.h> + +#include "../reqp.h" +#include "../chat/user.h" +#include "../thrd/pool.h" +#include "../maps/shashmap.h" + +#ifdef LOGGING +#include "../logd.h" +#endif + +using namespace std; + +class sock : protected shashmap +< string, uint32_t, self_hash<uint32_t>, equals_allocator<uint32_t> > +{ +private: +#ifdef LOGGING + logd *log_daemon; // the log daemon +#endif + + // total number of server requests. + unsigned long long i_req; + bool b_run; // true while socket manager is running. + reqp *req_parser; // parses the http requests from clients. + + char *c_buffer; // char buffer! + pthread_mutex_t mut_hits; + + static string inet_ntoa_callback(void* p_void); + +public: + // creates a server socket. + int make_server_socket( int i_port ); + + // small inline methods: + bool get_run() const + { + return b_run; + } + + bool set_run( bool b_run ) + { + this->b_run = b_run; + } + + sock( ); + ~sock( ); + + int read_write( int* p_sock ); + int start(); + void clean_ipcache(); + + // the chat stream there all the chat messages will sent through. + static void chat_stream( int i_sock, user* p_user, map<string,string> &map_params ); //<< + +#ifdef NCURSES + + void print_hits(); +#endif + +}; + +#endif |
