menu.c

   1 #include <stdio.h>
   2 #include <stdlib.h>
   3 #include <stdint.h>
   4 #include <string.h>
   5 #include <sys/socket.h>
   6 #include <sys/stat.h>
   7 #include <pthread.h>
   8 #include <time.h>
   9 #include "telnet.h"
  10 #include "users.h"
  11 #include "games.h"
  12 #include "server.h"
  13 #include "run.h"
  14 #include "log.h"
  15
  16
  17 extern server_config config;
  18
  19 extern session_info *sessionlist;
  20 extern pthread_mutex_t sessionlist_mutex;
  21
  22 #define BUF_SIZE        1 * 1024
  23
  24
  25 // return 1 if enter is pressed
  26 static int input (char c, char *str, int *pos, int len, char *out, int *out_pos)
  27 {
  28         // enter
  29         if (c == '\n')
  30         {
  31                 str[*pos] = 0;
  32                 out[*out_pos] = 0;
  33                 return 1;
  34         }
  35
  36         // backspace
  37         if (c == 8 || c == 127)
  38         {
  39                 if (*pos > 0) *pos -= 1;
  40
  41                 out[*out_pos] = 8;
  42                 if (*out_pos < len) *out_pos += 1;
  43         }
  44         else if (*pos < len - 1)
  45         {
  46                 str[*pos] = c;
  47                 if (*pos < len) *pos += 1;
  48
  49                 out[*out_pos] = c;
  50                 if (*out_pos < len) *out_pos += 1;
  51         }
  52
  53         return 0;
  54 }
  55
  56
  57
  58 static int recv_input (session_info *sess, char *inp, int len)
  59 {
  60         char buf[BUF_SIZE], out[BUF_SIZE];
  61         int pos = 0;
  62
  63
  64         while (1)
  65         {
  66                 int opos = 0;
  67
  68                 int n = recv(sess->socket, buf, BUF_SIZE, 0);
  69                 if (n == -1) return -1;
  70
  71                 n = translate_telnet_input(buf, n, sess);
  72                 if (n > 0)
  73                 {
  74                         int i;
  75                         for (i = 0; i < n; i++)
  76                         {
  77                                 if (input(buf[i], inp, &pos, len, out, &opos))
  78                                 {
  79                                         return pos;
  80                                 }
  81                         }
  82                 }
  83
  84                 // echo
  85                 if (send(sess->socket, out, opos, 0) == -1) return -1;
  86         }
  87 }
  88
  89
  90
  91 static int print_and_wait (session_info *sess, const char *msg)
  92 {
  93         char buf[BUF_SIZE];
  94
  95
  96         if (send(sess->socket, msg, strlen(msg) + 1, 0) == -1) return -1;
  97
  98         while (1)
  99         {
 100                 // wait for a keypress
 101                 int n = recv(sess->socket, buf, BUF_SIZE, 0);
 102                 if (n == -1) return -1;
 103
 104                 if (translate_telnet_input(buf, n, sess) > 0) break;
 105         }
 106
 107         return 0;
 108 }
 109
 110
 111 #define REG_INPUTS      3
 112
 113 static int register_menu (session_info *sess)
 114 {
 115         char data[REG_INPUTS][BUF_SIZE];
 116         int id, r = 0;
 117         const char *const prompts[REG_INPUTS] =
 118         {
 119                 "\r\nUser name (3-16 chars, alphabet and digits only, start with a letter): ",
 120                 "\r\nPassword (4-16 chars, it is INSECURE, so do not use the same as email): ",
 121                 "\r\nEmail (optional): "
 122         };
 123         int lengths[REG_INPUTS][2] = { {MIN_USER_NAME_LEN, USER_NAME_LEN}, {MIN_USER_PASS_LEN, USER_PASS_LEN}, {0, USER_EMAIL_LEN} };
 124         char buf[BUF_SIZE];
 125         user usr;
 126
 127
 128         // ask user for data
 129         while (r < REG_INPUTS)
 130         {
 131                 int len;
 132
 133                 if (send(sess->socket, prompts[r], strlen(prompts[r]), 0) == -1) return -1;
 134
 135                 len = recv_input(sess, data[r], lengths[r][1]);
 136                 if (len < 0) return -1;
 137
 138                 if (len < lengths[r][0])
 139                 {
 140                         int l = sprintf(buf, "\r\nminimum %d symbols, try again: ", lengths[r][0]);
 141                         if (send(sess->socket, buf, l, 0) == -1) return -1;
 142                         continue;
 143                 }
 144
 145                 r++;
 146         }
 147
 148
 149         // try to register using the data
 150         id = add_user(&usr, data[0], data[1], data[2]);
 151         if (id > 0)
 152         {
 153                 int l;
 154
 155                 sess->user_id = usr.id;
 156                 strcpy (sess->user_name, usr.name);
 157
 158                 write_log (LOG_DEBUG, "registered user %s", usr.name);
 159
 160                 l = sprintf(buf, "\r\nRegistration succeded\r\n");
 161
 162                 if (send(sess->socket, buf, l, 0) == -1) return -1;
 163         }
 164         else
 165         {
 166                 int l;
 167                 if (id == -1)
 168                 {
 169                         l = sprintf(buf, "\r\nInvalid name\r\n");
 170                 }
 171                 else if (id == -4)
 172                 {
 173                         l = sprintf(buf, "\r\nUser with such name already exists\r\n");
 174                 }
 175                 else
 176                 {
 177                         write_log (LOG_ERROR, "user registration failed, error %d", id);
 178
 179                         l = sprintf(buf, "\r\nRegistration failed\r\n");
 180                 }
 181
 182                 if (send(sess->socket, buf, l, 0) == -1) return -1;
 183         }
 184
 185         // wait for a keypress
 186         int n = recv(sess->socket, buf, BUF_SIZE, 0);
 187         if (n == -1) return -1;
 188
 189         return 0;
 190 }
 191
 192
 193
 194 static int login_menu (session_info *sess)
 195 {
 196         const char login_prompt[] = "\r\nUser name: ";
 197         const char pass_prompt[] = "\r\nPassword: ";
 198         char name[USER_NAME_LEN], pass[USER_PASS_LEN];
 199         int id = 0;
 200         user usr;
 201
 202
 203         if (send(sess->socket, login_prompt, sizeof(login_prompt) - 1, 0) == -1) return -1;
 204         if (recv_input(sess, name, USER_NAME_LEN) < 0) return -1;
 205
 206         if (send(sess->socket, pass_prompt, sizeof(pass_prompt) - 1, 0) == -1) return -1;
 207         if (recv_input(sess, pass, USER_PASS_LEN) < 0) return -1;
 208
 209         // check user name and password
 210         if (check_user_login(&usr, name, pass) < 0)
 211         {
 212                 if (print_and_wait(sess, "\r\nLogin failed\r\n") == -1) return -1;
 213                 return -2;
 214         }
 215
 216         // disallow simultaneous connections
 217         if (user_is_connected(usr.id))
 218         {
 219                 if (print_and_wait(sess, "\r\nYou are already logged in\r\n") == -1) return -1;
 220                 return -3;
 221         }
 222
 223         sess->user_id = usr.id;
 224         strcpy (sess->user_name, usr.name);
 225         write_log (LOG_DEBUG, "user %s logs in", usr.name);
 226
 227         return id;
 228 }
 229
 230
 231
 232 static int passwd_menu (session_info *sess)
 233 {
 234         const char old_prompt[] = "\r\nOld Password: ";
 235         const char new_prompt[] = "\r\nNew Password: ";
 236         char pass[USER_PASS_LEN];
 237         user usr;
 238         int len;
 239
 240
 241         // ask for old password first
 242         if (send(sess->socket, old_prompt, sizeof(old_prompt) - 1, 0) == -1) return -1;
 243         if (recv_input(sess, pass, USER_PASS_LEN) < 0) return -1;
 244
 245         // check old password
 246         if (check_user_login(&usr, sess->user_name, pass) < 0)
 247         {
 248                 if (print_and_wait(sess, "\r\nWrong password\r\n") == -1) return -1;
 249                 return -2;
 250         }
 251
 252         // ask for new password
 253         if (send(sess->socket, new_prompt, sizeof(new_prompt) - 1, 0) == -1) return -1;
 254         len = recv_input(sess, pass, USER_PASS_LEN);
 255         if (len < 0) return -1;
 256
 257         if (len < MIN_USER_PASS_LEN)
 258         {
 259                 if (print_and_wait(sess, "\r\nNew password is too short\r\n") == -1) return -1;
 260                 return -3;
 261         }
 262
 263         if (change_password(sess->user_id, pass) == 0)
 264         {
 265                 if (print_and_wait(sess, "\r\nPassword has been changed\r\n") == -1) return -1;
 266         }
 267
 268         return 0;
 269 }
 270
 271
 272
 273 static int observe (session_info *sess)
 274 {
 275         uint8_t buf[BUF_SIZE];
 276
 277
 278         while (1)
 279         {
 280                 int n = recv(sess->socket, buf, BUF_SIZE, 0);
 281                 if (n == -1) return -1;
 282
 283                 if (buf[0] == 'q') break;
 284         }
 285
 286         return 0;
 287 }
 288
 289
 290
 291 static int observer_menu (session_info *sess)
 292 {
 293         char header[] = TELNET_CLS "\tGame\t\t\t\tUser\t\t  Idle\t\tTerminal\r\n";
 294         char buf[BUF_SIZE];
 295         session_info *s;
 296         int n, l, gn, count = 0;
 297         time_t now;
 298
 299         char inp[20];
 300
 301
 302         if (send(sess->socket, header, sizeof(header) - 1, 0) == -1) return -1;
 303
 304         pthread_mutex_lock (&sessionlist_mutex);
 305         now = time(NULL);
 306         for (s = sessionlist; s != NULL; s = s->next)
 307         {
 308                 int idle;
 309
 310                 if (s->game[0] == 0 || s->observer_count == -1) continue;
 311
 312                 idle = now - s->last_activity;
 313
 314                 l = sprintf(buf, "%3d\t%s", count + 1, s->game);
 315                 for (; l < 4 * 8; l++) buf[l] = ' ';
 316
 317                 l += sprintf(&buf[l], "\t%s", s->user_name);
 318                 for (; l < 6 * 8; l++) buf[l] = ' ';
 319
 320                 l += sprintf(&buf[l], "\t%3dm %02ds\t%dx%d\r\n", idle / 60, idle % 60, s->term_wid, s->term_hgt);
 321                 if (send(sess->socket, buf, l, 0) == -1) return -1;
 322                 count++;
 323         }
 324         pthread_mutex_unlock (&sessionlist_mutex);
 325         l = sprintf(buf, "\r\n%d games total\r\n\r\n"
 326                 "Input number to observe or anything else to return to the main menu\r\n"
 327                 "You can press 'q' anytime to stop observing\r\n", count);
 328         if (send(sess->socket, buf, l, 0) == -1) return -1;
 329
 330
 331         // get number
 332         n = recv_input(sess, inp, 20);
 333         if (n == -1) return -1;
 334
 335         gn = atoi(inp);
 336         if (gn > 0 && gn <= count)
 337         {
 338                 // try to observe the specified game
 339                 if (add_observer(gn - 1, sess->socket))
 340                 {
 341                         int r = observe(sess);
 342                         stop_observing (sess);
 343                         if (r == -1) return -1;
 344                 }
 345         }
 346
 347         return 0;
 348 }
 349
 350
 351
 352 // if score == 1, display highscore list and exit rather than playing
 353 static int play_menu (session_info *sess, int score)
 354 {
 355         const char msg[] = TELNET_CLS "Select a game by entering its id\r\n";
 356         game_info *gamelist, *g;
 357         char buf[BUF_SIZE];
 358         int n;
 359         char inp[20];
 360
 361
 362         gamelist = load_gamelist();
 363
 364         if (send(sess->socket, msg, sizeof(msg), 0) == -1)
 365         {
 366                 free_gamelist (gamelist);
 367                 return -1;
 368         }
 369
 370         for (g = gamelist; g != NULL; g = g->next)
 371         {
 372                 if (score && !g->cmd_score[0]) continue;
 373
 374                 int l = sprintf(buf, "%8s: %s\r\n", g->id, g->name);
 375
 376                 if (send(sess->socket, buf, l, 0) == -1)
 377                 {
 378                         free_gamelist (gamelist);
 379                         return -1;
 380                 }
 381         }
 382
 383
 384         // get id
 385         n = recv_input(sess, inp, 20);
 386         if (n == -1) return -1;
 387
 388         // find and run game with such id
 389         for (g = gamelist; g != NULL; g = g->next)
 390         {
 391                 if (score && !g->cmd_score[0]) continue;
 392
 393                 if (strcmp(inp, g->id) == 0)
 394                 {
 395                         // hack for displaying score
 396                         if (score)
 397                         {
 398                                 sess->observer_count = -1;
 399                                 strcpy (g->cmd, g->cmd_score);
 400                         }
 401
 402                         run_game (sess, g);
 403                         free_gamelist (gamelist);
 404
 405                         if (print_and_wait(sess, "\r\nYou have left the game\r\n") == -1) return -1;
 406
 407                         if (sess->observer_count == -1) sess->observer_count = 0;
 408
 409                         return 1;
 410                 }
 411         }
 412
 413         free_gamelist (gamelist);
 414
 415         return 0;
 416 }
 417
 418
 419
 420 static void run_editor (session_info *sess, const game_info *g, int i)
 421 {
 422         game_info ed;
 423         char name[BUF_SIZE];
 424         int dirln;
 425
 426
 427         // prefix with game directory name
 428         if (g->dir[0]) dirln = sprintf(name, "%s/", g->dir);
 429         else dirln = 0;
 430
 431         // copy file name to bufer
 432         strcat (&name[dirln], g->files[i].copy);
 433
 434         // set up a a game struct for editor
 435         memset (&ed, 0, sizeof(game_info));
 436         strcpy (ed.id, "editor");
 437         strcpy (ed.name, "editor");
 438         ed.enter = '\n';
 439         snprintf (ed.cmd, GAME_COMMAND_LEN, "%s %s", config.editor, name);
 440
 441         // disable observing
 442         sess->observer_count = -1;
 443
 444         run_game (sess, &ed);
 445
 446         // enable observing again
 447         sess->observer_count = 0;
 448 }
 449
 450
 451
 452 static int edit_menu (session_info *sess)
 453 {
 454         const char msg[] = TELNET_CLS "Select a file to edit by entering its number\r\n";
 455         game_info *gamelist, *g;
 456         char buf[BUF_SIZE], name[BUF_SIZE];
 457         int n, idx = 1;
 458         char inp[20];
 459
 460
 461         gamelist = load_gamelist();
 462
 463         if (send(sess->socket, msg, sizeof(msg), 0) == -1)
 464         {
 465                 free_gamelist (gamelist);
 466                 return -1;
 467         }
 468
 469         g = gamelist;
 470         while (g != NULL)
 471         {
 472                 struct stat st;
 473                 int l, i, dirln = 0;
 474
 475                 for (i = 0; i < GAME_FILES; i++)
 476                 {
 477                         if (!g->files[i].file[0]) break;
 478                         if (!g->files[i].allow_edit) continue;
 479
 480                         // prefix with game directory name, expanding user name
 481                         if (g->dir[0])
 482                         {
 483                                 str_replace (name, g->dir, "$USER$", sess->user_name);
 484                                 strcat (name, "/");
 485                                 dirln = strlen(name);
 486                         }
 487
 488                         // copy file name to bufer, expanding user name
 489                         str_replace (&name[dirln], g->files[i].copy, "$USER$", sess->user_name);
 490                         // file does not exist
 491                         if (stat(name, &st) != 0) continue;
 492
 493                         l = sprintf(buf, "%3d: [%s] %s\r\n", idx, g->name, name);
 494
 495                         if (send(sess->socket, buf, l, 0) == -1)
 496                         {
 497                                 free_gamelist (gamelist);
 498                                 return -1;
 499                         }
 500
 501                         // we (ab)use the allow_edit to store index
 502                         // to find the file when the user enters a number
 503                         g->files[i].allow_edit = 0x80000000 | idx;
 504
 505                         idx++;
 506                 }
 507
 508                 g = g->next;
 509         }
 510
 511
 512         // get id
 513         n = recv_input(sess, inp, 20);
 514         if (n == -1) return -1;
 515
 516         n = 0x80000000 | atoi(inp);
 517
 518         // find and run game with such id
 519         g = gamelist;
 520         while (g != NULL)
 521         {
 522                 int i;
 523
 524                 for (i = 0; i < GAME_FILES; i++)
 525                 {
 526                         if (!g->files[i].file[0]) break;
 527                         if (g->files[i].allow_edit != n) continue;
 528
 529                         run_editor (sess, g, i);
 530
 531                         free_gamelist (gamelist);
 532
 533                         return 0;
 534                 }
 535
 536                 g = g->next;
 537         }
 538
 539         free_gamelist (gamelist);
 540
 541         return 0;
 542 }
 543
 544
 545
 546 static int print_main_menu (session_info *sess)
 547 {
 548         char msg[BUF_SIZE];
 549         const char *reg, *name, *play;
 550
 551
 552         // XXX this is ugly, clean it up someday
 553
 554         if (sess->user_id < 0)
 555         {
 556                 reg = "r) Register as a user (needed in order to play)";
 557                 name = "player of roguelikes";
 558                 play = "l) Log in to play";
 559         }
 560         else
 561         {
 562                 reg = "c) Change password";
 563                 name = sess->user_name;
 564                 play = "p) Play\r\n\te) Edit game options files";
 565         }
 566
 567         int l = snprintf(msg, BUF_SIZE, TELNET_CLS "Welcome to the roguelike server, %s!\r\n"
 568                 "\t%s\r\n"
 569                 "\to) Observe running games\r\n"
 570                 "\ts) View high-score lists\r\n"
 571                 "\t%s\r\n"
 572                 "\tq) Leave\r\n", name, play, reg);
 573
 574         if (send(sess->socket, msg, l, 0) == -1) return -1;
 575         return 0;
 576 }
 577
 578
 579
 580 int main_menu (session_info *sess)
 581 {
 582         if (print_main_menu(sess) == -1) return -1;
 583
 584         while (1)
 585         {
 586                 char buf[BUF_SIZE];
 587                 int n = recv(sess->socket, buf, BUF_SIZE, 0);
 588                 if (n == -1) return -1;
 589
 590                 n = translate_telnet_input(buf, n, sess);
 591                 if (n > 0)
 592                 {
 593                         if (buf[0] == 'p' && sess->user_id >= 0)
 594                         {
 595                                 if (play_menu(sess, 0) == -1) return -1;
 596                                 if (print_main_menu(sess) == -1) return -1;
 597                         }
 598
 599                         if (buf[0] == 's')
 600                         {
 601                                 if (play_menu(sess, 1) == -1) return -1;
 602                                 if (print_main_menu(sess) == -1) return -1;
 603                         }
 604
 605                         if (buf[0] == 'e' && sess->user_id >= 0)
 606                         {
 607                                 if (edit_menu(sess) == -1) return -1;
 608                                 if (print_main_menu(sess) == -1) return -1;
 609                         }
 610
 611                         if (buf[0] == 'l' && sess->user_id < 0)
 612                         {
 613                                 if (login_menu(sess) == -1) return -1;
 614                                 if (print_main_menu(sess) == -1) return -1;
 615                         }
 616
 617                         if (buf[0] == 'r' && sess->user_id < 0)
 618                         {
 619                                 if (register_menu(sess) == -1) return -1;
 620                                 if (print_main_menu(sess) == -1) return -1;
 621                         }
 622
 623                         if (buf[0] == 'c' && sess->user_id >= 0)
 624                         {
 625                                 if (passwd_menu(sess) == -1) return -1;
 626                                 if (print_main_menu(sess) == -1) return -1;
 627                         }
 628
 629                         if (buf[0] == 'o')
 630                         {
 631                                 if (observer_menu(sess) == -1) return -1;
 632                                 if (print_main_menu(sess) == -1) return -1;
 633                         }
 634
 635                         if (buf[0] == 'q') return 0;
 636                 }
 637         }
 638 }
 639