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 #ifdef __DOS__
   7 #include <direct.h>
   8 #else
   9 #include <unistd.h>
  10 #include <dirent.h>
  11 #endif
  12 #include <sys/stat.h>
  13 #include <sys/select.h>
  14 #include "tgfx.h"
  15 #include "input.h"
  16 #include "util.h"
  17 #include "tui.h"
  18 #include "ftp.h"
  19 #include "darray.h"
  20
  21 #ifdef __DOS__
  22 #define select  select_s
  23 #endif
  24
  25 void updateui(void);
  26 int update_localdir(void);
  27 int proc_input(void);
  28 int keypress(int key);
  29 int parse_args(int argc, char **argv);
  30
  31 static struct ftp *ftp;
  32 static struct tui_widget *uilist[2];
  33
  34 static int focus;
  35
  36 static char *host = "localhost";
  37 static int port = 21;
  38
  39 static char curdir[PATH_MAX + 1];
  40 static struct ftp_dirent *localdir;
  41
  42
  43 int main(int argc, char **argv)
  44 {
  45         int i, numsock, maxfd;
  46         int ftpsock[16];
  47         fd_set rdset;
  48         struct timeval tv;
  49
  50         if(parse_args(argc, argv) == -1) {
  51                 return 1;
  52         }
  53
  54         localdir = darr_alloc(0, sizeof *localdir);
  55         getcwd(curdir, sizeof curdir);
  56         update_localdir();
  57
  58         if(!(ftp = ftp_alloc())) {
  59                 return 1;
  60         }
  61         if(ftp_connect(ftp, host, port) == -1) {
  62                 ftp_free(ftp);
  63                 return 1;
  64         }
  65
  66         init_input();
  67
  68         tg_init();
  69
  70         tg_bgchar(' ');
  71         tg_clear();
  72
  73         uilist[0] = tui_list("Remote", 0, 0, 40, 23, 0, 0);
  74         uilist[1] = tui_list("Local", 40, 0, 40, 23, 0, 0);
  75         focus = 0;
  76         tui_focus(uilist[focus], 1);
  77
  78         tg_setcursor(0, 23);
  79
  80         tui_draw(uilist[0]);
  81         tui_draw(uilist[1]);
  82
  83         for(;;) {
  84                 FD_ZERO(&rdset);
  85                 maxfd = 0;
  86
  87                 numsock = ftp_sockets(ftp, ftpsock, sizeof ftpsock);
  88                 for(i=0; i<numsock; i++) {
  89                         FD_SET(ftpsock[i], &rdset);
  90                         if(ftpsock[i] > maxfd) maxfd = ftpsock[i];
  91                 }
  92
  93 #ifdef __unix__
  94                 FD_SET(0, &rdset);
  95                 tv.tv_sec = 120;
  96                 tv.tv_usec = 0;
  97 #else
  98                 tv.tv_sec = tv.tv_usec = 0;
  99 #endif
 100
 101                 if(select(maxfd + 1, &rdset, 0, 0, &tv) == -1 && errno == EINTR) {
 102                         continue;
 103                 }
 104
 105 #ifdef __unix__
 106                 if(FD_ISSET(0, &rdset)) {
 107                         if(proc_input() == -1) {
 108                                 break;
 109                         }
 110                 }
 111 #else
 112                 if(proc_input() == -1) {
 113                         break;
 114                 }
 115 #endif
 116
 117                 for(i=0; i<numsock; i++) {
 118                         if(FD_ISSET(ftpsock[i], &rdset)) {
 119                                 ftp_handle(ftp, ftpsock[i]);
 120                         }
 121                 }
 122
 123                 updateui();
 124         }
 125
 126         tg_cleanup();
 127         cleanup_input();
 128         ftp_close(ftp);
 129         ftp_free(ftp);
 130         return 0;
 131 }
 132
 133 void updateui(void)
 134 {
 135         int i, num;
 136         struct ftp_dirent *ent;
 137         unsigned int upd = 0;
 138         char buf[128];
 139         const char *remdir;
 140
 141         remdir = ftp_curdir(ftp);
 142         if(remdir && strcmp(tui_get_title(uilist[0]), remdir) != 0) {
 143                 tui_set_title(uilist[0], remdir);
 144                 upd |= 1;
 145         }
 146
 147         if(ftp->modified & FTP_MOD_DIR) {
 148                 tui_clear_list(uilist[0]);
 149
 150                 num = ftp_num_dirent(ftp);
 151                 for(i=0; i<num; i++) {
 152                         ent = ftp_dirent(ftp, i);
 153                         if(ent->type == FTP_DIR) {
 154                                 sprintf(buf, "%s/", ent->name);
 155                                 tui_add_list_item(uilist[0], buf);
 156                         } else {
 157                                 tui_add_list_item(uilist[0], ent->name);
 158                         }
 159                 }
 160
 161                 tui_list_select(uilist[0], 0);
 162
 163                 ftp->modified &= ~FTP_MOD_DIR;
 164                 upd |= 1;
 165         }
 166
 167         if(strcmp(tui_get_title(uilist[1]), curdir) != 0) {
 168                 tui_clear_list(uilist[1]);
 169                 num = darr_size(localdir);
 170                 for(i=0; i<num; i++) {
 171                         ent = localdir + i;
 172                         if(ent->type == FTP_DIR) {
 173                                 sprintf(buf, "%s/", ent->name);
 174                                 tui_add_list_item(uilist[1], buf);
 175                         } else {
 176                                 tui_add_list_item(uilist[1], ent->name);
 177                         }
 178                 }
 179                 tui_list_select(uilist[1], 0);
 180                 tui_set_title(uilist[1], curdir);
 181                 upd |= 2;
 182         }
 183
 184         if(tui_isdirty(uilist[0]) || upd & 1) {
 185                 tui_draw(uilist[0]);
 186         }
 187         if(tui_isdirty(uilist[1]) || upd & 2) {
 188                 tui_draw(uilist[1]);
 189         }
 190 }
 191
 192 int update_localdir(void)
 193 {
 194         DIR *dir;
 195         struct dirent *dent;
 196         struct stat st;
 197         struct ftp_dirent ent;
 198
 199         if(!(dir = opendir(curdir))) {
 200                 errmsg("failed to open directory: %s\n", curdir);
 201                 return -1;
 202         }
 203
 204         darr_clear(localdir);
 205         while((dent = readdir(dir))) {
 206                 ent.name = strdup_nf(dent->d_name);
 207                 if(strcmp(dent->d_name, ".") == 0) continue;
 208
 209                 if(stat(dent->d_name, &st) == 0) {
 210                         if(S_ISDIR(st.st_mode)) {
 211                                 ent.type = FTP_DIR;
 212                         } else {
 213                                 ent.type = FTP_FILE;
 214                         }
 215                         ent.size = st.st_size;
 216                 } else {
 217                         ent.type = FTP_FILE;
 218                         ent.size = 0;
 219                 }
 220
 221                 darr_push(localdir, &ent);
 222         }
 223         closedir(dir);
 224
 225         qsort(localdir, darr_size(localdir), sizeof *localdir, ftp_direntcmp);
 226         return 0;
 227 }
 228
 229 int proc_input(void)
 230 {
 231         union event ev;
 232
 233         while(poll_input(&ev)) {
 234                 switch(ev.type) {
 235                 case EV_KEY:
 236                         if(keypress(ev.key.key) == -1) {
 237                                 return -1;
 238                         }
 239                         break;
 240
 241                 default:
 242                         break;
 243                 }
 244         }
 245         return 0;
 246 }
 247
 248 int keypress(int key)
 249 {
 250         int sel;
 251
 252         switch(key) {
 253         case 27:
 254         case 'q':
 255         case KB_F10:
 256                 return -1;
 257
 258         case '\t':
 259                 tui_focus(uilist[focus], 0);
 260                 focus ^= 1;
 261                 tui_focus(uilist[focus], 1);
 262                 break;
 263
 264         case KB_UP:
 265                 tui_list_sel_prev(uilist[focus]);
 266                 break;
 267         case KB_DOWN:
 268                 tui_list_sel_next(uilist[focus]);
 269                 break;
 270         case KB_LEFT:
 271                 tui_list_sel_start(uilist[focus]);
 272                 break;
 273         case KB_RIGHT:
 274                 tui_list_sel_end(uilist[focus]);
 275                 break;
 276
 277         case '\n':
 278                 sel = tui_get_list_sel(uilist[focus]);
 279                 if(focus == 0) {
 280                         struct ftp_dirent *ent = ftp_dirent(ftp, sel);
 281                         if(ent->type == FTP_DIR) {
 282                                 ftp_queue(ftp, FTP_CHDIR, ent->name);
 283                         } else {
 284                                 /* TODO */
 285                         }
 286                 } else {
 287                         if(localdir[sel].type == FTP_DIR) {
 288                                 if(chdir(localdir[sel].name) == -1) {
 289                                         errmsg("failed to change directory: %s\n", localdir[sel].name);
 290                                 } else {
 291                                         getcwd(curdir, sizeof curdir);
 292                                         update_localdir();
 293                                 }
 294                         } else {
 295                                 /* TODO */
 296                         }
 297                 }
 298                 break;
 299
 300         case '\b':
 301                 if(focus == 0) {
 302                         ftp_queue(ftp, FTP_CDUP, 0);
 303                 } else {
 304                         if(chdir("..") == 0) {
 305                                 getcwd(curdir, sizeof curdir);
 306                                 update_localdir();
 307                         }
 308                 }
 309                 break;
 310
 311         default:
 312                 break;
 313         }
 314         return 0;
 315 }
 316
 317 static const char *usage = "Usage: %s [options] [hostname] [port]\n"
 318         "Options:\n"
 319         "  -h: print usage information and exit\n";
 320
 321 int parse_args(int argc, char **argv)
 322 {
 323         int i, argidx = 0;
 324
 325         for(i=1; i<argc; i++) {
 326                 if(argv[i][0] == '-') {
 327                         if(argv[i][2] != 0) {
 328                                 goto inval;
 329                         }
 330                         switch(argv[i][1]) {
 331                         case 'h':
 332                                 printf(usage, argv[0]);
 333                                 exit(0);
 334
 335                         default:
 336                                 goto inval;
 337                         }
 338
 339                 } else {
 340                         switch(argidx++) {
 341                         case 0:
 342                                 host = argv[i];
 343                                 break;
 344
 345                         case 1:
 346                                 if((port = atoi(argv[i])) <= 0) {
 347                                         fprintf(stderr, "invalid port number: %s\n", argv[i]);
 348                                         return -1;
 349                                 }
 350                                 break;
 351
 352                         default:
 353                                 goto inval;
 354                         }
 355                 }
 356         }
 357
 358         return 0;
 359 inval:
 360         fprintf(stderr, "invalid argument: %s\n", argv[i]);
 361         fprintf(stderr, usage, argv[0]);
 362         return -1;
 363 }