git.mutantstargoat.com Git - oftp/blob - src/main.c

   1 #include <stdio.h>
   2 #include <string.h>
   3 #include <signal.h>
   4 #include <errno.h>
   5 #include <assert.h>
   6 #include <sys/select.h>
   7 #include "tgfx.h"
   8 #include "input.h"
   9 #include "util.h"
  10 #include "tui.h"
  11 #include "ftp.h"
  12
  13 void updateui(void);
  14 int proc_input(void);
  15 int keypress(int key);
  16 int parse_args(int argc, char **argv);
  17
  18 static struct ftp *ftp;
  19 static struct tui_widget *uilist;
  20
  21 static char *host = "localhost";
  22 static int port = 21;
  23
  24 int main(int argc, char **argv)
  25 {
  26         int i, numsock, maxfd;
  27         int ftpsock[16];
  28         fd_set rdset;
  29         struct timeval tv;
  30
  31         if(parse_args(argc, argv) == -1) {
  32                 return 1;
  33         }
  34
  35         if(!(ftp = ftp_alloc())) {
  36                 return 1;
  37         }
  38         if(ftp_connect(ftp, host, port) == -1) {
  39                 ftp_free(ftp);
  40                 return 1;
  41         }
  42
  43         init_input();
  44
  45         tg_init();
  46
  47         tg_bgchar(' ');
  48         tg_clear();
  49
  50         uilist = tui_list("Remote", 0, 0, 40, 23, 0, 0);
  51
  52         tg_setcursor(0, 23);
  53
  54         tui_draw(uilist);
  55
  56         for(;;) {
  57                 FD_ZERO(&rdset);
  58                 maxfd = 0;
  59
  60                 numsock = ftp_sockets(ftp, ftpsock, sizeof ftpsock);
  61                 for(i=0; i<numsock; i++) {
  62                         FD_SET(ftpsock[i], &rdset);
  63                         if(ftpsock[i] > maxfd) maxfd = ftpsock[i];
  64                 }
  65
  66 #ifdef __unix__
  67                 FD_SET(0, &rdset);
  68                 tv.tv_sec = 120;
  69                 tv.tv_usec = 0;
  70 #else
  71                 tv.tv_sec = tv.tv_usec = 0;
  72 #endif
  73
  74                 if(select(maxfd + 1, &rdset, 0, 0, &tv) == -1 && errno == EINTR) {
  75                         continue;
  76                 }
  77
  78 #ifdef __unix__
  79                 if(FD_ISSET(0, &rdset)) {
  80                         if(proc_input() == -1) {
  81                                 break;
  82                         }
  83                 }
  84 #endif
  85
  86                 for(i=0; i<numsock; i++) {
  87                         if(FD_ISSET(ftpsock[i], &rdset)) {
  88                                 ftp_handle(ftp, ftpsock[i]);
  89                         }
  90                 }
  91
  92                 updateui();
  93         }
  94
  95         tg_cleanup();
  96         cleanup_input();
  97         ftp_close(ftp);
  98         ftp_free(ftp);
  99         return 0;
 100 }
 101
 102 void updateui(void)
 103 {
 104         int i, num;
 105         struct ftp_dirent *ent;
 106         unsigned int upd = 0;
 107         char buf[128];
 108         const char *remdir;
 109
 110         remdir = ftp_curdir(ftp, FTP_REMOTE);
 111         if(remdir && strcmp(tui_get_title(uilist), remdir) != 0) {
 112                 tui_set_title(uilist, remdir);
 113                 upd |= 1;
 114         }
 115
 116         if(ftp->modified & FTP_MOD_REMDIR) {
 117                 tui_clear_list(uilist);
 118
 119                 num = ftp_num_dirent(ftp, FTP_REMOTE);
 120                 for(i=0; i<num; i++) {
 121                         ent = ftp_dirent(ftp, FTP_REMOTE, i);
 122                         if(ent->type == FTP_DIR) {
 123                                 sprintf(buf, "%s/", ent->name);
 124                                 tui_add_list_item(uilist, buf);
 125                         } else {
 126                                 tui_add_list_item(uilist, ent->name);
 127                         }
 128                 }
 129
 130                 tui_list_select(uilist, 0);
 131
 132                 ftp->modified &= ~FTP_MOD_REMDIR;
 133                 upd |= 1;
 134         }
 135
 136         if(tui_isdirty(uilist) || upd & 1) {
 137                 tui_draw(uilist);
 138         }
 139 }
 140
 141 int proc_input(void)
 142 {
 143         union event ev;
 144
 145         while(poll_input(&ev)) {
 146                 switch(ev.type) {
 147                 case EV_KEY:
 148                         if(keypress(ev.key.key) == -1) {
 149                                 return -1;
 150                         }
 151                         break;
 152
 153                 default:
 154                         break;
 155                 }
 156         }
 157         return 0;
 158 }
 159
 160 int keypress(int key)
 161 {
 162         int sel;
 163         const char *name;
 164
 165         switch(key) {
 166         case 27:
 167         case 'q':
 168                 return -1;
 169
 170         case KB_UP:
 171                 tui_list_sel_prev(uilist);
 172                 break;
 173         case KB_DOWN:
 174                 tui_list_sel_next(uilist);
 175                 break;
 176         case KB_LEFT:
 177                 tui_list_sel_start(uilist);
 178                 break;
 179         case KB_RIGHT:
 180                 tui_list_sel_end(uilist);
 181                 break;
 182
 183         case '\n':
 184                 sel = tui_get_list_sel(uilist);
 185                 name = ftp_dirent(ftp, FTP_REMOTE, sel)->name;
 186                 ftp_queue(ftp, FTP_CHDIR, name);
 187                 break;
 188
 189         case '\b':
 190                 infomsg("CDUP\n");
 191                 ftp_queue(ftp, FTP_CDUP, 0);
 192                 break;
 193
 194         default:
 195                 break;
 196         }
 197         return 0;
 198 }
 199
 200 static const char *usage = "Usage: %s [options] [hostname] [port]\n"
 201         "Options:\n"
 202         "  -h: print usage information and exit\n";
 203
 204 int parse_args(int argc, char **argv)
 205 {
 206         int i, argidx = 0;
 207
 208         for(i=1; i<argc; i++) {
 209                 if(argv[i][0] == '-') {
 210                         if(argv[i][2] != 0) {
 211                                 goto inval;
 212                         }
 213                         switch(argv[i][1]) {
 214                         case 'h':
 215                                 printf(usage, argv[0]);
 216                                 exit(0);
 217
 218                         default:
 219                                 goto inval;
 220                         }
 221
 222                 } else {
 223                         switch(argidx++) {
 224                         case 0:
 225                                 host = argv[i];
 226                                 break;
 227
 228                         case 1:
 229                                 if((port = atoi(argv[i])) <= 0) {
 230                                         fprintf(stderr, "invalid port number: %s\n", argv[i]);
 231                                         return -1;
 232                                 }
 233                                 break;
 234
 235                         default:
 236                                 goto inval;
 237                         }
 238                 }
 239         }
 240
 241         return 0;
 242 inval:
 243         fprintf(stderr, "invalid argument: %s\n", argv[i]);
 244         fprintf(stderr, usage, argv[0]);
 245         return -1;
 246 }