clients/tty/dcsh.c

   1 /*
   2  *  Dolda Connect - Modular multiuser Direct Connect-style client
   3  *  Copyright (C) 2007 Fredrik Tolf <fredrik@dolda2000.com>
   4  *
   5  *  This program is free software; you can redistribute it and/or modify
   6  *  it under the terms of the GNU General Public License as published by
   7  *  the Free Software Foundation; either version 2 of the License, or
   8  *  (at your option) any later version.
   9  *
  10  *  This program is distributed in the hope that it will be useful,
  11  *  but WITHOUT ANY WARRANTY; without even the implied warranty of
  12  *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  13  *  GNU General Public License for more details.
  14  *
  15  *  You should have received a copy of the GNU General Public License
  16  *  along with this program; if not, write to the Free Software
  17  *  Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
  18 */
  19
  20 #include <stdlib.h>
  21 #include <stdio.h>
  22 #include <unistd.h>
  23 #include <wchar.h>
  24 #include <sys/poll.h>
  25 #include <string.h>
  26 #include <errno.h>
  27 #include <locale.h>
  28
  29 #ifdef HAVE_CONFIG_H
  30 #include <config.h>
  31 #endif
  32 #include <doldaconnect/uilib.h>
  33 #include <doldaconnect/uimisc.h>
  34 #include <doldaconnect/utils.h>
  35
  36 int verbose = 0;
  37 int dcfd;
  38
  39 struct cmd {
  40     wchar_t *name;
  41     int (*handler)(int argc, wchar_t **argv);
  42 };
  43
  44 int cmd_lsnodes(int argc, wchar_t **argv)
  45 {
  46     int i;
  47     struct dc_response *resp;
  48     struct dc_intresp *ires;
  49
  50     resp = dc_gettaggedrespsync(dc_queuecmd(NULL, NULL, L"lsnodes", NULL));
  51     if(resp->code == 200) {
  52         printf("   ID  NET  USERS S NAME\n");
  53         while((ires = dc_interpret(resp)) != NULL) {
  54             if(wcslen(ires->argv[2].val.str) > 58) {
  55                 for(i = 55; i < 58; i++)
  56                     ires->argv[2].val.str[i] = L'.';
  57                 ires->argv[2].val.str[i] = L'\0';
  58             }
  59             printf("%5i %4ls %6i %c %ls\n", ires->argv[0].val.num, ires->argv[1].val.str, ires->argv[3].val.num, "SHED"[ires->argv[4].val.num], ires->argv[2].val.str);
  60             dc_freeires(ires);
  61         }
  62     } else if(resp->code == 201) {
  63     } else {
  64         fprintf(stderr, "dcsh: %ls\n", resp->rlines[0].argv[0]);
  65     }
  66     return(0);
  67 }
  68
  69 struct cmd commands[] = {
  70     {L"hubs", cmd_lsnodes},
  71     {NULL, NULL}
  72 };
  73
  74 int run1(int argc, wchar_t **argv)
  75 {
  76     struct cmd *c;
  77
  78     for(c = commands; c->handler != NULL; c++) {
  79         if(!wcscmp(argv[0], c->name))
  80             return(c->handler(argc, argv));
  81     }
  82     fprintf(stderr, "dcsh: no such command\n");
  83     return(1);
  84 }
  85
  86 int runchar(int argc, char **argv) {
  87     int i, rv;
  88     wchar_t **wargv, *buf;
  89     size_t wargvsize, wargvdata;
  90
  91     wargv = NULL;
  92     wargvsize = wargvdata = 0;
  93     for(i = 0; i < argc; i++) {
  94         if((buf = icmbstowcs(argv[i], NULL)) == NULL) {
  95             fprintf(stderr, "dcsh: could not convert %s to wide char: %s\n", argv[i], strerror(errno));
  96             for(i = 0; i < wargvdata; i++)
  97                 free(wargv[i]);
  98             if(wargv != NULL)
  99                 free(wargv);
 100             return(1);
 101         }
 102         addtobuf(wargv, buf);
 103     }
 104     addtobuf(wargv, NULL);
 105     rv = run1(wargvdata - 1, wargv);
 106     dc_freewcsarr(wargv);
 107     return(rv);
 108 }
 109
 110 int shell(void)
 111 {
 112     int ret, argc;
 113     wchar_t *buf, **argv;
 114     char *p, cmdbuf[128];
 115     int cmddata;
 116     struct pollfd pfd[2];
 117
 118     cmddata = 0;
 119     while(1) {
 120         pfd[0].fd = dcfd;
 121         pfd[0].events = POLLIN;
 122         if(dc_wantwrite())
 123             pfd[0].events |= POLLOUT;
 124         pfd[1].fd = 0;
 125         pfd[1].events = POLLIN;
 126         pfd[1].revents = 0;
 127         if(poll(pfd, 2, -1) < 0) {
 128             if(errno != EINTR) {
 129                 perror("dcsh: poll");
 130                 exit(1);
 131             }
 132         }
 133         if(((pfd[0].revents & POLLIN) && dc_handleread()) || ((pfd[0].revents & POLLOUT) && dc_handlewrite()))
 134             return(1);
 135         if(pfd[1].revents) {
 136             ret = read(0, cmdbuf + cmddata, sizeof(cmdbuf) - cmddata);
 137             if(ret < 0) {
 138                 fprintf(stderr, "dcsh: stdin: %s\n", strerror(errno));
 139                 return(1);
 140             } else if(ret == 0) {
 141                 return(0);
 142             }
 143             cmddata += ret;
 144             while((p = memchr(cmdbuf, '\n', cmddata)) != NULL) {
 145                 *(p++) = 0;
 146                 if((buf = icmbstowcs(cmdbuf, NULL)) == NULL) {
 147                     fprintf(stderr, "dcsh: could not convert command to wide chars: %s\n", strerror(errno));
 148                 } else {
 149                     argv = dc_lexsexpr(buf);
 150                     free(buf);
 151                     for(argc = 0; argv[argc] != NULL; argc++);
 152                     if(argc > 0)
 153                         run1(argc, argv);
 154                     dc_freewcsarr(argv);
 155                 }
 156                 memmove(cmdbuf, p, cmddata -= (p - cmdbuf));
 157             }
 158         }
 159     }
 160 }
 161
 162 int main(int argc, char **argv)
 163 {
 164     int c, rv;
 165     char *server;
 166     char *username;
 167     int authless, authia;
 168
 169     setlocale(LC_ALL, "");
 170     server = username = NULL;
 171     authless = authia = 1;
 172     while((c = getopt(argc, argv, "hvIAs:u:")) != -1) {
 173         switch(c) {
 174         case 's':
 175             server = optarg;
 176             break;
 177         case 'u':
 178             username = optarg;
 179             break;
 180         case 'I':
 181             authia = 0;
 182             break;
 183         case 'A':
 184             authless = 0;
 185             break;
 186         case 'v':
 187             verbose++;
 188             break;
 189         default:
 190             fprintf(stderr, "usage: dcsh [-AIhv] [-s SERVER] [-u USERNAME] [COMMAND [ARGS...]]\n");
 191             exit((c == 'h')?0:1);
 192         }
 193     }
 194     dc_init();
 195
 196     if(verbose)
 197         fprintf(stderr, "connecting...\n");
 198     if((dcfd = dc_connectsync2(server, 1)) < 0) {
 199         perror("dcsh: could not connect to server");
 200         exit(1);
 201     }
 202     if(verbose)
 203         fprintf(stderr, "authenticating...\n");
 204     if(dc_login(username, authless, authia?dc_convtty:dc_convnone, NULL) != DC_LOGIN_ERR_SUCCESS) {
 205         fprintf(stderr, "dcsh: authentication unsuccessful\n");
 206         exit(1);
 207     }
 208     if(verbose)
 209         fprintf(stderr, "done\n");
 210     if(optind < argc)
 211         rv = runchar(argc - optind, argv + optind);
 212     else
 213         rv = shell();
 214     return(rv);
 215 }