tagging ychat-0.7.5ychat-0.7.5

4 files changed, 231 insertions, 251 deletions

diff --git a/src/thrd/pool.cpp b/src/thrd/pool.cpp
index a27beac..ef7b0a3 100644..100755
--- a/src/thrd/pool.cpp
+++ b/src/thrd/pool.cpp
@@ -1,27 +1,3 @@
-/*:*
- *: File: ./src/thrd/pool.cpp
- *: 
- *: yChat; Homepage: www.yChat.org; Version 0.7.9.5-RELEASE
- *: 
- *: Copyright (C) 2003 Paul C. Buetow, Volker Richter
- *: Copyright (C) 2004 Paul C. Buetow
- *: Copyright (C) 2005 EXA Digital Solutions GbR
- *: 
- *: 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 the Free Software Foundation; either version 2
- *: of the License, or (at your option) any later version.
- *: 
- *: This program 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 this program; if not, write to the Free Software
- *: Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.
- *:*/
-
 #ifndef POOL_CPP
 #define POOL_CPP
 
@@ -29,153 +5,220 @@
 
 using namespace std;
 
+int pool::i_thrd_used = 0;
+
 pool::pool()
 {
-  pthread_mutex_init(&mut_threads, 0);
-  pthread_mutex_init(&mut_queue_tasks, 0);
-  pthread_mutex_init(&mut_num_avail_threads, 0);
-  pthread_cond_init(&cond_new_task, 0);
-
-  i_num_total_threads = 0;
-  i_num_avail_threads = tool::string2int( wrap::CONF->get_elem( "httpd.thread.initpoolsize" ) );
-  increase_pool(i_num_avail_threads);
+    i_thrd_pool_size  = tool::string2int( wrap::CONF->get_elem( "httpd.thread.initpoolsize" ) );
+    i_thrd_pool_queue = tool::string2int( wrap::CONF->get_elem( "httpd.thread.queuesize" ) );
+    tpool_init( &thread_pool, i_thrd_pool_size, i_thrd_pool_queue);
 }
 
-pool::~pool()
+void
+pool::tpool_init( tpool_t *tpoolp, int num_worker_threads, int max_queue_size)
 {
-  pthread_mutex_lock(&mut_queue_tasks);
-  while (!queue_tasks.empty())
-  {
-    delete queue_tasks.front();
-    queue_tasks.pop();
-  }
-  pthread_mutex_unlock(&mut_queue_tasks);
-
-  pthread_mutex_destroy(&mut_threads);
-  pthread_mutex_destroy(&mut_queue_tasks);
-  pthread_mutex_destroy(&mut_num_avail_threads);
-  pthread_cond_destroy(&cond_new_task);
-}
+    int i, rtn;
+    tpool_t tpool;
 
