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