path: root/util/shmctl.c

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>
#include <ctype.h>
#include <signal.h>
#include <sys/types.h>
#include <sys/shm.h>
#include <errno.h>
#include "config.h"
#include "pttstruct.h"
#include "util.h"
#include "proto.h"

extern struct utmpfile_t *utmpshm;
extern struct pttcache_t *ptt;
extern bcache_t *brdshm;

int logout_friend_online(userinfo_t *utmp)
{
    int i, j, k;
    int offset=(int) (utmp - &utmpshm->uinfo[0]);
    userinfo_t *ui;
    while(utmp->friendtotal){
    i = utmp->friendtotal-1;
    j = (utmp->friend_online[i] & 0xFFFFFF);
    if( !(0 <= j && j < MAX_ACTIVE) )
        printf("\tonline friend error(%d)\n", j);
    else{
        utmp->friend_online[i]=0;
        ui = &utmpshm->uinfo[j]; 
        if(ui->pid && ui!=utmp){
        for(k=0; k<ui->friendtotal && 
            (int)(ui->friend_online[k] & 0xFFFFFF) !=offset; k++);
        if(k<ui->friendtotal){
            ui->friendtotal--;
            ui->friend_online[k]=ui->friend_online[ui->friendtotal];
            ui->friend_online[ui->friendtotal]=0;
        }
        }
    }
    utmp->friendtotal--;
    utmp->friend_online[utmp->friendtotal]=0;
    }
    return 0;
}

void purge_utmp(userinfo_t *uentp)
{
    logout_friend_online(uentp);
    //memset(uentp, 0, sizeof(userinfo_t));
}

int utmpfix(int argc, char **argv)
{
    int     i, fast = 0;
    time_t  now, timeout = -1;
    char    *clean, buf[1024], ch;

    while( (ch = getopt(argc, argv, "nt:")) != -1 )
    switch( ch ){
    case 'n':
        fast = 1;
        break;
    case 't':
        timeout = atoi(optarg);
        break;
    default:
        printf("usage:\tshmctl\tutmpfix[-n] [-t timeout]\n");
        return 1;
    }

    time(&now);
    for( i = 0 ; i < 5 ; ++i )
    if( !utmpshm->busystate )
        break;
    else{
        puts("utmpshm is busy....");
        sleep(1);
    }
    printf("starting scaning... %s \n", (fast ? "(fast mode)" : ""));
    utmpshm->busystate = 1;
    for( i = 0 ; i < USHM_SIZE ; ++i )
    if( utmpshm->uinfo[i].pid ){
        clean = NULL;
        if( !isalpha(utmpshm->uinfo[i].userid[0]) )
        clean = "userid error";
        else if( kill(utmpshm->uinfo[i].pid, 0) < 0 ){
        clean = "process error";
        purge_utmp(&utmpshm->uinfo[i]);
        }
        else if( !fast ){
#ifdef DOTIMEOUT
        if( now - utmpshm->uinfo[i].lastact > 
            (timeout == -1 ? IDLE_TIMEOUT : timeout) ){
            sprintf(buf, "timeout(%s",
                ctime(&utmpshm->uinfo[i].lastact));
            buf[strlen(buf) - 1] = 0;
            strcat(buf, ")");
            clean = buf;
            kill(utmpshm->uinfo[i].pid, SIGHUP);
            purge_utmp(&utmpshm->uinfo[i]);
        }
        else
#endif
            if( searchuser(utmpshm->uinfo[i].userid) == 0 ){
            clean = "user not exist";
            } 
        }
        
        if( clean ){
        printf("clean %06d(%s), userid: %s\n",
               i, clean, utmpshm->uinfo[i].userid);
        memset(&utmpshm->uinfo[i], 0, sizeof(userinfo_t));
        }
        else if ( utmpshm->uinfo[i].mind > 40 ){
        printf("mind fix: %06d, userid: %s, mind: %d\n",
               i, utmpshm->uinfo[i].userid, utmpshm->uinfo[i].mind);
        utmpshm->uinfo[i].mind %= 40;
        }
    }
    utmpshm->busystate = 0;
    return 0;
}

char *CTIME(char *buf, time_t t)
{
    strcpy(buf, ctime(&t));
    buf[strlen(buf) - 1] = 0;
    return buf;
}