-int
-pool::increase_pool(int i_num)
-{
-  wrap::system_message(POOLFLL + tool::int2string(i_num) +","+tool::int2string(i_num_total_threads)+")");
-  int i_max_pool_size = tool::string2int( wrap::CONF->get_elem( "httpd.thread.maxpoolsize" ) );
+    // allocate a pool data structure
+    if (( tpool = (tpool_t) malloc( sizeof( struct tpool ) ) ) == 0 )
+    {
+     wrap::system_message( POOLERR );
+     exit(-1);
+    }
+
+    // initialize th fields
+    tpool->num_threads    = num_worker_threads;
+    tpool->max_queue_size = max_queue_size;
 
-  for ( int i = 0; i < i_num; ++i )
-  {
-    if ( i_max_pool_size != 0 && i_num_total_threads >= i_max_pool_size )
+    if ( ( tpool->threads = (pthread_t*) malloc( sizeof(pthread_t)*num_worker_threads ) ) == 0 )
     {
-      wrap::system_message(POOLER2+tool::int2string(i_max_pool_size)+")");
-      wrap::system_message(POOLER1+tool::int2string(i)+")");
-      return i;
+        wrap::system_message( POOLERR );
+        exit(-1);
     }
 
-    ++i_num_total_threads;
-    pthread_t p_pthread;
-    pthread_create(&p_pthread, 0, wait_for_task, (void*) this );
-  }
+    tpool->cur_queue_size = 0;
+    tpool->queue_head     = 0;
+    tpool->queue_tail     = 0;
 
-  return i_num;
-}
+    if ( ( rtn = pthread_mutex_init( &(tpool->queue_lock), 0 ) ) != 0 )
+    {
+        string s_err( "pthread_mutex_init " );
+        s_err.append( strerror( rtn ) );
 
-void
-pool::add_task( void(*p_func)(void*), void* p_void )
-{
-  pthread_mutex_lock(&mut_queue_tasks);
-  queue_tasks.push(new task(p_func, p_void));
-  pthread_mutex_unlock(&mut_queue_tasks);
+        wrap::system_message( s_err );
+
+        exit(-1);
+    }
+
+    else if ( ( rtn = pthread_cond_init( &(tpool->queue_not_empty), 0 ) ) != 0 )
+    {
+        string s_err( "pthread_cond_init (1): " );
+        s_err.append( strerror( rtn ) );
 
-  pthread_cond_signal(&cond_new_task);
+        wrap::system_message( s_err );
 
+        exit(1);
+    }
+
+    else if ( ( rtn = pthread_cond_init( &(tpool->queue_not_full), 0 ) ) != 0 )
+    {
+        string s_err( "pthread_cond_init (2): " );
+        s_err.append( strerror( rtn ) );
+
+        wrap::system_message( s_err );
+
+        exit(1);
+    }
+
+    else if ( ( rtn = pthread_cond_init( &(tpool->queue_empty), 0 ) ) != 0 )
+    {
+        string s_err( "pthread_mutex_init " );
+        s_err.append( strerror( rtn ) );
+
+        wrap::system_message( s_err );
+
+        exit(1);
+    }
+
+    // create threads
+    for ( i = 0; i < num_worker_threads; ++i )
+        pthread_create( &(tpool->threads[i]) , 0, tpool_thread, (void*)tpool );
+
+    *tpoolp = tpool;
 }
 
 void*
-pool::wait_for_task( void* p_void )
+pool::tpool_thread( void* p_void )
 {
-  pool* p_pool = static_cast<pool*>(p_void);
+    tpool_t tpool = (tpool_t) p_void;
+    tpool_work_t *my_workp;
 
-  for (;;)
-  {
+    	for( pthread_mutex_lock( &(tpool->queue_lock) );;
+	     pthread_mutex_lock( &(tpool->queue_lock) ), --i_thrd_used )
+ 	{
 
 #ifdef NCURSES
-    p_pool->print_pool_size();
+        print_threads(i_thrd_used);
 #endif
 
-    pthread_mutex_lock(&p_pool->mut_threads);
-    pthread_cond_wait(&p_pool->cond_new_task, &p_pool->mut_threads);
+        while (tpool->cur_queue_size == 0) 
+           pthread_cond_wait( &(tpool->queue_not_empty), &(tpool->queue_lock) );
 
-    pthread_mutex_lock(&p_pool->mut_num_avail_threads);
-    if ( --p_pool->i_num_avail_threads < 5 )
-    {
-      int i_size = 9 - p_pool->i_num_avail_threads;
-      i_size = p_pool->increase_pool(i_size);
-      p_pool->i_num_avail_threads += i_size;
-    }
-    pthread_mutex_unlock(&p_pool->mut_num_avail_threads);
+        my_workp = tpool->queue_head;
+        tpool->cur_queue_size--;
 
-    pthread_mutex_lock(&p_pool->mut_queue_tasks);
-    task* p_task = p_pool->queue_tasks.front();
-    p_pool->queue_tasks.pop();
-    pthread_mutex_unlock(&p_pool->mut_queue_tasks);
+        if ( tpool->cur_queue_size == 0)
+            tpool->queue_head = tpool->queue_tail = 0;
 
-    pthread_mutex_unlock(&p_pool->mut_threads);
+        else
+            tpool->queue_head = my_workp->next;
 
-    (*(p_task->p_func))(p_task->p_void);
-    delete p_task;
+        if ( tpool->cur_queue_size == ( tpool->max_queue_size - 1 ) )
+            pthread_cond_signal( &(tpool->queue_not_full) );
 
-    pthread_mutex_lock(&p_pool->mut_num_avail_threads);
-    p_pool->i_num_avail_threads++;
-    pthread_mutex_unlock(&p_pool->mut_num_avail_threads);
-  }
+        if ( tpool->cur_queue_size == 0 ) 
+            pthread_cond_signal( &(tpool->queue_empty) );
 
-  return 0;
-}
+        pthread_mutex_unlock( &(tpool->queue_lock) );
 
-void
-pool::run(void* p_void)
-{
-  add_task(run_func, p_void);
+        (*(my_workp->routine))(my_workp->p_void);
+
+        free(my_workp);
+    }
 }
 
