void syncutmp(int cfd) {
int i;
int like[MAX_FRIEND];
int hate[MAX_REJECT];
#ifdef UTMPLOG
int x=-1;
if(logfp && ftell(logfp)> 500*(1<<20)) {
fclose(logfp);
logfp=NULL;
}
if(logfp) fwrite(&x, sizeof(x), 1, logfp);
#endif
printf("logout all\n");
utmplogoutall();
fprintf(stderr,"sync begin\n");
for(i=0; i<USHM_SIZE; i++) {
int uid;
if( toread(cfd, &uid, sizeof(uid)) <= 0 ||
toread(cfd, like, sizeof(like)) <= 0 ||
toread(cfd, hate, sizeof(hate)) <= 0)
break;
#ifdef UTMPLOG
if(logfp) {
fwrite(&uid, sizeof(uid), 1, logfp);
fwrite(like, sizeof(like), 1, logfp);
fwrite(hate, sizeof(hate), 1, logfp);
}
#endif
if(uid != 0)
utmplogin(uid, i, like, hate);
}
if(i<USHM_SIZE) {
#ifdef UTMPLOG
int x=-2;
if(logfp) fwrite(&x, sizeof(x), 1, logfp);
#endif
}
fprintf(stderr,"sync end\n");
}
int main(int argc, char *argv[])
{
int fd, ret = 0;
int op = 0;
regmaildb_req req = {0};
if (argc < 4) {
fprintf(stderr, "Usage: %s operation userid email\n", argv[0]);
return 0;
}
if ( (fd = toconnect(REGMAILD_ADDR)) < 0 ) {
perror("toconnect");
return 1;
}
if (strcmp(argv[1], "count") == 0)
{
op = REGMAILDB_REQ_COUNT;
strlcpy(req.userid, argv[2], sizeof(req.userid));
strlcpy(req.email, argv[3], sizeof(req.email));
}
else if (strcmp(argv[1], "set") == 0)
{
op = REGMAILDB_REQ_SET;
strlcpy(req.userid, argv[2], sizeof(req.userid));
strlcpy(req.email, argv[3], sizeof(req.email));
}
else if (strcmp(argv[1], "amb") == 0)
{
op = REGCHECK_REQ_AMBIGUOUS;
strlcpy(req.userid, argv[2], sizeof(req.userid));
strlcpy(req.email, "*****@*****.**", sizeof(req.email));
}
else
return 0;
req.cb = sizeof(req);
req.operation = op;
if (towrite(fd, &req, sizeof(req)) != sizeof(req)) {
perror("towrite");
return 1;
}
if (toread(fd, &ret, sizeof(ret)) != sizeof(ret)) {
perror("toread");
return 1;
}
printf("result: %d\n", ret);
return 0;
}
static void
client_cb(int fd, short event, void *arg)
{
struct event *ev = (struct event*) arg;
regmaildb_req req = {0};
int ret = -1;
assert(ev);
if ( (event & EV_TIMEOUT) ||
!(event & EV_READ) ||
toread(fd, &req, sizeof(req)) != sizeof(req) ||
req.cb != sizeof(req))
{
fprintf(stderr, "error: corrupted request.\r\n");
close(fd);
free(ev);
return;
}
switch(req.operation)
{
case REGMAILDB_REQ_COUNT:
if (!*req.userid || !*req.email) {
err_request(&req, fd, ev);
return;
}
ret = regmaildb_check_email(req.email, strlen(req.email), req.userid);
fprintf(stderr, "%-*s check mail (result: %d): [%s]\r\n",
IDLEN, req.userid, ret, req.email);
if (towrite(fd, &ret, sizeof(ret)) != sizeof(ret))
{
fprintf(stderr, " error: cannot write response...\r\n");
}
break;
case REGMAILDB_REQ_SET:
if (!*req.userid || !*req.email) {
err_request(&req, fd, ev);
return;
}
ret = regmaildb_update_email(req.userid, strlen(req.userid),
req.email, strlen(req.email));
fprintf(stderr, "%-*s UPDATE mail (result: %d): [%s]\r\n",
IDLEN, req.userid, ret, req.email);
if (towrite(fd, &ret, sizeof(ret)) != sizeof(ret))
{
fprintf(stderr, " error: cannot write response...\r\n");
}
break;
case REGCHECK_REQ_AMBIGUOUS:
ret = regcheck_ambiguous_id(req.userid);
fprintf(stderr, "%-*s check ambiguous id exist (result: %d)\r\n",
IDLEN, req.userid, ret);
if (towrite(fd, &ret, sizeof(ret)) != sizeof(ret))
{
fprintf(stderr, " error: cannot write response...\r\n");
}
break;
default:
fprintf(stderr, "error: invalid operation: %d.\r\n", req.operation);
close(fd);
free(ev);
return;
}
// close connection anyway
close(fd);
free(ev);
}
int main(int argc, char *argv[])
{
int fd;
char buf[64];
Signal(SIGPIPE, SIG_IGN);
memset(buf, 0, sizeof(buf));
if (argc < 2) {
fprintf(stderr, "Usage: %s tunnel_path\n", argv[0]);
return 0;
}
if ( (fd = toconnect(argv[1])) < 0 ) {
perror("toconnect");
return 1;
}
puts("start waiting!\n");
while (1)
{
int xfd = 0, i;
const char *encoding = "";
login_data dat = {0};
if ((xfd = recv_remote_fd(fd, argv[1])) < 0)
{
fprintf(stderr, "recv_remote_fd error. abort.\r\n");
break;
}
puts("got recv_remote_fd");
if (toread(fd, &dat, sizeof(dat)) <= 0)
{
fprintf(stderr, "toread error. abort.\r\n");
break;
}
if (towrite(fd, &dat.ack, sizeof(dat.ack)) <= 0) {
fprintf(stderr, "towrite error. abort.\r\n");
break;
}
#ifdef CONVERT
switch (dat.encoding)
{
case CONV_UTF8:
encoding = "[UTF-8] ";
break;
}
#endif
fprintf(stderr, "got login data: userid=%s, (%dx%d) %sfrom: %s\r\n",
dat.userid, dat.t_cols, dat.t_lines,
encoding, dat.hostip);
if (xfd != 0) dup2(xfd, 0);
if (xfd != 1) dup2(xfd, 1);
// write something to user!
printf("\r\nwelcome, %s from %s! greetings from loginc!\r\n", dat.userid, dat.hostip);
printf("please give me 3 key hits to test i/o!\r\n");
for (i = 0; i < 3; i++)
{
char c;
read(0, &c, sizeof(c));
printf("you hit %02X\r\n", c);
}
printf("\r\ntest complete. connection closed.\r\n");
close(0);
close(1);
if (xfd != 0 && xfd != 1) close(xfd);
}
return 0;
}
int main(int argc, char *argv[])
{
int fd;
angel_beats_data req = {0};
angel_beats_report rpt = {0};
angel_beats_uid_list list = {0};
attach_SHM();
if (argc < 2) {
fprintf(stderr, "Usage: %s operation [uid]\n", argv[0]);
return 0;
}
if ( (fd = toconnect(ANGELBEATS_ADDR)) < 0 ) {
perror("toconnect");
return 1;
}
// start commands
if (strcmp(argv[1], "reload") == 0) {
req.operation = ANGELBEATS_REQ_RELOAD;
} else if (strcmp(argv[1], "suggest") == 0) {
req.operation = ANGELBEATS_REQ_SUGGEST;
if (argc > 2) {
req.operation = ANGELBEATS_REQ_SUGGEST_AND_LINK;
req.master_uid = searchuser(argv[2], NULL);
}
} else if (strcmp(argv[1], "unlink") == 0) {
if (argc != 3) {
printf("need target id.\n");
return -1;
}
req.operation = ANGELBEATS_REQ_REMOVE_LINK;
req.master_uid = 0; // anyone, just unlink
req.angel_uid = searchuser(argv[2], NULL);
if (!req.angel_uid) {
printf("invalid user id: %s\n", argv[2]);
return -1;
}
} else if (strcmp(argv[1], "report") == 0) {
req.operation = ANGELBEATS_REQ_REPORT;
if (argc > 2 && !(req.angel_uid = searchuser(argv[2], NULL))) {
printf("invalid user id: %s\n", argv[2]);
return -1;
}
} else if (strcmp(argv[1], "list") == 0) {
req.operation = ANGELBEATS_REQ_GET_ONLINE_LIST;
} else if (strcmp(argv[1], "save") == 0) {
req.operation = ANGELBEATS_REQ_SAVE_STATE;
} else if (strcmp(argv[1], "perf") == 0) {
req.operation = ANGELBEATS_REQ_EXPORT_PERF;
} else {
fprintf(stderr, "Sorry, unknown command: %s\n", argv[1]);
return 0;
}
req.cb = sizeof(req);
if (towrite(fd, &req, sizeof(req)) != sizeof(req)) {
perror("towrite");
return 1;
}
if (req.operation == ANGELBEATS_REQ_REPORT) {
if (toread(fd, &rpt, sizeof(rpt)) != sizeof(rpt)) {
perror("toread");
return 1;
}
assert(rpt.cb == sizeof(rpt));
printf("total_angels=%d\n"
"total_online_angels=%d\n"
"total_active_angels=%d\n"
"min_masters_of_online_angels=%d\n"
"max_masters_of_online_angels=%d\n"
"min_masters_of_active_angels=%d\n"
"max_masters_of_active_angels=%d\n"
"my_index=%d\n"
"my_active_index=%d\n"
"my_active_masters=%d\n",
rpt.total_angels,
rpt.total_online_angels,
rpt.total_active_angels,
rpt.min_masters_of_online_angels,
rpt.max_masters_of_online_angels,
rpt.min_masters_of_active_angels,
rpt.max_masters_of_active_angels,
rpt.my_index,
rpt.my_active_index,
rpt.my_active_masters);
} else if (req.operation == ANGELBEATS_REQ_GET_ONLINE_LIST) {
int i;
if (toread(fd, &list, sizeof(list)) != sizeof(list)) {
perror("toread");
return 1;
}
assert(list.cb == sizeof(list));
printf("angels=%d\n", list.angels);
for (i = 0; i < list.angels; i++) {
printf("%3d. %s\n", i + 1, getuserid(list.uids[i]));
}
} else {
if (toread(fd, &req, sizeof(req)) != sizeof(req)) {
perror("toread");
return 1;
}
printf("result: angel_uid=%d\n", req.angel_uid);
}
return 0;
}