#define MODULE_LOG_PREFIX "work"

#include "globals.h"

#include "module-cacheex.h"

#include "oscam-client.h"

#include "oscam-ecm.h"

#include "oscam-emm.h"

#include "oscam-lock.h"

#include "oscam-net.h"

#include "oscam-reader.h"

#include "oscam-string.h"

#include "oscam-work.h"

#include "reader-common.h"

#include "module-cccam-data.h"

#include "module-cccshare.h"

#include "oscam-time.h"

extern CS_MUTEX_LOCK system_lock;

extern int32_t thread_pipe[2];

struct job_data

};

static void free_job_data(struct job_data *data)

{

if(!data)

{ return; }

if(data->len && data->ptr)

{

//special free checks

if(data->action==ACTION_ECM_ANSWER_CACHE)

{

NULLFREE(((struct s_write_from_cache *)data->ptr)->er_cache);

}

NULLFREE(data->ptr);

}

NULLFREE(data);

}

void free_joblist(struct s_client *cl)

{

int32_t lock_status = pthread_mutex_trylock(&cl->thread_lock);

LL_ITER it = ll_iter_create(cl->joblist);

struct job_data *data;

while((data = ll_iter_next(&it)))

{

free_job_data(data);

}

ll_destroy(&cl->joblist);

cl->account = NULL;

if(cl->work_job_data) // Free job_data that was not freed by work_thread

{ free_job_data(cl->work_job_data); }

cl->work_job_data = NULL;

if(lock_status == 0)

{ SAFE_MUTEX_UNLOCK(&cl->thread_lock); }

pthread_mutex_destroy(&cl->thread_lock);

}

/*

Work threads are named like this:

w[r|c]XX-[rdr->label|client->username]

w - work thread prefix

[r|c] - depending whether the the action is related to reader or client

XX - two digit action code from enum actions

label - reader label or client username (see username() function)

*/

static void set_work_thread_name(struct job_data *data)

{

char thread_name[16 + 1];

snprintf(thread_name, sizeof(thread_name), "w%c%02d-%s",

data->action < ACTION_CLIENT_FIRST ? 'r' : 'c',

data->action,

username(data->cl)

);

set_thread_name(thread_name);

}

#define __free_job_data(client, job_data) \

do { \

client->work_job_data = NULL; \

if (job_data && job_data != &tmp_data) { \

free_job_data(job_data); \

} \

job_data = NULL; \

} while(0)

void *work_thread(void *ptr)