-void
-pool::run_func(void *p_void)
+void pool::run_func( void *p_void )
 {
-  socketcontainer* p_sock = static_cast<socketcontainer*>(p_void);
-  wrap::SOCK->read_write(p_sock);
+    int* p_sock = (int*)p_void;
+    wrap::SOCK->read_write( p_sock );
+    delete p_sock;
 }
 
-bool
-pool::allow_user_login()
+int
+pool::tpool_add_work( tpool_t tpool, void(*routine)(void*), void* p_void ) ///
 {
-  pthread_mutex_lock(&mut_num_avail_threads);
-  if ( i_num_avail_threads < 2 )
-  {
-    int i_max_pool_size = tool::string2int( wrap::CONF->get_elem( "httpd.thread.maxpoolsize" ) );
-    if ( i_max_pool_size != 0 && i_max_pool_size == i_num_total_threads )
+    tpool_work_t *workp;
+    pthread_mutex_lock( &(tpool->queue_lock) );
+
+    if ( ++i_thrd_used == tpool->num_threads ) 
     {
-      pthread_mutex_unlock(&mut_num_avail_threads);
-      return false;
+      int i_max_pool_size = tool::string2int( wrap::CONF->get_elem( "httpd.thread.maxpoolsize" ) );
+      if ( i_max_pool_size != 0 && i_thrd_used > i_max_pool_size )
+      {
+        wrap::system_message(POOLER2+tool::int2string(i_thrd_used)+")");
+      }
+
+      else
+      {
+        int i_size = tpool->num_threads + 1;
+
+        wrap::system_message(POOLFLL+tool::int2string(i_size)+")");
+
+        tpool->threads = (pthread_t*)realloc((void*)tpool->threads, sizeof(pthread_t)*tpool->num_threads);
+
+        for ( int i = tpool->num_threads; i < i_size; ++i )
+          pthread_create( &(tpool->threads[i]) , 0, tpool_thread, (void*)tpool );
+
+        i_thrd_pool_size = tpool->num_threads = i_size;
+#ifdef NCURSES
+        print_pool_size();
+#endif
+      }
     }
-  }
-  pthread_mutex_unlock(&mut_num_avail_threads);
 
-  return true;
+#ifdef NCURSES
+    print_threads(i_thrd_used);
+#endif
+
+    while (tpool->cur_queue_size == tpool->max_queue_size) 
+        pthread_cond_wait( &(tpool->queue_not_full), &(tpool->queue_lock) );
+
+    // allocate work structure:
+    workp = (tpool_work_t*) malloc( sizeof( tpool_work_t ) );
+
+    workp->routine = routine;
+    workp->p_void  = p_void;
+    workp->next    = 0;
+
+    if (tpool->cur_queue_size == 0 )
+    {
+        tpool->queue_tail = tpool->queue_head = workp;
+    }
+
+    else
+    {
+        (tpool->queue_tail)->next = workp;
+        tpool->queue_tail         = workp;
+    }
+
+    tpool->cur_queue_size++;
+    pthread_cond_signal( &tpool->queue_not_empty );
+    pthread_mutex_unlock( &(tpool->queue_lock) );
+
+    return 0;
 }
 
 #ifdef NCURSES
 void
