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 static int cmpnames(const void *a, const void *b)
 103 {
 104         const char *sa = *(const char**)a;
 105         const char *sb = *(const char**)b;
 106         infomsg("cmp(%s, %s)\n", sa, sb);
 107         return strcmp(sa, sb);
 108 }
 109
 110 void updateui(void)
 111 {
 112         struct ftp_dirent *ent;
 113         unsigned int upd = 0;
 114         char buf[128];
 115
 116         if(ftp->curdir_rem && strcmp(tui_get_title(uilist), ftp->curdir_rem) != 0) {
 117                 tui_set_title(uilist, ftp->curdir_rem);
 118                 upd |= 1;
 119         }
 120
 121         if(ftp->modified & FTP_MOD_REMDIR) {
 122                 tui_clear_list(uilist);
 123
 124                 ent = ftp->dent_rem;
 125                 while(ent) {
 126                         if(ent->type == FTP_DIR) {
 127                                 sprintf(buf, "%s/", ent->name);
 128                                 tui_add_list_item(uilist, buf);
 129                         } else {
 130                                 tui_add_list_item(uilist, ent->name);
 131                         }
 132                         ent = ent->next;
 133                 }
 134
 135                 //tui_sort_list(uilist, cmpnames);
 136                 tui_list_select(uilist, 0);
 137
 138                 ftp->modified &= ~FTP_MOD_REMDIR;
 139                 upd |= 1;
 140         }
 141
 142         if(tui_isdirty(uilist) || upd & 1) {
 143                 tui_draw(uilist);
 144         }
 145 }
 146
 147 int proc_input(void)
 148 {
 149         union event ev;
 150
 151         while(poll_input(&ev)) {
 152                 switch(ev.type) {
 153                 case EV_KEY:
 154                         if(keypress(ev.key.key) == -1) {
 155                                 return -1;
 156                         }
 157                         break;
 158
 159                 default:
 160                         break;
 161                 }
 162         }
 163         return 0;
 164 }
 165
 166 int keypress(int key)
 167 {
 168         switch(key) {
 169         case 27:
 170         case 'q':
 171                 return -1;
 172
 173         case KB_UP:
 174                 tui_list_sel_prev(uilist);
 175                 break;
 176         case KB_DOWN:
 177                 tui_list_sel_next(uilist);
 178                 break;
 179         case KB_LEFT:
 180                 tui_list_sel_start(uilist);
 181                 break;
 182         case KB_RIGHT:
 183                 tui_list_sel_end(uilist);
 184                 break;
 185
 186         default:
 187                 break;
 188         }
 189         return 0;
 190 }
 191
 192 static const char *usage = "Usage: %s [options] [hostname] [port]\n"
 193         "Options:\n"
 194         "  -h: print usage information and exit\n";
 195
 196 int parse_args(int argc, char **argv)
 197 {
 198         int i, argidx = 0;
 199
 200         for(i=1; i<argc; i++) {
 201                 if(argv[i][0] == '-') {
 202                         if(argv[i][2] != 0) {
 203                                 goto inval;
 204                         }
 205                         switch(argv[i][1]) {
 206                         case 'h':
 207                                 printf(usage, argv[0]);
 208                                 exit(0);
 209
 210                         default:
 211                                 goto inval;
 212                         }
 213
 214                 } else {
 215                         switch(argidx++) {
 216                         case 0:
 217                                 host = argv[i];
 218                                 break;
 219
 220                         case 1:
 221                                 if((port = atoi(argv[i])) <= 0) {
 222                                         fprintf(stderr, "invalid port number: %s\n", argv[i]);
 223                                         return -1;
 224                                 }
 225                                 break;
 226
 227                         default:
 228                                 goto inval;
 229                         }
 230                 }
 231         }
 232
 233         return 0;
 234 inval:
 235         fprintf(stderr, "invalid argument: %s\n", argv[i]);
 236         fprintf(stderr, usage, argv[0]);
 237         return -1;
 238 }