{

struct job_data *data = (struct job_data *)ptr;

struct s_client *cl = data->cl;

struct s_reader *reader = cl->reader;

struct timeb start, end; // start time poll, end time poll

struct job_data tmp_data;

struct pollfd pfd[1];

SAFE_SETSPECIFIC(getclient, cl);

cl->thread = pthread_self();

cl->thread_active = 1;

set_work_thread_name(data);

struct s_module *module = get_module(cl);

uint16_t bufsize = module->bufsize; //CCCam needs more than 1024bytes!

if(!bufsize)

{ bufsize = DEFAULT_MODULE_BUFSIZE; }

uint8_t *mbuf;

if(!cs_malloc(&mbuf, bufsize))

{ return NULL; }

cl->work_mbuf = mbuf; // Track locally allocated data, because some callback may call cs_exit/cs_disconect_client/pthread_exit and then mbuf would be leaked

int32_t n = 0, rc = 0, i, idx, s;

uint8_t dcw[16];

int8_t restart_reader = 0;

while(cl->thread_active)

{

cs_ftime(&start); // register start time

while(cl->thread_active)

{

if(!cl || cl->kill || !is_valid_client(cl))

{

SAFE_MUTEX_LOCK(&cl->thread_lock);

cl->thread_active = 0;

SAFE_MUTEX_UNLOCK(&cl->thread_lock);

cs_log_dbg(D_TRACE, "ending thread (kill)");

__free_job_data(cl, data);

cl->work_mbuf = NULL; // Prevent free_client from freeing mbuf (->work_mbuf)

free_client(cl);

if(restart_reader)

{ restart_cardreader(reader, 0); }

NULLFREE(mbuf);

pthread_exit(NULL);

return NULL;

}

if(data && data->action != ACTION_READER_CHECK_HEALTH)

{ cs_log_dbg(D_TRACE, "data from add_job action=%d client %c %s", data->action, cl->typ, username(cl)); }

if(!data)

{

if(!cl->kill && cl->typ != 'r')

{ client_check_status(cl); } // do not call for physical readers as this might cause an endless job loop

SAFE_MUTEX_LOCK(&cl->thread_lock);

if(cl->joblist && ll_count(cl->joblist) > 0)

{

LL_ITER itr = ll_iter_create(cl->joblist);

data = ll_iter_next_remove(&itr);

if(data)

{ set_work_thread_name(data); }

//cs_log_dbg(D_TRACE, "start next job from list action=%d", data->action);

}

SAFE_MUTEX_UNLOCK(&cl->thread_lock);

}

if(!data)

{

/* for serial client cl->pfd is file descriptor for serial port not socket

for example: pfd=open("/dev/ttyUSB0"); */

if(!cl->pfd || module->listenertype == LIS_SERIAL)

{ break; }

pfd[0].fd = cl->pfd;

pfd[0].events = POLLIN | POLLPRI;

SAFE_MUTEX_LOCK(&cl->thread_lock);

cl->thread_active = 2;

SAFE_MUTEX_UNLOCK(&cl->thread_lock);

rc = poll(pfd, 1, 3000);

SAFE_MUTEX_LOCK(&cl->thread_lock);

cl->thread_active = 1;

SAFE_MUTEX_UNLOCK(&cl->thread_lock);

if(rc > 0)

{

cs_ftime(&end); // register end time

cs_log_dbg(D_TRACE, "[OSCAM-WORK] new event %d occurred on fd %d after %"PRId64" ms inactivity", pfd[0].revents,

pfd[0].fd, comp_timeb(&end, &start));

data = &tmp_data;

data->ptr = NULL;

cs_ftime(&start); // register start time for new poll next run

if(reader)

{ data->action = ACTION_READER_REMOTE; }

else

{

if(cl->is_udp)

{

data->action = ACTION_CLIENT_UDP;

data->ptr = mbuf;

data->len = bufsize;

}

else

{ data->action = ACTION_CLIENT_TCP; }

if(pfd[0].revents & (POLLHUP | POLLNVAL | POLLERR))

{ cl->kill = 1; }

}

}

}

if(!data)

{ continue; }

if(!reader && data->action < ACTION_CLIENT_FIRST)

{

__free_job_data(cl, data);

break;

}

if(!data->action)

{ break; }

struct timeb actualtime;

cs_ftime(&actualtime);

int64_t gone = comp_timeb(&actualtime, &data->time);

if(data != &tmp_data && gone > (int) cfg.ctimeout+1000)

{

cs_log_dbg(D_TRACE, "dropping client data for %s time %"PRId64" ms", username(cl), gone);

__free_job_data(cl, data);

continue;

}

if(data != &tmp_data)

{ cl->work_job_data = data; } // Track the current job_data

switch(data->action)

{

case ACTION_READER_IDLE:

reader_do_idle(reader);

break;

case ACTION_READER_REMOTE:

s = check_fd_for_data(cl->pfd);

if(s == 0) // no data, another thread already read from fd?

{ break; }

if(s < 0)

{

if(reader->ph.type == MOD_CONN_TCP)

{ network_tcp_connection_close(reader, "disconnect"); }

break;

}

rc = reader->ph.recv(cl, mbuf, bufsize);

if(rc < 0)

{

if(reader->ph.type == MOD_CONN_TCP)

{ network_tcp_connection_close(reader, "disconnect on receive"); }

break;

}

cl->last = time(NULL); // *********************************** TO BE REPLACE BY CS_FTIME() LATER ****************

idx = reader->ph.c_recv_chk(cl, dcw, &rc, mbuf, rc);

if(idx < 0) { break; } // no dcw received

if(!idx) { idx = cl->last_idx; }

reader->last_g = time(NULL); // *********************************** TO BE REPLACE BY CS_FTIME() LATER **************** // for reconnect timeout

for(i = 0, n = 0; i < cfg.max_pending && n == 0; i++)

{

if(cl->ecmtask[i].idx == idx)

{

cl->pending--;

casc_check_dcw(reader, i, rc, dcw);

n++;

}

}

break;

case ACTION_READER_RESET:

cardreader_do_reset(reader);

break;

case ACTION_READER_ECM_REQUEST:

reader_get_ecm(reader, data->ptr);

break;

case ACTION_READER_EMM:

reader_do_emm(reader, data->ptr);

break;

case ACTION_READER_CARDINFO:

reader_do_card_info(reader);

break;

case ACTION_READER_POLL_STATUS:

cardreader_poll_status(reader);

break;

case ACTION_READER_INIT:

if(!cl->init_done)

{ reader_init(reader); }

break;

case ACTION_READER_RESTART:

cl->kill = 1;

restart_reader = 1;

break;

case ACTION_READER_RESET_FAST:

reader->card_status = CARD_NEED_INIT;

cardreader_do_reset(reader);

break;

case ACTION_READER_CHECK_HEALTH:

cardreader_do_checkhealth(reader);

break;

case ACTION_READER_CAPMT_NOTIFY:

if(reader->ph.c_capmt) { reader->ph.c_capmt(cl, data->ptr); }

break;

case ACTION_CLIENT_UDP:

n = module->recv(cl, data->ptr, data->len);

if(n < 0) { break; }

module->s_handler(cl, data->ptr, n);

break;

case ACTION_CLIENT_TCP:

s = check_fd_for_data(cl->pfd);

if(s == 0) // no data, another thread already read from fd?

{ break; }

if(s < 0) // system error or fd wants to be closed

{

cl->kill = 1; // kill client on next run

continue;

}

n = module->recv(cl, mbuf, bufsize);

if(n < 0)

{

cl->kill = 1; // kill client on next run

continue;

}

module->s_handler(cl, mbuf, n);

break;

case ACTION_CACHEEX1_DELAY:

cacheex_mode1_delay(data->ptr);

break;

case ACTION_CACHEEX_TIMEOUT:

cacheex_timeout(data->ptr);

break;

case ACTION_FALLBACK_TIMEOUT:

fallback_timeout(data->ptr);

break;

case ACTION_CLIENT_TIMEOUT:

ecm_timeout(data->ptr);

break;

case ACTION_ECM_ANSWER_READER:

chk_dcw(data->ptr);

break;

case ACTION_ECM_ANSWER_CACHE:

write_ecm_answer_fromcache(data->ptr);

break;

case ACTION_CLIENT_INIT:

if(module->s_init)

{ module->s_init(cl); }

cl->is_udp = module->type == MOD_CONN_UDP;

cl->init_done = 1;

break;

case ACTION_CLIENT_IDLE:

if(module->s_idle)

{ module->s_idle(cl); }

else

{

cs_log("user %s reached %d sec idle limit.", username(cl), cfg.cmaxidle);

cl->kill = 1;

}

break;

case ACTION_CACHE_PUSH_OUT:

{

cacheex_push_out(cl, data->ptr);

break;

}

case ACTION_CLIENT_KILL:

cl->kill = 1;

break;

case ACTION_CLIENT_SEND_MSG:

{

if (config_enabled(MODULE_CCCAM))

{

struct s_clientmsg *clientmsg = (struct s_clientmsg *)data->ptr;

cc_cmd_send(cl, clientmsg->msg, clientmsg->len, clientmsg->cmd);

}

break;

}

} // switch

__free_job_data(cl, data);

}

if(thread_pipe[1] && (mbuf[0] != 0x00))

{

cs_log_dump_dbg(D_TRACE, mbuf, 1, "[OSCAM-WORK] Write to pipe:");

if(write(thread_pipe[1], mbuf, 1) == -1) // wakeup client check

{

cs_log_dbg(D_TRACE, "[OSCAM-WORK] Writing to pipe failed (errno=%d %s)", errno, strerror(errno));

}

}

// Check for some race condition where while we ended, another thread added a job

SAFE_MUTEX_LOCK(&cl->thread_lock);

if(cl->joblist && ll_count(cl->joblist) > 0)

{

SAFE_MUTEX_UNLOCK(&cl->thread_lock);

continue;

}

else

{

cl->thread_active = 0;

SAFE_MUTEX_UNLOCK(&cl->thread_lock);

break;

}

}

cl->thread_active = 0;

cl->work_mbuf = NULL; // Prevent free_client from freeing mbuf (->work_mbuf)

NULLFREE(mbuf);

pthread_exit(NULL);

return NULL;

}