+pool::print_threads(int i_thrd_used)
+{
+    if ( wrap::NCUR->is_ready() )
+    {
+     mvprintw( NCUR_POOL_RUNNING_X,NCUR_POOL_RUNNING_Y, "In use: %d ", i_thrd_used);
+     refresh();
+   }
+}
+
+void
 pool::print_pool_size()
 {
-  if ( wrap::NCUR->is_ready() )
-  {
-    pthread_mutex_lock(&mut_num_avail_threads);
-    mvprintw( NCUR_POOL_WAIT_X,NCUR_POOL_WAIT_Y, "Wait/Tot: %d/%d  ", i_num_avail_threads, i_num_total_threads);
-    mvprintw( NCUR_POOL_RUNNING_X,NCUR_POOL_RUNNING_Y, "Running: %d  ", i_num_total_threads-i_num_avail_threads);
-    pthread_mutex_unlock(&mut_num_avail_threads);
-    refresh();
-  }
+    if ( wrap::NCUR->is_ready() )
+    {
+     mvprintw( NCUR_POOL_SIZE_X,NCUR_POOL_SIZE_Y, "Size: %d %d", i_thrd_pool_size, i_thrd_pool_queue );
+     refresh();
+   }
 }
 #endif
+
 #endif
diff --git a/src/thrd/pool.h b/src/thrd/pool.h
index 0fe97f3..cbd1e83 100644..100755
--- a/src/thrd/pool.h
+++ b/src/thrd/pool.h
@@ -1,78 +1,64 @@
-/*:*
- *: File: ./src/thrd/pool.h
- *: 
- *: yChat; Homepage: www.yChat.org; Version 0.7.9.5-RELEASE
- *: 
- *: Copyright (C) 2003 Paul C. Buetow, Volker Richter
- *: Copyright (C) 2004 Paul C. Buetow
- *: Copyright (C) 2005 EXA Digital Solutions GbR
- *: 
- *: 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 the Free Software Foundation; either version 2
- *: of the License, or (at your option) any later version.
- *: 
- *: This program 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 this program; if not, write to the Free Software
- *: Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.
- *:*/
-
 #include "../incl.h"
 
 #ifndef POOL_H
 #define POOL_H
 
-#include <queue>
-
 using namespace std;
 
 class pool
 {
 private:
-  friend class thro;
+    static int i_thrd_used;
 
-  struct task
-  {
-    void(*p_func)(void*);
-    void *p_void;
+    typedef struct tpool_work {
+        void (*routine)(void*); ///
+        void *p_void;
+        struct tpool_work *next;
+    }
+    tpool_work_t;
 
-    task(void(*p_func)(void*), void *p_void)
+    typedef struct tpool
     {
-      this->p_func = p_func;
-      this->p_void = p_void;
-    }
-  };
+        // pool characteristics:
+        int num_threads;
+        int max_queue_size;
+
+        // pool state
+        pthread_t *threads;
+        int cur_queue_size;
 
-  pthread_mutex_t mut_threads;
-  pthread_mutex_t mut_queue_tasks;
-  pthread_mutex_t mut_num_avail_threads;
-  pthread_cond_t cond_new_task;
+        tpool_work_t *queue_head;
+        tpool_work_t *queue_tail;
 
-  int i_num_avail_threads;
-  int i_num_total_threads;
+        pthread_mutex_t queue_lock;
+        pthread_cond_t  queue_not_empty;
+        pthread_cond_t  queue_not_full;
+        pthread_cond_t  queue_empty;
+    }
+    *tpool_t;
+
+    int i_thrd_pool_size;
+    int i_thrd_pool_queue;
 
-  queue<task*> queue_tasks;
+    tpool_t thread_pool;
 
-  int increase_pool(int i_num);
-  void add_task( void(*p_func)(void*), void* p_void );
-  static void* wait_for_task(void *p_void);
-  static void run_func(void *p_void);
+    void tpool_init( tpool_t *tpoolp, int num_worker_threads, int max_queue_size); 
+    int tpool_add_work( tpool_t tpool, void(*routine)(void*), void* p_void );
+    static void* tpool_thread( void *p_void);
+    static void run_func( void *p_void );
 
 public:
