w.cpp

#include <stdio.h>
#include <pthread.h>
#include <unistd.h>
#include <curses.h>
#include <string.h>

#include "w.h"


pthread_t tid_http_log;
pthread_t tid_http_stat;
pthread_mutex_t _lock_L1;
pthread_mutex_t _lock_R1;

#define CMD_HTTP_LOG "tail -n 1 /usr/local/httpd/logs/error_log"
#define FILE_HTTP_LOG "/usr/local/httpd/logs/access_log"
#define CMD_HTTP_STAT "sh showlogerror.sh"

static int running = 1;
static Win *win_L1 = NULL;
static Win *win_R1 = NULL;


void update_win(Win *win, char *log, int clean)
{
    int len = 0;
    int i = 0;

    len = strlen(log);
    for (i=len - 1; i>0 && len>0; i--) {
        if (log[i] == '\n' || log[i] == '\r') {
            log[i] = '\0';
            continue;
        }
        break;
    }
    if (NULL == win) {
        printf("%s[%d]\n", log, (int)strlen(log));
        return;
    }

    len = strlen(log);
    if (len >= win->locate.w) {
        log[win->locate.w] = '\0';
    }

    if (clean) {
        wclear(win->win);
    }

    wprintw(win->win, "%s\n", log);
    wrefresh(win->win);
}

void lock_R1()
{
    pthread_mutex_lock(&_lock_R1);
}

void unlock_R1()
{
    pthread_mutex_unlock(&_lock_R1);
}
void lock_L1()
{
    pthread_mutex_lock(&_lock_L1);
}

void unlock_L1()
{
    pthread_mutex_unlock(&_lock_L1);
}

void *http_log(void *arg)
{
    char *ret = NULL;
    FILE *fp = NULL;
    char buf[1024] = {0};\
    int lastpos = 0;

    // TODO: Read from config file
    fp = fopen(FILE_HTTP_LOG, "r");
    fseek(fp, lastpos, SEEK_END);
    lastpos = ftell(fp);
    fclose(fp);
    fp = NULL;

    while(running) {
        if (NULL == fp) {
            usleep(100);
            fp = fopen(FILE_HTTP_LOG, "r");
            fseek(fp, lastpos, SEEK_SET);
        }
        ret = fgets(buf, sizeof(buf), fp);
        lastpos = ftell(fp);
        // TODO:Save to config file
        if (NULL == ret) {
            fclose(fp);
            fp = NULL;
            continue;
        }
        lock_L1();
        update_win(win_L1, buf, 0);
        unlock_L1();
    }
    
/*
    printf("thread\n");
    fp = popen(CMD_HTTP_LOG, "r");
    while(running) {
        fread(buf, sizeof(char), sizeof(buf), fp);
        printf("%s\n", buf);
    }
*/
/*
    char buffer[1024] = {0};
    FILE *fp = NULL;
    int fd = 0;
    int wd = 0;
    fp = fopen(CMD_HTTP_LOG, "r");
    if (fp < 0) {
        goto quit;
    }  
    fd = inotify_init();
    wd = inotify_add_watch(fd, CMD_HTTP_LOG, IN_MODIFY);
    while(running) {
        read(fp, buffer, sizeof(buffer));  
        printf("%s\n", buffer);
        pthread_mutex_lock(&lock_http);
        pthread_mutex_unlock(&lock_http);
    }
quit:
*/

    return (void *)0;
}

void *http_stat(void *arg)
{
    FILE *fp = NULL;
    char buf[256] = {0};
    int ret = 0;

    while(running) {
        fp = popen(CMD_HTTP_STAT, "r");
        ret = fread(buf, sizeof(char), sizeof(buf), fp);
        lock_R1();
        update_win(win_R1, buf, 1);
        unlock_R1();
        pclose(fp);
        fp = NULL;
        sleep(10);
    }

    return (void *)0;
}

void wait_work_thread()
{
    int ret = 0;
    void *tret;
    
    ret = pthread_join(tid_http_log, &tret);
    if (ret != 0) {
    }
}

int start_work_thread(Win *L1, Win *R1)
{
    int ret = 0;

    win_L1 = L1;
    win_R1 = R1;
    
    start_http_log();
    start_http_stat();
    start_free_mem();
    
    return ret;
}

int start_http_log()
{
    int ret = 0;
    
    ret = pthread_create(&tid_http_log, NULL, http_log, NULL);
    
    return ret;
}

int start_http_stat()
{
    int ret = 0;

    ret = pthread_create(&tid_http_stat, NULL, http_stat, NULL);
    
    return ret;
}

int start_free_mem()
{
    int ret = 0;
    
    return ret;
}

#ifdef DEBUG_MAIN
int main()
{  
    start_work_thread(NULL, NULL);
    
    sleep(3);
    running = 0;
    
    // wait_work_thread();
    
    return 0;
}
#endif