/**

* adds a job to the job queue

* if ptr should be free() after use, set len to the size

* else set size to 0

**/

int32_t add_job(struct s_client *cl, enum actions action, void *ptr, int32_t len)

{

if(!cl || cl->kill)

{

if(!cl)

{ cs_log("WARNING: add_job failed. Client killed!"); } // Ignore jobs for killed clients

if(len && ptr)

{ NULLFREE(ptr); }

return 0;

}

if(action == ACTION_CACHE_PUSH_OUT && cacheex_check_queue_length(cl))

{

if(len && ptr)

{ NULLFREE(ptr); }

return 0;

}

struct job_data *data;

if(!cs_malloc(&data, sizeof(struct job_data)))

{

if(len && ptr)

{ NULLFREE(ptr); }

return 0;

}

data->action = action;

data->ptr = ptr;

data->cl = cl;

data->len = len;

cs_ftime(&data->time);

SAFE_MUTEX_LOCK(&cl->thread_lock);

if(cl && !cl->kill && cl->thread_active)

{

if(!cl->joblist)

{ cl->joblist = ll_create("joblist"); }

ll_append(cl->joblist, data);

if(cl->thread_active == 2)

{ pthread_kill(cl->thread, OSCAM_SIGNAL_WAKEUP); }

SAFE_MUTEX_UNLOCK(&cl->thread_lock);

cs_log_dbg(D_TRACE, "add %s job action %d queue length %d %s",

action > ACTION_CLIENT_FIRST ? "client" : "reader", action,

ll_count(cl->joblist), username(cl));

return 1;

}

/* pcsc doesn't like this; segfaults on x86, x86_64 */

int8_t modify_stacksize = 0;

struct s_reader *rdr = cl->reader;

if(cl->typ != 'r' || !rdr || rdr->typ != R_PCSC)

{ modify_stacksize = 1; }

if(action != ACTION_READER_CHECK_HEALTH)

{

cs_log_dbg(D_TRACE, "start %s thread action %d",

action > ACTION_CLIENT_FIRST ? "client" : "reader", action);

}

int32_t ret = start_thread("client work", work_thread, (void *)data, &cl->thread, 1, modify_stacksize);

if(ret)

{

cs_log("ERROR: can't create thread for %s (errno=%d %s)",

action > ACTION_CLIENT_FIRST ? "client" : "reader", ret, strerror(ret));

free_job_data(data);

}

cl->thread_active = 1;

SAFE_MUTEX_UNLOCK(&cl->thread_lock);

return 1;

}