summaryrefslogtreecommitdiff
path: root/sock.cpp
diff options
context:
space:
mode:
authorPaul Buetow <paul@buetow.org>2013-04-06 13:14:44 +0200
committerPaul Buetow <paul@buetow.org>2013-04-06 13:14:44 +0200
commitca28c0e618890330d429c0dc12429255b20f0c90 (patch)
treeecc02da0184cf4e8bdba94dcdd831abdd1e51b3c /sock.cpp
parentb3a99e6e15af3be25394e66d1138bb2682f565c3 (diff)
tagging ychat-0.5.0ychat-0.5.0
Diffstat (limited to 'sock.cpp')
-rwxr-xr-xsock.cpp466
1 files changed, 155 insertions, 311 deletions
diff --git a/sock.cpp b/sock.cpp
index 5d64367..fb041fb 100755
--- a/sock.cpp
+++ b/sock.cpp
@@ -9,12 +9,10 @@
#include "sock.h"
#include "s_chat.h"
#include "s_conf.h"
+#include "s_mutx.h"
#include "s_tool.h"
#include "s_lang.h"
#include "s_sman.h"
-#include "s_mman.h"
-#include "s_mutx.h"
-#include "s_ncur.h"
#include "chat.h"
#include "user.h"
@@ -22,373 +20,219 @@ using namespace std;
sock::sock()
{
- this->b_run = true;
- this->i_req = 0;
- this->i_threads = 2; // currently there are 2 threads running ( admin interface + server listening );
- this->req_parser = new reqp();
- this->thrd_pool = new pool();
- this->log_daemon = new logd(s_conf::get
- ().get_val( "ACCESS_LOG" ));
-
-
- pthread_mutex_init( &mut_threads, NULL );
-}
-
-sock::~sock()
-{
- pthread_mutex_destroy( &mut_threads );
+ this->b_run = true;
+ this->i_req = 0;
+ this->req_parser = new reqp();
+ this->thrd_pool = new pool();
+ this->log_daemon = new logd(s_conf::get().get_val( "ACCESS_LOG" ));
}
void
sock::chat_stream( int i_sock, user* p_user, map_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 );
+ string s_msg( "\n" );
- do
- {
- pthread_mutex_lock ( &(p_user->mut_message) );
- s_msg = p_user->get_mess( );
- if ( 0 > send( i_sock, s_msg.c_str(), s_msg.size(), 0 ) )
- p_user->set_online( false );
- pthread_cond_wait( &(p_user->cond_message), &(p_user->mut_message) );
- }
- while( p_user->get_online() );
+ pthread_mutex_lock ( &(p_user->mut_message) );
- // 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 );
+ for ( int i = 0; i < PUSHSTR; i++ )
+ send( i_sock, s_msg.c_str(), s_msg.size(), 0 );
- pthread_mutex_unlock( &(p_user->mut_message) );
+ 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 );
+ pthread_cond_wait( &(p_user->cond_message), &(p_user->mut_message) );
+ }
+ while( p_user->get_online() );
- // remove the user from its room.
- string s_user( p_user->get_name() );
- p_user->get_room()->del_elem( s_user );
+ // 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 );
-#ifdef NCURSES
+ pthread_mutex_unlock( &(p_user->mut_message) );
- {
- string s_tmp( REMUSER );
- s_tmp.append( s_user );
- s_ncur::get
- ().print( s_tmp.c_str() );
- }
-#endif
- // post the room that the user has left the chat.
- p_user->get_room()->msg_post( new string( p_user->get_colored_name().append( s_lang::get
- ().get_val( "USERLEAV" ) ) ) );
- s_sman::get
- ().destroy_session( p_user->get_id() );
+ // remove the user from its room.
+ string s_user( p_user->get_name() );
+ p_user->get_p_room()->del_elem( s_user );
-#ifdef NCURSES
- {
- string s_tmp( SESSION );
- s_tmp.append( s_tool::int2string( s_sman::get
- ().get_session_count() ) );
- s_ncur::get
- ().print( s_tmp );
- }
-#endif
-#ifdef VERBOSE
- // cout << REMUSER << s_user << endl
- // << SESSION << s_man::get().get_session_count() << endl;
-
-#endif
+ // post the room that the user has left the chat.
+ p_user->get_p_room()->msg_post( new string( p_user->get_name().append( s_lang::get().get_val( "USERLEAV" ) ) ) );
+ s_sman::get().destroySession( p_user->get_id() );
+ #ifdef VERBOSE
+ cout << s_user << " left | SessionCount: " << s_sman::get().getSessionCount() << endl;
+ #endif
- p_user->~user();
+ p_user->~user();
}
int
sock::make_socket( uint16_t i_port )
{
- int sock;
- struct sockaddr_in name;
-
- // create the server socket.
- sock = socket (PF_INET, SOCK_STREAM, 0);
- if (sock < 0)
- {
-
-#ifdef NCURSES
- s_ncur::get
- ().print( SOCKERR );
-#endif
-
- if ( ++i_port > MAXPORT )
- exit(-1);
+ int sock;
+ struct sockaddr_in name;
-#ifdef NCURSES
+ // create the server socket.
+ sock = socket (PF_INET, SOCK_STREAM, 0);
+ if (sock < 0)
+ {
+ cerr << "Sock: socket error" << endl;
- s_ncur::get
- ().print( SOCKERR );
-#endif
-
- return make_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 optval=1;
-
- setsockopt(sock, SOL_SOCKET, SO_REUSEADDR, (char*)&optval, sizeof(int));
-
- if (bind (sock, (struct sockaddr *) &name, sizeof (name)) < 0)
- {
-
-#ifdef NCURSES
- s_ncur::get
- ().print( BINDERR );
-#endif
+ if ( ++i_port > MAXPORT )
+ exit(-1);
- if ( ++i_port > MAXPORT )
- exit(-1);
+ cerr << SOCKERR << i_port << endl;
+ return make_socket( i_port );
+ }
-#ifdef SERVMSG
+ // 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 optval=1;
- cout << SOCKERR << i_port << endl;
-#endif
-#ifdef NCURSES
+ setsockopt(sock, SOL_SOCKET, SO_REUSEADDR, (char*)&optval, sizeof(int));
- s_ncur::get
- ().print( SOCKERR );
-#endif
+ if (bind (sock, (struct sockaddr *) &name, sizeof (name)) < 0)
+ {
+ cerr << "Sock: bind error" << endl;
- return make_socket( i_port );
- }
+ if ( ++i_port > MAXPORT )
+ exit(-1);
-#ifdef VERBOSE
- cout << SOCKCRT << "localhost:" << i_port << endl;
-#endif
-#ifdef NCURSES
-
- string s_tmp( SOCKCRT );
- s_tmp.append( "localhost:" );
- s_tmp.append( s_tool::int2string(i_port) );
- s_ncur::get
- ().print( s_tmp );
- mvprintw( NCUR_PORT_X,NCUR_PORT_Y, "Port: %d ", i_port);
- refresh();
-#endif
+ cout << SOCKERR << i_port << endl;
+ return make_socket( i_port );
+ }
- return sock;
+ return sock;
}
int
sock::read_write( thrd* p_thrd, int i_sock )
{
- char c_req[2048];
-
- int i_bytes;
- i_bytes = read (i_sock, c_req, 2048);
-
- if (i_bytes < 0)
- {
-#ifdef NCURSES
- s_ncur::get
- ().print( new string( READERR ) );
-#endif
+ char c_req[2048];
- }
- else
- {
- // stores the request params.
- map_string map_params;
+ int i_bytes;
+ i_bytes = read (i_sock, c_req, 2048);
- // get the s_rep ( s_html response which will be send imediatly to the client
- // and fill map_params with request values.
- auto string s_temp=(string)c_req;
- struct sockaddr_in client;
- size_t size=sizeof(client);
+ if (i_bytes < 0)
+ {
+ cerr << "Sock: read error " << endl;
+ }
- getpeername( i_sock, (struct sockaddr *)&client, &size);
+ else
+ {
+ // stores the request params.
+ map_string map_params;
- map_params["REMOTE_ADDR"]=inet_ntoa(client.sin_addr);
- map_params["REMOTE_PORT"]=ntohs( client.sin_port);
+ // get the s_rep ( s_html response which will be send imediatly to the client
+ // and fill map_params with request values.
+ auto string s_temp=(string)c_req;
+ struct sockaddr_in client;
+ size_t size=sizeof(client);
+ getpeername( i_sock, (struct sockaddr *)&client, &size);
+
+ map_params["REMOTE_ADDR"]=inet_ntoa(client.sin_addr);
+ map_params["REMOTE_PORT"]=ntohs( client.sin_port);
- string s_rep = req_parser->parse( p_thrd, string( c_req ), map_params );
- // send s_rep to the client.
- log_daemon->log(map_params);
- send( i_sock, s_rep.c_str(), s_rep.size(), 0 );
+ string s_rep = req_parser->parse( p_thrd, string( c_req ), map_params );
+ // send s_rep to the client.
+ log_daemon->log(map_params);
- // dont need those vals anymore.
- map_params.clear();
+ send( i_sock, s_rep.c_str(), s_rep.size(), 0 );
- return 0;
- }
+ // dont need those vals anymore.
+ map_params.clear();
- return -1;
+ return 0;
+ }
+
+ return -1;
}
int
sock::start()
{
-#ifdef NCURSES
- s_ncur::get
- ().print( STARTMS );
- print_hits();
- print_threads();
-#endif
+ auto int i_port = s_tool::string2int( s_conf::get().get_val( "SRVRPORT" ) );
- auto int i_port = s_tool::string2int( s_conf::get
- ().get_val( "SRVRPORT" ) );
-
- int sock;
- fd_set active_fd_set, read_fd_set;
- int i;
- struct sockaddr_in clientname;
- size_t size;
-
-
- // create the server socket and set it up to accept connections.
- sock = make_socket ( i_port );
-
- if (listen (sock, 1) < 0)
- {
-#ifdef NCURSES
- s_ncur::get
- ().print( LISTERR );
-#endif
-
- exit( EXIT_FAILURE );
- }
+ int sock;
+ fd_set active_fd_set, read_fd_set;
+ int i;
+ struct sockaddr_in clientname;
+ size_t size;
#ifdef VERBOSE
- cout << SOCKRDY << endl;
+ cout << SOCKCRT << "localhost:" << i_port << endl;
#endif
-#ifdef NCURSES
- s_ncur::get
- ().print( SOCKRDY );
-#endif
+ // create the server socket and set it up to accept connections.
+ sock = make_socket ( i_port );
- // initialize the set of active sockets.
- FD_ZERO (&active_fd_set);
- FD_SET (sock, &active_fd_set);
+ if (listen (sock, 1) < 0)
+ {
+ cerr << "Sock: listen error" << endl;
+ exit( EXIT_FAILURE );
+ }
- while( b_run )
+#ifdef VERBOSE
+ cout << SOCKRDY << endl;
+#endif
+
+ // initialize the set of active sockets.
+ FD_ZERO (&active_fd_set);
+ FD_SET (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)
+ {
+ cerr << "Sock: select error" << endl;
+ 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 == sock )
{
- // 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)
- {
-
-#ifdef NCURSES
- s_ncur::get
- ().print( SELCERR );
-#endif
-
- exit( EXIT_FAILURE );
- }
+ // connection request on original socket.
+ i_req++;
+ int new_sock;
+ size = sizeof (clientname);
+ new_sock = accept (sock,
+ (struct sockaddr *) &clientname,
+ &size);
+
+ if (new_sock < 0)
+ {
+ cerr << "Sock: accept error" << endl;
+ close ( new_sock );
+ }
- // service all the sockets with input pending.
- for ( i = 0; i < FD_SETSIZE; i++ )
- if ( FD_ISSET (i, &read_fd_set) )
- {
- if ( i == sock )
- {
- // connection request on original socket.
- i_req++;
-#ifdef NCURSES
-
- print_hits();
-#endif
-
- int new_sock;
- size = sizeof (clientname);
- new_sock = accept (sock,
- (struct sockaddr *) &clientname,
- &size);
-
- if (new_sock < 0)
- {
-
-#ifdef NCURSES
- s_ncur::get
- ().print( ACCPERR);
-#endif
-
- close ( new_sock );
-
- }
-
- else
- {
#ifdef VERBOSE
- cout << NEWREQU << i_req << " "
- << inet_ntoa( clientname.sin_addr )
- << ":"
- << ntohs ( clientname.sin_port )
- << endl;
-#endif
- FD_SET (new_sock, &active_fd_set);
- }
- }
-
- else
- {
-
-#ifdef NCURSES
- /// s_ncur::get().print( NEWREQU );
-#endif
-
- thrd_pool->run( (void*) new thrd( i ) );
- FD_CLR( i, &active_fd_set );
- }
- }
+ cout << CONNECT << i_req << " "
+ << inet_ntoa( clientname.sin_addr )
+ << ":"
+ << ntohs ( clientname.sin_port )
+ << endl;
+#endif
+
+ FD_SET (new_sock, &active_fd_set);
}
-}
-
-void
-sock::increase_num_threads()
-{
- pthread_mutex_lock( &mut_threads );
- i_threads++;
- pthread_mutex_unlock( &mut_threads );
-#ifdef NCURSES
- print_threads();
-#endif
-
-}
-
-void
-sock::decrease_num_threads()
-{
- pthread_mutex_lock( &mut_threads );
- i_threads--;
- pthread_mutex_unlock( &mut_threads );
-
-#ifdef NCURSES
- print_threads();
-#endif
+ else
+ {
+ thrd_pool->run( (void*) new thrd( i ) );
+ FD_CLR( i, &active_fd_set );
+ }
+ }
+ }
}
-#ifdef NCURSES
-void
-sock::print_threads()
-{
- pthread_mutex_lock ( &s_mutx::get().mut_stdout );
- mvprintw( NCUR_THREADS_X,NCUR_THREADS_Y, "Threads: %d ", i_threads);
- refresh();
- pthread_mutex_unlock( &s_mutx::get().mut_stdout );
-}
-
-void
-sock::print_hits()
-{
- pthread_mutex_lock ( &s_mutx::get().mut_stdout );
- mvprintw( NCUR_HITS_X,NCUR_HITS_Y, "Hits: %d ", i_req);
- refresh();
- pthread_mutex_unlock( &s_mutx::get().mut_stdout );
-}
-#endif
-
#endif