kernel/core/tty.c

   1 /*
   2  *  ZeX/OS
   3  *  Copyright (C) 2007  Tomas 'ZeXx86' Jedrzejek (zexx86@gmail.com)
   4  *  Copyright (C) 2008  Tomas 'ZeXx86' Jedrzejek (zexx86@gmail.com)
   5  *  Copyright (C) 2009  Tomas 'ZeXx86' Jedrzejek (zexx86@gmail.com)
   6  *
   7  *  This program is free software: you can redistribute it and/or modify
   8  *  it under the terms of the GNU General Public License as published by
   9  *  the Free Software Foundation, either version 3 of the License, or
  10  *  (at your option) any later version.
  11  *
  12  *  This program is distributed in the hope that it will be useful,
  13  *  but WITHOUT ANY WARRANTY; without even the implied warranty of
  14  *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  15  *  GNU General Public License for more details.
  16  *
  17  *  You should have received a copy of the GNU General Public License
  18  *  along with this program.  If not, see <http://www.gnu.org/licenses/>.
  19  */
  20
  21
  22 #include <build.h>
  23 #include <system.h>
  24 #include <string.h>
  25 #include <vfs.h>
  26
  27 #ifdef ARCH_i386                        /* i386 specific things */
  28 extern int attrib;
  29 extern int gets (void);
  30 extern int csr_x, csr_y;
  31 extern unsigned char vgagui;
  32 extern unsigned short *textmemptr;
  33 #else                                   /* fake things for other archs */
  34 unsigned int attrib;
  35 unsigned char vgagui;
  36 #endif
  37
  38 static char con_screen[TTY_CON_BUF];    /* console buffer for image of screen */
  39 static unsigned short tty_count;        /* number of created ttys */
  40 tty_font_t tty_font;                    /* global font settings */
  41 tty_char_t tty_char;                    /* global character array settings */
  42
  43 tty_t tty_list;                         /* list for all created ttys */
  44 tty_t *tty0, *tty1;                     /* structures for default ttys (0, 1) */
  45
  46 static unsigned task_tty ();
  47
  48 /* setup of font */
  49 bool tty_font_set (char *c, unsigned char x, unsigned char y)
  50 {
  51         if (!c || !x || !y)
  52                 return 0;
  53
  54         tty_font.c = c;
  55         tty_font.x = x;
  56         tty_font.y = y;
  57
  58         return 1;
  59 }
  60
  61 /* setup of character array */
  62 bool tty_char_set (unsigned short x, unsigned short y)
  63 {
  64         if ((x * y * 2) > TTY_CON_BUF)
  65                 return 0;
  66
  67         tty_char.x = x;
  68         tty_char.y = y;
  69
  70         return 1;
  71 }
  72
  73 /* clear selected screen */
  74 bool tty_cls (tty_t *tty)
  75 {
  76         if (!tty)
  77                 return 0;
  78
  79         if (!tty->screen)
  80                 return 0;
  81
  82         memset (tty->screen, 0, tty_char.x * tty_char.y * 2);
  83
  84         tty->screen_x = 0;
  85         tty->screen_y = 0;
  86         tty->read_len = 0;
  87
  88         if (tty != currtty)
  89                 return 1;
  90
  91 #ifdef ARCH_i386
  92         if (vgagui) {
  93                 if (vgagui == 2)
  94                         video_gfx_cls (0);
  95
  96                 return 1;
  97         }
  98
  99         csr_x = 0;
 100         csr_y = 0;
 101
 102         video_scroll ();
 103         video_move_csr ();
 104 #endif
 105         video_cls ();
 106
 107         return 1;
 108 }
 109
 110 /* lock manipulation with selected tty */
 111 bool tty_lock (tty_t *tty)
 112 {
 113         if (!tty)
 114                 return 0;
 115
 116         tty->active = false;
 117
 118         return 1;
 119 }
 120
 121 /* unlock manipulation with selected tty */
 122 bool tty_unlock (tty_t *tty)
 123 {
 124         if (!tty)
 125                 return 0;
 126
 127         tty->active = true;
 128
 129         return 1;
 130 }
 131
 132 /* startup function for tty init */
 133 bool tty_startup ()
 134 {
 135         tty_t *tty = tty0;
 136
 137         if (tty == NULL)
 138                 return 0;
 139
 140         currtty = tty;
 141
 142         strcpy (currtty->pwd, (char *) env_get ("PWD"));
 143
 144         currtty = tty;
 145
 146         env_set ("PWD", tty->pwd);
 147
 148         /* always default colors */
 149         attrib = 0x07;
 150
 151         /* clear screen*/
 152         video_cls ();
 153
 154         return 1;
 155 }
 156
 157 /* change current tty to selected one */
 158 bool tty_change (tty_t *tty)
 159 {
 160 #ifndef ARCH_i386
 161         return 0;
 162 #endif
 163         if (tty == NULL)
 164                 return 0;
 165
 166 #ifdef ARCH_i386
 167         /* redraw screen with selected tty */
 168         memcpy (textmemptr, tty->screen, tty_char.x * tty_char.y * 2);
 169
 170         csr_x = tty->screen_x;
 171         csr_y = tty->screen_y;
 172
 173         video_move_csr ();
 174 #endif
 175         strcpy (currtty->pwd, (char *) env_get ("PWD"));
 176
 177         if (currtty->user && currtty->user->name)
 178                 strcpy (currtty->user->name, (char *) env_get ("USER"));
 179
 180         currtty = tty;
 181
 182         env_set ("PWD", tty->pwd);
 183
 184         if (tty->user)
 185                 env_set ("USER", tty->user->name);
 186
 187         return 1;
 188 }
 189
 190 /* move cursor position to selected place */
 191 bool tty_gotoxy (tty_t *tty, unsigned char x, unsigned char y)
 192 {
 193         if (tty == NULL)
 194                 return 0;
 195
 196         if (x >= tty_char.x || y >= tty_char.y)
 197                 return 0;
 198
 199         video_gotoxy (x, y);
 200
 201         tty->screen_x = x;
 202         tty->screen_y = y;
 203
 204         return 1;
 205 }
 206
 207 /* find wanted tty by name */
 208 tty_t *tty_find (char *name)
 209 {
 210         tty_t *tty;
 211
 212         for (tty = tty_list.next; tty != &tty_list; tty = tty->next) {
 213                 if (!strcmp (tty->name, name))
 214                         return tty;
 215         }
 216
 217         return NULL;
 218 }
 219
 220 /* find tty by task */
 221 tty_t *tty_findbytask (task_t *task)
 222 {
 223         tty_t *tty;
 224
 225         for (tty = tty_list.next; tty != &tty_list; tty = tty->next) {
 226                 if (tty->task == task)
 227                         return tty;
 228         }
 229
 230         return NULL;
 231 }
 232
 233 /* display list of created tty consoles */
 234 void tty_listview ()
 235 {
 236         unsigned short id = 1;
 237
 238         tty_t *tty;
 239
 240         for (tty = tty_list.next; tty != &tty_list; tty = tty->next) {
 241                 printf ("%d.\t%s\t%s\n", id, tty->name, currtty == tty ? "<- current" : "");
 242
 243                 id ++;
 244         }
 245 }
 246
 247 /* main function for print strings into selected tty */
 248 bool tty_write (tty_t *tty, char *str, unsigned len)
 249 {
 250         if (!tty)
 251                 return 0;
 252
 253         unsigned l = 0;
 254         while (l != len) {
 255                 if (!tty_putnch (tty, str[l]))
 256                         break;
 257
 258                 l ++;
 259         }
 260
 261         return 1;
 262 }
 263
 264 /* read strings from selected tty */
 265 int tty_read (tty_t *tty, char *str, unsigned len)
 266 {
 267         if (!tty || !str)
 268                 return 0;
 269
 270         char *p = (char *) tty->screen;
 271
 272         unsigned l = 0;
 273         unsigned n = 2000;
 274
 275         unsigned x = 0;
 276         unsigned y = 0;
 277
 278         if (len+tty->read_len > n)
 279                 len = n-tty->read_len;
 280
 281         for (l = 0; l < len; l ++) {
 282                 if (p[((l+tty->read_len)*2)+1])
 283                         str[l] = p[((l+tty->read_len)*2)+1];
 284
 285                 x ++;
 286
 287                 if (x >= tty_char.x) {
 288                         x = 0;
 289                         y ++;
 290                         str[l] = '\n';
 291                 }
 292         }
 293
 294         str[l] = '\0';
 295
 296         tty->read_len += l;
 297
 298         return l;
 299 }
 300
 301 /* print character to current tty */
 302 bool tty_putch (char c)
 303 {
 304         tty_putnch (currtty, c);
 305
 306         return 1;
 307 }
 308
 309 /* print character to selected tty */
 310 bool tty_putnch (tty_t *tty, char c)
 311 {
 312         /* we wont null character */
 313         if (!c)
 314                 return 0;
 315
 316         switch (c) {
 317                 case '\n':
 318                         tty->screen_y ++;
 319                         tty->screen_x = 0;
 320                         break;
 321                 case '\b':
 322                         if (tty->screen_x)
 323                                 tty->screen_x --;
 324
 325                         /* save font attributes (colors) */
 326                         tty->screen[(tty->screen_x+(tty->screen_y*tty_char.x))*2] = ' ';
 327                         /* save character into buffer */
 328                         tty->screen[((tty->screen_x+(tty->screen_y*tty_char.x))*2)+1] = attrib;
 329                         break;
 330                 case '\t':
 331                         tty->screen_x = (tty->screen_x + 8) & ~(8 - 1);
 332                         break;
 333                 default:
 334                         /* save font attributes (colors) */
 335                         tty->screen[(tty->screen_x+(tty->screen_y*tty_char.x))*2] = c;
 336                         /* save character into buffer */
 337                         tty->screen[((tty->screen_x+(tty->screen_y*tty_char.x))*2)+1] = attrib;
 338                         tty->screen_x ++;
 339                         break;
 340         }
 341
 342         if (tty->screen_x >= tty_char.x) {
 343                 tty->screen_x -= tty_char.x;
 344                 tty->screen_y ++;
 345         }
 346
 347         if (tty->screen_y >= tty_char.y) {
 348                 unsigned i;
 349                 /* save old screen, but with one line offset */
 350                 for (i = 0; i < (tty_char.x * tty_char.y * 2) - (tty_char.x * 2); i ++)
 351                         con_screen[i] = tty->screen[i+(tty_char.x * 2)];
 352                 /* copy new image into tty screen */
 353                 for (i = 0; i < (tty_char.x * tty_char.y * 2); i ++)
 354                         tty->screen[i] = con_screen[i];
 355
 356                 tty->read_len += tty_char.x;
 357
 358                 tty->screen_y = tty_char.y - 1;
 359         }
 360
 361         if (!vgagui)
 362         if (tty == currtty)
 363                 video_putch (c);
 364
 365         return 1;
 366 }
 367
 368 /* create Message Of The Day in /etc/motd */
 369 bool tty_motd ()
 370 {
 371         vfs_list_add ("motd", VFS_FILEATTR_FILE | VFS_FILEATTR_READ | VFS_FILEATTR_SYSTEM, "/etc/");
 372
 373         char *motd =    "\t\t\t\tWelcome !\n"
 374                         "This is ZeX/OS - operating system created by Tomas 'ZeXx86' Jedrzejek\n"
 375                         "Please visit web page www.zexos.org for more information ..\n"
 376                         "Latest source code is available on git repository: git://repo.or.cz/ZeXOS.git\n"
 377                         "You can display list of the built-in commands by \"help\".\n"
 378                         "Default login name is \"root\" secured by password \"root\".\n";
 379
 380         vfs_mmap ("/etc/motd", 9, motd, strlen (motd));
 381
 382         video_color (15, 0);
 383
 384         vfs_cat ("/etc/motd", 9);
 385
 386         video_color (7, 0);
 387
 388         return 1;
 389 }
 390
 391 /* create new tty console */
 392 tty_t *tty_create ()
 393 {
 394         /* Software limit - max. 1K of ttys */
 395         if (tty_count > TTY_COUNT_MAX)
 396                 return 0;
 397
 398         char name[8];
 399         sprintf (name, "tty%d", tty_count);
 400
 401         unsigned name_len = strlen (name);
 402
 403         /* alloc and init context */
 404         tty_t *tty = (tty_t *) kmalloc (sizeof (tty_t));
 405
 406         if (!tty)
 407                 return 0;
 408
 409         memset (tty, 0, sizeof (tty_t));
 410
 411         tty->screen = (char *) kmalloc (tty_char.x * tty_char.y * 2);
 412
 413         if (!tty->screen) {
 414                 kfree (tty);
 415                 return 0;
 416         }
 417
 418         tty->name = (char *) kmalloc (sizeof (char) * name_len + 1);
 419
 420         if (!tty->name) {
 421                 kfree (tty->screen);
 422                 kfree (tty);
 423                 return 0;
 424         }
 425
 426         memcpy (tty->name, name, name_len);
 427         tty->name[name_len] = '\0';
 428
 429         memset (tty->screen, 0, tty_char.x * tty_char.y * 2);
 430
 431         tty->screen_x = 0;
 432         tty->screen_y = 0;
 433
 434         tty->logged = false;
 435
 436         tty->active = true;
 437
 438         tty->read_len = 0;
 439
 440         strcpy (tty->pwd, (char *) env_get ("PWD"));
 441
 442         tty->task = (task_t *) task_create (name, (unsigned) task_tty, 32);
 443
 444         /* add into list */
 445         tty->next = &tty_list;
 446         tty->prev = tty_list.prev;
 447         tty->prev->next = tty;
 448         tty->next->prev = tty;
 449
 450         tty_count ++;
 451
 452         return tty;
 453 }
 454
 455 /* thread for running tty consoles - handle selected console */
 456 void task_thread (tty_t *tty)
 457 {
 458         schedule ();
 459
 460         if (!tty)
 461                 return;
 462
 463         if (tty != currtty)
 464                 return;
 465
 466         if (tty->active) {
 467                 /* handle console on kernel level */
 468                 int id = gets ();
 469
 470                 if (id == 1)
 471                         if (tty->user) {
 472                                 console (id);
 473                         } else
 474                                 getlogin (id);
 475
 476 #ifdef ARCH_i386
 477                 /* VESA mode - graphical font */
 478                 if (vgagui == 2) {
 479                         unsigned i = 0;
 480                         unsigned j = 0;
 481                         unsigned k = 0;
 482                         unsigned m = tty_char.x * tty_char.y; //tty->screen_x + (tty->screen_y*80);
 483
 484                         while (i < m) {
 485                                 gputch (1+(j*tty_font.x), 11+(k*(tty_font.y+1)), vesa_16cto16b (tty->screen[i*2+1] & 0x0F), tty->screen[i*2]);
 486
 487                                 j ++;
 488
 489                                 if (j >= tty_char.x) {
 490                                         j = 0;
 491                                         k ++;
 492                                 }
 493
 494                                 i ++;
 495                         }
 496
 497                         video_gfx_fbswap ();
 498                 }
 499 #endif
 500         }
 501 }
 502
 503 /* startup function for created tty console */
 504 static unsigned task_tty ()
 505 {
 506         tty_t *tty = tty_findbytask (_curr_task);
 507
 508         if (!tty)
 509                 return 0;
 510
 511         while (1)
 512                 task_thread (tty);
 513 }
 514
 515 /* change tty console to next (+) or previous (-) in list */
 516 void tty_switch (char act)
 517 {
 518         switch (act) {
 519                 case '+':
 520                 {
 521                         tty_t *tty = currtty->next;
 522
 523                         if (tty == &tty_list)
 524                                 tty = tty_list.next;
 525
 526                         if (tty)
 527                                 tty_change (tty);
 528
 529                         break;
 530                 }
 531                 case '-':
 532                 {
 533                         tty_t *tty = currtty->prev;
 534
 535                         if (tty == &tty_list)
 536                                 tty = tty_list.prev;
 537
 538                         if (tty)
 539                                 tty_change (tty);
 540
 541                         break;
 542                 }
 543         }
 544 }
 545
 546 /* special function for initialization of graphical tty */
 547 tty_t *gtty_init ()
 548 {
 549         tty_t *tty = tty_create ();
 550
 551         tty_change (tty);
 552
 553         return tty;
 554 }
 555
 556 /* initialization of tty */
 557 unsigned int init_tty ()
 558 {
 559         /* pre-set tty list */
 560         tty_list.next = &tty_list;
 561         tty_list.prev = &tty_list;
 562
 563         /* default values */
 564         tty_count = 0;
 565
 566         tty_font_set ((char *) font5x8, 5, 8);
 567
 568         switch (vgagui) {
 569                 case 0: /* VGA-Textual console */
 570                         tty_char_set (80, 25);
 571                         break;
 572                 case 1: /* VGA console */
 573                         tty_char_set (53, 18);
 574                         break;
 575                 case 2: /* VESA concole */
 576                         tty_char_set (80, 65);
 577                         break;
 578         }
 579
 580         video_color (7, 0);
 581
 582         /* create consoles */
 583         tty0 = tty_create ();
 584
 585         if (!tty0)
 586                 return 0;
 587
 588 #ifdef ARCH_i386
 589         tty1 = tty_create ();
 590
 591         if (!tty1)
 592                 return 0;
 593 #endif
 594         tty_startup ();
 595
 596         /* printf MOTD into TTY0 console */
 597         tty_motd ();
 598
 599         /* then print login messages */
 600         tty_write (tty0, "\nlogin: ", 8);
 601         tty_write (tty1, "\nlogin: ", 8);
 602
 603         return 1;
 604 }
 605