/* $Id: shmctl.c,v 1.21 2002/06/13 15:10:50 in2 Exp $ */ #include "bbs.h" extern SHM_t *SHM; int logout_friend_online(userinfo_t *utmp) { int i, j, k; int offset=(int) (utmp - &SHM->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 = &SHM->uinfo[j]; if(ui->pid && ui!=utmp){ for(k=0; kfriendtotal && (int)(ui->friend_online[k] & 0xFFFFFF) !=offset; k++); if(kfriendtotal){ 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( !SHM->UTMPbusystate ) break; else{ puts("utmpshm is busy...."); sleep(1); } printf("starting scaning... %s \n", (fast ? "(fast mode)" : "")); SHM->UTMPbusystate = 1; for( i = 0 ; i < USHM_SIZE ; ++i ) if( SHM->uinfo[i].pid ){ clean = NULL; if( !isalpha(SHM->uinfo[i].userid[0]) ) clean = "userid error"; else if( kill(SHM->uinfo[i].pid, 0) < 0 ){ clean = "process error"; purge_utmp(&SHM->uinfo[i]); } else if( !fast ){ #ifdef DOTIMEOUT if( now - SHM->uinfo[i].lastact > (timeout == -1 ? IDLE_TIMEOUT : timeout) ){ sprintf(buf, "timeout(%s", ctime(&SHM->uinfo[i].lastact)); buf[strlen(buf) - 1] = 0; strcat(buf, ")"); clean = buf; kill(SHM->uinfo[i].pid, SIGHUP); purge_utmp(&SHM->uinfo[i]); } else #endif if( searchuser(SHM->uinfo[i].userid) == 0 ){ clean = "user not exist"; } } if( clean ){ printf("clean %06d(%s), userid: %s\n", i, clean, SHM->uinfo[i].userid); memset(&SHM->uinfo[i], 0, sizeof(userinfo_t)); } else if ( SHM->uinfo[i].mind > 40 ){ printf("mind fix: %06d, userid: %s, mind: %d\n", i, SHM->uinfo[i].userid, SHM->uinfo[i].mind); SHM->uinfo[i].mind %= 40; } } SHM->UTMPbusystate = 0; return 0; } char *CTIMEx(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); CTIMEx(upbuf, SHM->UTMPuptime); CTIMEx(nowbuf, time(NULL)); printf("now: %s\n", nowbuf); printf("currsorted: %d\n", SHM->currsorted); printf("uptime: %s\n", upbuf); printf("number: %d\n", SHM->UTMPnumber); printf("busystate: %d\n", SHM->UTMPbusystate); return 0; } int utmpreset(int argc, char **argv) { SHM->UTMPbusystate=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-SHM->UTMPuptime<60 && (now==SHM->UTMPuptime || SHM->UTMPbusystate)){ puts("lazy sort"); //return; /* lazy sort */ } SHM->UTMPbusystate=1; SHM->UTMPuptime = now; ns=(SHM->currsorted?0:1); for(uentp = &SHM->uinfo[0], count=0, i=0; i< USHM_SIZE; i++,uentp = &SHM->uinfo[i]) if(uentp->pid){ SHM->sorted[ns][0][count++]= uentp; } SHM->number = count; qsort(SHM->sorted[ns][0],count,sizeof(userinfo_t*),cmputmpuserid); memcpy(SHM->sorted[ns][1],SHM->sorted[ns][0], sizeof(userinfo_t *)*count); memcpy(SHM->sorted[ns][2],SHM->sorted[ns][0], sizeof(userinfo_t *)*count); memcpy(SHM->sorted[ns][3],SHM->sorted[ns][0], sizeof(userinfo_t *)*count); memcpy(SHM->sorted[ns][4],SHM->sorted[ns][0], sizeof(userinfo_t *)*count); memcpy(SHM->sorted[ns][5],SHM->sorted[ns][0], sizeof(userinfo_t *)*count); memcpy(SHM->sorted[ns][6],SHM->sorted[ns][0], sizeof(userinfo_t *)*count); memcpy(SHM->sorted[ns][7],SHM->sorted[ns][0], sizeof(userinfo_t *)*count); qsort(SHM->sorted[ns][1], count, sizeof(userinfo_t *), cmputmpmode ); qsort(SHM->sorted[ns][2], count, sizeof(userinfo_t *), cmputmpidle ); qsort(SHM->sorted[ns][3], count, sizeof(userinfo_t *), cmputmpfrom ); qsort(SHM->sorted[ns][4], count, sizeof(userinfo_t *), cmputmpfive ); //qsort(SHM->sorted[ns][5], count, sizeof(userinfo_t *), cmputmpsex ); qsort(SHM->sorted[ns][6], count, sizeof(userinfo_t *), cmputmpuid ); qsort(SHM->sorted[ns][7], count, sizeof(userinfo_t *), cmputmppid ); SHM->currsorted=ns; SHM->UTMPbusystate=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( !SHM->UTMPbusystate ) break; puts("busying..."); } if( i == TIMES ){ puts("reset!"); SHM->UTMPbusystate = 0; } } return 0; } int utmpnum(int argc, char **argv) { printf("%d.0\n", SHM->UTMPnumber); return 0; } int showglobal(int argc, char **argv) { int i; for( i = 0 ; i < 10 ; ++i ) printf("GLOBALVAR[%d] = %d\n", i, SHM->GLOBALVAR[i]); return 0; } int setglobal(int argc, char **argv) { int where; if( argc != 3 ) return 1; where = atoi(argv[1]); if( !(0 <= where && where <= 9) ){ puts("only GLOBALVAR[0] ~ GLOBALVAR[9]"); return 1; } printf("GLOBALVAR[%d] = %d -> ", where, SHM->GLOBALVAR[where]); printf("%d\n", SHM->GLOBALVAR[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(SHM->bcache[i].brdname[0]) || isalpha(SHM->bcache[i].brdname[0]) ){ for( count = 0, ptr = SHM->bcache[i].u ; ptr != NULL && count < 256 ; ++count, ptr = ptr->nextbfriend ) ; printf("counting %s\n", SHM->bcache[i].brdname); SHM->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", "SHM->busystate=0"}, {utmpsort, "utmpsort", "sort ulist"}, {utmpwatch, "utmpwatch", "to see if busystate is always 1 then fix it"}, {utmpnum, "utmpnum", "print SHM->number for snmpd"}, {showglobal, "showglobal", "show GLOBALVAR[]"}, {setglobal, "setglobal", "set GLOBALVAR[]"}, {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(SHM_KEY, sizeof(SHM_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; }