1 /*- 2 * Copyright (c) 2002 Jonathan Belson <jon@witchspace.com> 3 * All rights reserved. 4 * 5 * Redistribution and use in source and binary forms, with or without 6 * modification, are permitted provided that the following conditions 7 * are met: 8 * 1. Redistributions of source code must retain the above copyright 9 * notice, this list of conditions and the following disclaimer. 10 * 2. Redistributions in binary form must reproduce the above copyright 11 * notice, this list of conditions and the following disclaimer in the 12 * documentation and/or other materials provided with the distribution. 13 * 14 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND 15 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE 16 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE 17 * ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE 18 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL 19 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS 20 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) 21 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT 22 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY 23 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF 24 * SUCH DAMAGE. 25 * 26 * $FreeBSD: src/usr.sbin/kbdmap/kbdmap.c,v 1.2 2002/10/27 17:44:33 wollman Exp $ 27 */ 28 29 #include <sys/types.h> 30 #include <sys/queue.h> 31 32 #include <assert.h> 33 #include <ctype.h> 34 #include <dirent.h> 35 #include <limits.h> 36 #include <stdio.h> 37 #include <stdlib.h> 38 #include <string.h> 39 #include <stringlist.h> 40 #include <unistd.h> 41 42 #include "kbdmap.h" 43 44 45 static const char *lang_default = DEFAULT_LANG; 46 static const char *font; 47 static const char *lang; 48 static const char *program; 49 static const char *keymapdir = DEFAULT_KEYMAP_DIR; 50 static const char *fontdir = DEFAULT_FONT_DIR; 51 static const char *sysconfig = DEFAULT_SYSCONFIG; 52 static const char *font_default = DEFAULT_FONT; 53 static const char *font_current; 54 static const char *dir; 55 static const char *menu = ""; 56 57 static int x11; 58 static int show; 59 static int verbose; 60 static int print; 61 62 63 struct keymap { 64 char *desc; 65 char *keym; 66 int mark; 67 SLIST_ENTRY(keymap) entries; 68 }; 69 static SLIST_HEAD(slisthead, keymap) head = SLIST_HEAD_INITIALIZER(head); 70 71 72 /* 73 * Get keymap entry for 'key', or NULL of not found 74 */ 75 static struct keymap * 76 get_keymap(const char *key) 77 { 78 struct keymap *km; 79 80 SLIST_FOREACH(km, &head, entries) 81 if (!strcmp(km->keym, key)) 82 return km; 83 84 return NULL; 85 } 86 87 /* 88 * Count the number of keymaps we found 89 */ 90 static int 91 get_num_keymaps(void) 92 { 93 struct keymap *km; 94 int count = 0; 95 96 SLIST_FOREACH(km, &head, entries) 97 count++; 98 99 return count; 100 } 101 102 /* 103 * Remove any keymap with given keym 104 */ 105 static void 106 remove_keymap(const char *keym) 107 { 108 struct keymap *km; 109 110 SLIST_FOREACH(km, &head, entries) { 111 if (!strcmp(keym, km->keym)) { 112 SLIST_REMOVE(&head, km, keymap, entries); 113 free(km); 114 break; 115 } 116 } 117 } 118 119 /* 120 * Add to hash with 'key' 121 */ 122 static void 123 add_keymap(const char *desc, int mark, const char *keym) 124 { 125 struct keymap *km, *km_new; 126 127 /* Is there already an entry with this key? */ 128 SLIST_FOREACH(km, &head, entries) { 129 if (!strcmp(km->keym, keym)) { 130 /* Reuse this entry */ 131 free(km->desc); 132 km->desc = strdup(desc); 133 km->mark = mark; 134 return; 135 } 136 } 137 138 km_new = (struct keymap *) malloc (sizeof(struct keymap)); 139 km_new->desc = strdup(desc); 140 km_new->keym = strdup(keym); 141 km_new->mark = mark; 142 143 /* Add to keymap list */ 144 SLIST_INSERT_HEAD(&head, km_new, entries); 145 } 146 147 /* 148 * Figure out the default language to use. 149 */ 150 static const char * 151 get_locale(void) 152 { 153 const char *locale; 154 155 if ((locale = getenv("LC_ALL")) == NULL && 156 (locale = getenv("LC_CTYPE")) == NULL && 157 (locale = getenv("LANG")) == NULL) 158 locale = lang_default; 159 160 /* Check for alias */ 161 if (!strcmp(locale, "C")) 162 locale = DEFAULT_LANG; 163 164 return locale; 165 } 166 167 /* 168 * Extract filename part 169 */ 170 static const char * 171 extract_name(const char *name) 172 { 173 char *p; 174 175 p = strrchr(name, '/'); 176 if (p != NULL && p[1] != '\0') 177 return p + 1; 178 179 return name; 180 } 181 182 /* 183 * Return file extension or NULL 184 */ 185 static char * 186 get_extension(const char *name) 187 { 188 char *p; 189 190 p = strrchr(name, '.'); 191 192 if (p != NULL && p[1] != '\0') 193 return p; 194 195 return NULL; 196 } 197 198 /* 199 * Read font from /etc/rc.conf else return default. 200 * Freeing the memory is the caller's responsibility. 201 */ 202 static char * 203 get_font(void) 204 { 205 char line[256], buf[20]; 206 char *fnt = NULL; 207 208 FILE *fp = fopen(sysconfig, "r"); 209 if (fp) { 210 while (fgets(line, sizeof(line), fp)) { 211 int a, b, matches; 212 213 if (line[0] == '#') 214 continue; 215 216 matches = sscanf(line, 217 " font%dx%d = \"%20[-.0-9a-zA-Z_]", 218 &a, &b, buf); 219 if (matches==3) { 220 if (strcmp(buf, "NO")) { 221 if (fnt) 222 free(fnt); 223 fnt = (char *) malloc(strlen(buf) + 1); 224 strcpy(fnt, buf); 225 } 226 } 227 } 228 } else 229 fprintf(stderr, "Could not open %s for reading\n", sysconfig); 230 231 return fnt; 232 } 233 234 /* 235 * Set a font using 'vidcontrol' 236 */ 237 static void 238 vidcontrol(const char *fnt) 239 { 240 char *tmp, *p, *q; 241 char ch; 242 int i; 243 244 /* syscons test failed */ 245 if (x11) 246 return; 247 248 tmp = strdup(fnt); 249 250 /* Extract font size */ 251 p = strrchr(tmp, '-'); 252 if (p && p[1] != '\0') { 253 p++; 254 /* Remove any '.fnt' extension */ 255 if ((q = strstr(p, ".fnt"))) 256 *q = '\0'; 257 258 /* 259 * Check font size is valid, with no trailing characters 260 * ('&ch' should not be matched) 261 */ 262 if (sscanf(p, "%dx%d%c", &i, &i, &ch) != 2) 263 fprintf(stderr, "Which font size? %s\n", fnt); 264 else { 265 char *cmd; 266 asprintf(&cmd, "vidcontrol -f %s %s", p, fnt); 267 if (verbose) 268 fprintf(stderr, "%s\n", cmd); 269 system(cmd); 270 free(cmd); 271 } 272 } else 273 fprintf(stderr, "Which font size? %s\n", fnt); 274 275 free(tmp); 276 } 277 278 /* 279 * Execute 'kbdcontrol' with the appropriate arguments 280 */ 281 static void 282 do_kbdcontrol(struct keymap *km) 283 { 284 char *kbd_cmd; 285 asprintf(&kbd_cmd, "kbdcontrol -l %s/%s", dir, km->keym); 286 287 if (!x11) 288 system(kbd_cmd); 289 290 printf("keymap=%s\n", km->keym); 291 free(kbd_cmd); 292 } 293 294 /* 295 * Call 'vidcontrol' with the appropriate arguments 296 */ 297 static void 298 do_vidfont(struct keymap *km) 299 { 300 char *vid_cmd, *tmp, *p, *q; 301 302 asprintf(&vid_cmd, "%s/%s", dir, km->keym); 303 vidcontrol(vid_cmd); 304 free(vid_cmd); 305 306 tmp = strdup(km->keym); 307 p = strrchr(tmp, '-'); 308 if (p && p[1]!='\0') { 309 p++; 310 q = get_extension(p); 311 if (q) { 312 *q = '\0'; 313 printf("font%s=%s\n", p, km->keym); 314 } 315 } 316 free(tmp); 317 } 318 319 /* 320 * Display dialog from 'keymaps[]' 321 */ 322 static void 323 show_dialog(struct keymap **km_sorted, int num_keymaps) 324 { 325 FILE *fp; 326 char *cmd, *dialog; 327 char tmp_name[] = "/tmp/_kbd_lang.XXXX"; 328 int fd, i, size; 329 330 fd = mkstemp(tmp_name); 331 if (fd == -1) { 332 fprintf(stderr, "Could not open temporary file \"%s\"\n", 333 tmp_name); 334 exit(1); 335 } 336 asprintf(&dialog, "/usr/bin/dialog --clear --title \"Keyboard Menu\" " 337 "--menu \"%s\" -1 -1 10", menu); 338 339 /* start right font, assume that current font is equal 340 * to default font in /etc/rc.conf 341 * 342 * $font is the font which require the language $lang; e.g. 343 * russian *need* a koi8 font 344 * $font_current is the current font from /etc/rc.conf 345 */ 346 if (font && strcmp(font, font_current)) 347 vidcontrol(font); 348 349 /* Build up the command */ 350 size = 0; 351 for (i=0; i<num_keymaps; i++) { 352 /* 353 * Each 'font' is passed as ' "font" ""', so allow the 354 * extra space 355 */ 356 size += strlen(km_sorted[i]->desc) + 6; 357 } 358 359 /* Allow the space for '2> tmpfilename' redirection */ 360 size += strlen(tmp_name) + 3; 361 362 cmd = (char *) malloc(strlen(dialog) + size + 1); 363 strcpy(cmd, dialog); 364 365 for (i=0; i<num_keymaps; i++) { 366 strcat(cmd, " \""); 367 strcat(cmd, km_sorted[i]->desc); 368 strcat(cmd, "\""); 369 strcat(cmd, " \"\""); 370 } 371 372 strcat(cmd, " 2>"); 373 strcat(cmd, tmp_name); 374 375 /* Show the dialog.. */ 376 system(cmd); 377 378 fp = fopen(tmp_name, "r"); 379 if (fp) { 380 char choice[64]; 381 if (fgets(choice, 64, fp) != NULL) { 382 /* Find key for desc */ 383 for (i=0; i<num_keymaps; i++) { 384 if (!strcmp(choice, km_sorted[i]->desc)) { 385 if (!strcmp(program, "kbdmap")) 386 do_kbdcontrol(km_sorted[i]); 387 else 388 do_vidfont(km_sorted[i]); 389 break; 390 } 391 } 392 } else { 393 if (font != NULL && strcmp(font, font_current)) 394 /* Cancelled, restore old font */ 395 vidcontrol(font_current); 396 } 397 fclose(fp); 398 } else 399 fprintf(stderr, "Failed to open temporary file"); 400 401 /* Tidy up */ 402 remove(tmp_name); 403 free(cmd); 404 free(dialog); 405 close(fd); 406 } 407 408 /* 409 * Search for 'token' in comma delimited array 'buffer'. 410 * Return true for found, false for not found. 411 */ 412 static int 413 find_token(const char *buffer, const char *token) 414 { 415 char *buffer_tmp, *buffer_copy, *inputstring; 416 char **ap; 417 int found; 418 419 buffer_copy = strdup(buffer); 420 buffer_tmp = buffer_copy; 421 inputstring = buffer_copy; 422 ap = &buffer_tmp; 423 424 found = 0; 425 426 while ((*ap = strsep(&inputstring, ",")) != NULL) { 427 if (strcmp(buffer_tmp, token) == 0) { 428 found = 1; 429 break; 430 } 431 } 432 433 free(buffer_copy); 434 435 return found; 436 } 437 438 /* 439 * Compare function for qsort 440 */ 441 static int 442 compare_keymap(const void *a, const void *b) 443 { 444 445 /* We've been passed pointers to pointers, so: */ 446 const struct keymap *km1 = *((const struct keymap * const *) a); 447 const struct keymap *km2 = *((const struct keymap * const *) b); 448 449 return strcmp(km1->desc, km2->desc); 450 } 451 452 /* 453 * Compare function for qsort 454 */ 455 static int 456 compare_lang(const void *a, const void *b) 457 { 458 const char *l1 = *((const char * const *) a); 459 const char *l2 = *((const char * const *) b); 460 461 return strcmp(l1, l2); 462 } 463 464 /* 465 * Change '8x8' to '8x08' so qsort will put it before eg. '8x14' 466 */ 467 static void 468 kludge_desc(struct keymap **km_sorted, int num_keymaps) 469 { 470 int i; 471 472 for (i=0; i<num_keymaps; i++) { 473 char *p; 474 char *km = km_sorted[i]->desc; 475 if ((p = strstr(km, "8x8")) != NULL) { 476 int len; 477 int j; 478 int offset; 479 480 offset = p - km; 481 482 /* Make enough space for the extra '0' */ 483 len = strlen(km); 484 km = realloc(km, len + 2); 485 486 for (j=len; j!=offset+1; j--) 487 km[j + 1] = km[j]; 488 489 km[offset+2] = '0'; 490 491 km_sorted[i]->desc = km; 492 } 493 } 494 } 495 496 /* 497 * Reverse 'kludge_desc()' - change '8x08' back to '8x8' 498 */ 499 static void 500 unkludge_desc(struct keymap **km_sorted, int num_keymaps) 501 { 502 int i; 503 504 for (i=0; i<num_keymaps; i++) { 505 char *p; 506 char *km = km_sorted[i]->desc; 507 if ((p = strstr(km, "8x08")) != NULL) { 508 p += 2; 509 while (*p++) 510 p[-1] = p[0]; 511 512 km = realloc(km, p - km - 1); 513 km_sorted[i]->desc = km; 514 } 515 } 516 } 517 518 /* 519 * Return 0 if file exists and is readable, else -1 520 */ 521 static int 522 check_file(const char *keym) 523 { 524 int status = 0; 525 526 if (access(keym, R_OK) == -1) { 527 char *fn; 528 asprintf(&fn, "%s/%s", dir, keym); 529 if (access(fn, R_OK) == -1) { 530 if (verbose) 531 fprintf(stderr, "%s not found!\n", fn); 532 status = -1; 533 } 534 free(fn); 535 } else { 536 if (verbose) 537 fprintf(stderr, "No read permission for %s!\n", keym); 538 status = -1; 539 } 540 541 return status; 542 } 543 544 /* 545 * Read options from the relevent configuration file, then 546 * present to user. 547 */ 548 static void 549 menu_read(void) 550 { 551 const char *lg; 552 char *p; 553 int mark, num_keymaps, items, i; 554 char buffer[256], filename[PATH_MAX]; 555 char keym[64], lng[64], desc[64]; 556 char dialect[64], lang_abk[64]; 557 struct keymap *km; 558 struct keymap **km_sorted; 559 struct dirent *dp; 560 StringList *lang_list; 561 FILE *fp; 562 DIR *dirp; 563 564 lang_list = sl_init(); 565 566 sprintf(filename, "%s/INDEX.%s", dir, extract_name(dir)); 567 568 /* en_US.ISO8859-1 -> en_..\.ISO8859-1 */ 569 strlcpy(dialect, lang, sizeof(dialect)); 570 if (strlen(dialect) >= 6 && dialect[2] == '-') { 571 dialect[3] = '.'; 572 dialect[4] = '.'; 573 } 574 575 576 /* en_US.ISO8859-1 -> en */ 577 strlcpy(lang_abk, lang, sizeof(lang_abk)); 578 if (strlen(lang_abk) >= 3 && lang_abk[2] == '-') 579 lang_abk[2] = '.'; 580 581 fprintf(stderr, "lang_default = %s\n", lang_default); 582 fprintf(stderr, "dialect = %s\n", dialect); 583 fprintf(stderr, "lang_abk = %s\n", lang_abk); 584 585 fp = fopen(filename, "r"); 586 if (fp) { 587 int matches; 588 while (fgets(buffer, sizeof(buffer), fp)) { 589 p = buffer; 590 if (p[0] == '#') 591 continue; 592 593 while (isspace(*p)) 594 p++; 595 596 if (*p == '\0') 597 continue; 598 599 /* Parse input, removing newline */ 600 matches = sscanf(p, "%64[^:]:%64[^:]:%64[^:\n]", 601 keym, lng, desc); 602 if (matches == 3) { 603 if (strcmp(keym, "FONT") 604 && strcmp(keym, "MENU")) { 605 /* Check file exists & is readable */ 606 if (check_file(keym) == -1) 607 continue; 608 } 609 } 610 611 if (show) { 612 /* 613 * Take note of supported languages, which 614 * might be in a comma-delimited list 615 */ 616 char *tmp = strdup(lng); 617 char *delim = tmp; 618 619 for (delim = tmp; ; ) { 620 char ch = *delim++; 621 if (ch == ',' || ch == '\0') { 622 delim[-1] = '\0'; 623 if (!sl_find(lang_list, tmp)) 624 sl_add(lang_list, tmp); 625 if (ch == '\0') 626 break; 627 tmp = delim; 628 } 629 } 630 } 631 /* Set empty language to default language */ 632 if (lng[0] == '\0') 633 lg = lang_default; 634 else 635 lg = lng; 636 637 638 /* 4) Your choice if it exists 639 * 3) Long match eg. en_GB.ISO8859-1 is equal to 640 * en_..\.ISO8859-1 641 * 2) short match 'de' 642 * 1) default langlist 'en' 643 * 0) any language 644 * 645 * Language may be a comma separated list 646 * A higher match overwrites a lower 647 * A later entry overwrites a previous if it exists 648 * twice in the database 649 */ 650 651 /* Check for favoured language */ 652 km = get_keymap(keym); 653 mark = (km) ? km->mark : 0; 654 655 if (find_token(lg, lang)) 656 add_keymap(desc, 4, keym); 657 else if (mark <= 3 && find_token(lg, dialect)) 658 add_keymap(desc, 3, keym); 659 else if (mark <= 2 && find_token(lg, lang_abk)) 660 add_keymap(desc, 2, keym); 661 else if (mark <= 1 && find_token(lg, lang_default)) 662 add_keymap(desc, 1, keym); 663 else if (mark <= 0) 664 add_keymap(desc, 0, keym); 665 } 666 fclose(fp); 667 668 } else 669 printf("Could not open file\n"); 670 671 if (show) { 672 qsort(lang_list->sl_str, lang_list->sl_cur, sizeof(char*), 673 compare_lang); 674 printf("Currently supported languages: "); 675 for (i=0; i< (int) lang_list->sl_cur; i++) 676 printf("%s ", lang_list->sl_str[i]); 677 puts(""); 678 exit(0); 679 } 680 681 km = get_keymap("MENU"); 682 if (km) 683 /* Take note of menu title */ 684 menu = strdup(km->desc); 685 km = get_keymap("FONT"); 686 if (km) 687 /* Take note of language font */ 688 font = strdup(km->desc); 689 690 /* Remove unwanted items from list */ 691 remove_keymap("MENU"); 692 remove_keymap("FONT"); 693 694 /* Look for keymaps not in database */ 695 dirp = opendir(dir); 696 if (dirp) { 697 while ((dp = readdir(dirp)) != NULL) { 698 const char *ext = get_extension(dp->d_name); 699 if (ext) { 700 if ((!strcmp(ext, ".fnt") || 701 !strcmp(ext, ".kbd")) && 702 !get_keymap(dp->d_name)) { 703 char *q; 704 705 /* Remove any .fnt or .kbd extension */ 706 q = strdup(dp->d_name); 707 *(get_extension(q)) = '\0'; 708 add_keymap(q, 0, dp->d_name); 709 free(q); 710 711 if (verbose) 712 fprintf(stderr, 713 "'%s' not in database\n", 714 dp->d_name); 715 } 716 } 717 } 718 closedir(dirp); 719 } else 720 fprintf(stderr, "Could not open directory '%s'\n", dir); 721 722 /* Sort items in keymap */ 723 num_keymaps = get_num_keymaps(); 724 725 km_sorted = (struct keymap **) 726 malloc(num_keymaps*sizeof(struct keymap *)); 727 728 /* Make array of pointers to items in hash */ 729 items = 0; 730 SLIST_FOREACH(km, &head, entries) 731 km_sorted[items++] = km; 732 733 /* Change '8x8' to '8x08' so sort works as we might expect... */ 734 kludge_desc(km_sorted, num_keymaps); 735 736 qsort(km_sorted, num_keymaps, sizeof(struct keymap *), compare_keymap); 737 738 /* ...change back again */ 739 unkludge_desc(km_sorted, num_keymaps); 740 741 if (print) { 742 for (i=0; i<num_keymaps; i++) 743 printf("%s\n", km_sorted[i]->desc); 744 exit(0); 745 } 746 747 show_dialog(km_sorted, num_keymaps); 748 749 free(km_sorted); 750 } 751 752 /* 753 * Display usage information and exit 754 */ 755 static void 756 usage(void) 757 { 758 759 fprintf(stderr, "usage: %s\t[-K] [-V] [-d|-default] [-h|-help] " 760 "[-l|-lang language]\n\t\t[-p|-print] [-r|-restore] [-s|-show] " 761 "[-v|-verbose]\n", program); 762 exit(1); 763 } 764 765 static void 766 parse_args(int argc, char **argv) 767 { 768 int i; 769 770 for (i=1; i<argc; i++) { 771 if (argv[i][0] != '-') 772 usage(); 773 else if (!strcmp(argv[i], "-help") || !strcmp(argv[i], "-h")) 774 usage(); 775 else if (!strcmp(argv[i], "-verbose") || !strcmp(argv[i], "-v")) 776 verbose = 1; 777 else if (!strcmp(argv[i], "-lang") || !strcmp(argv[i], "-l")) 778 if (i + 1 == argc) 779 usage(); 780 else 781 lang = argv[++i]; 782 else if (!strcmp(argv[i], "-default") || !strcmp(argv[i], "-d")) 783 lang = lang_default; 784 else if (!strcmp(argv[i], "-show") || !strcmp(argv[i], "-s")) 785 show = 1; 786 else if (!strcmp(argv[i], "-print") || !strcmp(argv[i], "-p")) 787 print = 1; 788 else if (!strcmp(argv[i], "-restore") || 789 !strcmp(argv[i], "-r")) { 790 vidcontrol(font_current); 791 exit(0); 792 } else if (!strcmp(argv[i], "-K")) 793 dir = keymapdir; 794 else if (!strcmp(argv[i], "-V")) 795 dir = fontdir; 796 else 797 usage(); 798 } 799 } 800 801 /* 802 * A front-end for the 'vidfont' and 'kbdmap' programs. 803 */ 804 int 805 main(int argc, char **argv) 806 { 807 808 x11 = system("kbdcontrol -d >/dev/null"); 809 810 if (x11) { 811 fprintf(stderr, "You are not on a virtual console - " 812 "expect certain strange side-effects\n"); 813 sleep(2); 814 } 815 816 SLIST_INIT(&head); 817 818 lang = get_locale(); 819 820 program = extract_name(argv[0]); 821 822 font_current = get_font(); 823 if (font_current == NULL) 824 font_current = font_default; 825 826 if (strcmp(program, "kbdmap")) 827 dir = fontdir; 828 else 829 dir = keymapdir; 830 831 /* Parse command line arguments */ 832 parse_args(argc, argv); 833 834 /* Read and display options */ 835 menu_read(); 836 837 return 0; 838 } 839