-  pool();
-  ~pool();
+    pool();
 
-  void run(void* p_void);
-  bool allow_user_login();
+    // inline (speed)!
+    void run( void *p_void )
+    {
+        tpool_add_work( thread_pool, run_func, p_void );
+    }
 
 #ifdef NCURSES
-
-  void print_pool_size();
+    void print_pool_size();
+    static void print_threads(int i_thrd_used);
 #endif
 };
 
diff --git a/src/thrd/thro.cpp b/src/thrd/thro.cpp
index 04372b3..739ba0e 100644
--- a/src/thrd/thro.cpp
+++ b/src/thrd/thro.cpp
@@ -1,27 +1,3 @@
-/*:*
- *: File: ./src/thrd/thro.cpp
- *: 
- *: yChat; Homepage: www.yChat.org; Version 0.7.9.5-RELEASE
- *: 
- *: Copyright (C) 2003 Paul C. Buetow, Volker Richter
- *: Copyright (C) 2004 Paul C. Buetow
- *: Copyright (C) 2005 EXA Digital Solutions GbR
- *: 
- *: 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 the Free Software Foundation; either version 2
- *: of the License, or (at your option) any later version.
- *: 
- *: This program 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 this program; if not, write to the Free Software
- *: Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.
- *:*/
-
 #ifndef THRO_CPP
 #define THRO_CPP
 
@@ -30,38 +6,39 @@
 using namespace std;
 
 thro::thro()
-{}
+{
+}
 
 thro::~thro()
-{}
+{
+}
 
 void
 thro::run()
 {
-  void *p_void;
-  run( p_void );
+ void *p_void;
+ run( p_void );
 }
 
 void
 thro::run( void *p_void )
 {
-  elem.p_thro = this;
-  elem.p_void = p_void;
-  //wrap::POOL->add_task(start_, &elem);
-  pthread_create( &pthread, NULL, start_, &elem );
+ elem.p_thro = this;
+ elem.p_void = p_void;
+ pthread_create( &pthread, NULL, start_, &elem );
 }
 
 void*
 thro::start_( void *p_void )
 {
-  elements *e = (elements*) p_void;
-  e->p_thro->start( e->p_void );
+ elements *e = (elements*) p_void;
+ e->p_thro->start( e->p_void ); 
 }
 
 void
 thro::start( void *p_void )
 {
-  wrap::system_message( THRDSTR );
+ wrap::system_message( THRDSTR );
 }
 
 #endif
diff --git a/src/thrd/thro.h b/src/thrd/thro.h
index 76e5a51..9ad1e3e 100644
--- a/src/thrd/thro.h
+++ b/src/thrd/thro.h
@@ -1,27 +1,3 @@
-/*:*
- *: File: ./src/thrd/thro.h
- *: 
- *: yChat; Homepage: www.yChat.org; Version 0.7.9.5-RELEASE
- *: 
- *: Copyright (C) 2003 Paul C. Buetow, Volker Richter
- *: Copyright (C) 2004 Paul C. Buetow
- *: Copyright (C) 2005 EXA Digital Solutions GbR
- *: 
- *: 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 the Free Software Foundation; either version 2
- *: of the License, or (at your option) any later version.
- *: 
- *: This program 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 this program; if not, write to the Free Software
- *: Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.
- *:*/
-
 #include "../incl.h"
 
 #ifndef THRO_H
@@ -32,23 +8,21 @@ using namespace std;
 class thro
 {
 private:
-  pthread_t pthread;
+ pthread_t pthread;
 
-  struct elements
-  {
-    thro *p_thro;
-    void *p_void;
-  }
-  elem;
+ struct elements {
+  thro *p_thro;
+  void *p_void;
+ } elem;  
 
-  static void *start_( void *p_void );
+ static void* start_( void *p_void );
 
 public:
-  thro( );
-  ~thro( );
-  void run();
-  void run( void *p_void );
-  virtual void start( void *p_void );
+ thro( );
+ ~thro( );
+ void run();
+ void run( void *p_void );
+ virtual void start( void *p_void );
 };
 
 #endif