int utmpstate(int argc, char **argv)
{
    time_t  now;
    char    upbuf[64], nowbuf[64];
    now = time(NULL);
    CTIME(upbuf,  utmpshm->uptime);
    CTIME(nowbuf, time(NULL));
    printf("now:        %s\n", nowbuf);
    printf("currsorted: %d\n", utmpshm->currsorted);
    printf("uptime:     %s\n", upbuf);
    printf("number:     %d\n", utmpshm->number);
    printf("busystate:  %d\n", utmpshm->busystate);
    return 0;
}

int utmpreset(int argc, char **argv)
{
    utmpshm->busystate=0;
    utmpstate(0, NULL);
    return 0;
}

/* ulistsort */
static int cmputmpuserid(const void *i, const void *j){
    return strcasecmp((*((userinfo_t**)i))->userid,
              (*((userinfo_t**)j))->userid);
}

static int cmputmpmode(const void *i, const void *j){
    return (*((userinfo_t**)i))->mode-
    (*((userinfo_t**)j))->mode;
} 

static int cmputmpidle(const void *i, const void *j){
    return (*((userinfo_t**)i))->lastact-
    (*((userinfo_t**)j))->lastact;
} 

static int cmputmpfrom(const void *i, const void *j){
  return strcasecmp((*((userinfo_t**)i))->from,
            (*((userinfo_t**)j))->from);
} 

static int cmputmpfive(const void *i, const void *j){
    int type;
    if((type=(*((userinfo_t**)j))->five_win - (*((userinfo_t**)i))->five_win))
    return type;
    if((type=(*((userinfo_t**)i))->five_lose-(*((userinfo_t**)j))->five_lose))
    return type;
    return (*((userinfo_t**)i))->five_tie-(*((userinfo_t**)j))->five_tie;
} 

static int cmputmpsex(const void *i, const void *j){
    static int ladyfirst[]={1,0,1,0,1,0,3,3};
    return ladyfirst[(*((userinfo_t**)i))->sex]-
    ladyfirst[(*((userinfo_t**)j))->sex];
}
static int cmputmppid(const void *i, const void *j){
    return (*((userinfo_t**)i))->pid-(*((userinfo_t**)j))->pid;
}
static int cmputmpuid(const void *i, const void *j){
    return (*((userinfo_t**)i))->uid-(*((userinfo_t**)j))->uid;
}

int utmpsort(int argc, char **argv)
{
    time_t now=time(NULL);
    int count, i, ns;
    userinfo_t *uentp;
    
    if(now-utmpshm->uptime<60 && (now==utmpshm->uptime || utmpshm->busystate)){
    puts("lazy sort");
    //return; /* lazy sort */
    }
    utmpshm->busystate=1;
    utmpshm->uptime = now;
    ns=(utmpshm->currsorted?0:1);

    for(uentp = &utmpshm->uinfo[0], count=0, i=0;
    i< USHM_SIZE; i++,uentp = &utmpshm->uinfo[i])
    if(uentp->pid){
        utmpshm->sorted[ns][0][count++]= uentp;
        }
    utmpshm->number = count;
    qsort(utmpshm->sorted[ns][0],count,sizeof(userinfo_t*),cmputmpuserid);
    memcpy(utmpshm->sorted[ns][1],utmpshm->sorted[ns][0],
                         sizeof(userinfo_t *)*count);
    memcpy(utmpshm->sorted[ns][2],utmpshm->sorted[ns][0],
                         sizeof(userinfo_t *)*count);
    memcpy(utmpshm->sorted[ns][3],utmpshm->sorted[ns][0],
                         sizeof(userinfo_t *)*count);
    memcpy(utmpshm->sorted[ns][4],utmpshm->sorted[ns][0],
                         sizeof(userinfo_t *)*count);
    memcpy(utmpshm->sorted[ns][5],utmpshm->sorted[ns][0],
                         sizeof(userinfo_t *)*count);
    memcpy(utmpshm->sorted[ns][6],utmpshm->sorted[ns][0],
                         sizeof(userinfo_t *)*count);
    memcpy(utmpshm->sorted[ns][7],utmpshm->sorted[ns][0],
                         sizeof(userinfo_t *)*count);
    qsort(utmpshm->sorted[ns][1], count, sizeof(userinfo_t *), cmputmpmode );
    qsort(utmpshm->sorted[ns][2], count, sizeof(userinfo_t *), cmputmpidle );
    qsort(utmpshm->sorted[ns][3], count, sizeof(userinfo_t *), cmputmpfrom );
    qsort(utmpshm->sorted[ns][4], count, sizeof(userinfo_t *), cmputmpfive );
    //qsort(utmpshm->sorted[ns][5], count, sizeof(userinfo_t *), cmputmpsex );
    qsort(utmpshm->sorted[ns][6], count, sizeof(userinfo_t *), cmputmpuid );
    qsort(utmpshm->sorted[ns][7], count, sizeof(userinfo_t *), cmputmppid );
    utmpshm->currsorted=ns;
    utmpshm->busystate=0;

    puts("new utmpstate");
    utmpstate(0, NULL);
    return 0;
}
/* end of ulistsort */

#define TIMES   10
int utmpwatch(int argc, char **argv)
{
    int     i;
    while( 1 ){
    for( i = 0 ; i < TIMES ; ++i ){
        usleep(300);
        if( !utmpshm->busystate )
        break;
        puts("busying...");
    }
    if( i == TIMES ){
        puts("reset!");
        utmpshm->busystate = 0;
    }
    }
    return 0;
}

int utmpnum(int argc, char **argv)
{
    printf("%d.0\n", utmpshm->number);
    return 0;
}

int showglobe(int argc, char **argv)
{
    int     i;
    for( i = 0 ; i < 10 ; ++i )
    printf("GLOBE[%d] = %d\n", i, ptt->GLOBE[i]);
    return 0;
}

int setglobe(int argc, char **argv)
{
    int     where;
    if( argc != 3 )
    return 1;
    where = atoi(argv[1]);
    if( !(0 <= where && where <= 9) ){
    puts("only GLOBE[0] ~ GLOBE[9]");
    return 1;
    }
    printf("GLOBE[%d] = %d -> ", where, ptt->GLOBE[where]);
    printf("%d\n", ptt->GLOBE[where] = atoi(argv[2]));
    return 0;
}

int fixbfriend(int argc, char **argv)
{
    userinfo_t      *ptr;
    int     count, i;

    for( i = 0 ; i < MAX_BOARD ; ++i ){
    if( isdigit(brdshm->bcache[i].brdname[0]) ||
        isalpha(brdshm->bcache[i].brdname[0])    ){
        for( count = 0, ptr = brdshm->bcache[i].u ; 
         ptr != NULL && count < 256           ;
         ++count, ptr = ptr->nextbfriend        )
        ;
        printf("counting %s\n", brdshm->bcache[i].brdname);
        brdshm->bcache[i].nuser = ((count == 256) ? 0 : count);
    }
    }

    return 0;
}

struct {
    int     (*func)(int, char **);
    char    *cmd, *descript;
} cmd[] =
    { {utmpfix,    "utmpfix",    "clear dead userlist entry"},
      {utmpstate,  "utmpstate",  "list utmpstate"},
      {utmpreset,  "utmpreset",  "utmpshm->busystate=0"},
      {utmpsort,   "utmpsort",   "sort ulist"},
      {utmpwatch,  "utmpwatch",  "to see if busystate is always 1 then fix it"},
      {utmpnum,    "utmpnum",    "print utmpshm->number for snmpd"},
      {showglobe,  "showglobe",  "show GLOBE"},
      {setglobe,   "setglobe",   "set GLOBE"},
      {fixbfriend, "fixbfriend", "recount numbers of board friends"},
      {NULL, NULL, NULL} };

int main(int argc, char **argv)
{
    int     i = 0;
    
    if( argc >= 2 ){
    /* shmctl shouldn't create shm itself */
        int     shmid = shmget(UHASH_KEY, sizeof(uhash_t), 0);
    if( shmid < 0 ){
      printf("%d\n", errno);
        perror("attach utmpshm");
        return 1;
    }
    chdir(BBSHOME);
    resolve_utmp();
    resolve_boards();
    resolve_garbage();
    resolve_fcache();
    for( i = 0 ; cmd[i].func != NULL ; ++i )
        if( strcmp(cmd[i].cmd, argv[1]) == 0 ){
        return cmd[i].func(argc - 1, &argv[1]);
        }
    }
    if( argc == 1 || cmd[i].func == NULL ){
    /* usage */
    printf("usage: bbsctl [command] [options]\n");
    printf("commands:\n");
    for( i = 0 ; cmd[i].func != NULL ; ++i )
        printf("\t%-15s%s\n", cmd[i].cmd, cmd[i].descript);
    }
    return 0;
}