stream/stream_ftp.c

   1 /*
   2  * This file is part of MPlayer.
   3  *
   4  * MPlayer is free software; you can redistribute it and/or modify
   5  * it under the terms of the GNU General Public License as published by
   6  * the Free Software Foundation; either version 2 of the License, or
   7  * (at your option) any later version.
   8  *
   9  * MPlayer is distributed in the hope that it will be useful,
  10  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  11  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  12  * GNU General Public License for more details.
  13  *
  14  * You should have received a copy of the GNU General Public License along
  15  * with MPlayer; if not, write to the Free Software Foundation, Inc.,
  16  * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
  17  */
  18
  19 #include "config.h"
  20
  21 #include <stdlib.h>
  22 #include <stdio.h>
  23
  24 #include <sys/types.h>
  25 #include <sys/stat.h>
  26 #include <fcntl.h>
  27 #include <unistd.h>
  28 #include <errno.h>
  29 #if !HAVE_WINSOCK2_H
  30 #include <sys/socket.h>
  31 #else
  32 #include <winsock2.h>
  33 #endif
  34
  35 #include "mp_msg.h"
  36 #include "network.h"
  37 #include "stream.h"
  38 #include "m_option.h"
  39 #include "m_struct.h"
  40 #include "tcp.h"
  41
  42 static struct stream_priv_s {
  43   char* user;
  44   char* pass;
  45   char* host;
  46   int port;
  47   char* filename;
  48
  49   char *cput,*cget;
  50   int handle;
  51   int cavail,cleft;
  52   char *buf;
  53 } stream_priv_dflts = {
  54   "anonymous","no@spam",
  55   NULL,
  56   21,
  57   NULL,
  58   NULL,
  59   NULL,
  60
  61   0,
  62   0,0,
  63   NULL
  64 };
  65
  66 #define BUFSIZE 2048
  67
  68 #define ST_OFF(f) M_ST_OFF(struct stream_priv_s,f)
  69 /// URL definition
  70 static const m_option_t stream_opts_fields[] = {
  71   {"username", ST_OFF(user), CONF_TYPE_STRING, 0, 0 ,0, NULL},
  72   {"password", ST_OFF(pass), CONF_TYPE_STRING, 0, 0 ,0, NULL},
  73   {"hostname", ST_OFF(host), CONF_TYPE_STRING, 0, 0 ,0, NULL},
  74   {"port", ST_OFF(port), CONF_TYPE_INT, 0, 0 ,65635, NULL},
  75   {"filename", ST_OFF(filename), CONF_TYPE_STRING, 0, 0 ,0, NULL},
  76   { NULL, NULL, 0, 0, 0, 0,  NULL }
  77 };
  78 static const struct m_struct_st stream_opts = {
  79   "ftp",
  80   sizeof(struct stream_priv_s),
  81   &stream_priv_dflts,
  82   stream_opts_fields
  83 };
  84
  85 #define TELNET_IAC      255             /* interpret as command: */
  86 #define TELNET_IP       244             /* interrupt process--permanently */
  87 #define TELNET_SYNCH    242             /* for telfunc calls */
  88
  89 // Check if there is something to read on a fd. This avoid hanging
  90 // forever if the network stop responding.
  91 static int fd_can_read(int fd,int timeout) {
  92   fd_set fds;
  93   struct timeval tv;
  94
  95   FD_ZERO(&fds);
  96   FD_SET(fd,&fds);
  97   tv.tv_sec = timeout;
  98   tv.tv_usec = 0;
  99
 100   return select(fd+1, &fds, NULL, NULL, &tv) > 0;
 101 }
 102
 103 /*
 104  * read a line of text
 105  *
 106  * return -1 on error or bytecount
 107  */
 108 static int readline(char *buf,int max,struct stream_priv_s *ctl)
 109 {
 110     int x,retval = 0;
 111     char *end,*bp=buf;
 112     int eof = 0;
 113
 114     do {
 115       if (ctl->cavail > 0) {
 116         x = (max >= ctl->cavail) ? ctl->cavail : max-1;
 117         end = memccpy(bp,ctl->cget,'\n',x);
 118         if (end != NULL)
 119           x = end - bp;
 120         retval += x;
 121         bp += x;
 122         *bp = '\0';
 123         max -= x;
 124         ctl->cget += x;
 125         ctl->cavail -= x;
 126         if (end != NULL) {
 127           bp -= 2;
 128           if (strcmp(bp,"\r\n") == 0) {
 129             *bp++ = '\n';
 130             *bp++ = '\0';
 131             --retval;
 132           }
 133           break;
 134         }
 135       }
 136       if (max == 1) {
 137         *buf = '\0';
 138         break;
 139       }
 140       if (ctl->cput == ctl->cget) {
 141         ctl->cput = ctl->cget = ctl->buf;
 142         ctl->cavail = 0;
 143         ctl->cleft = BUFSIZE;
 144       }
 145       if(eof) {
 146         if (retval == 0)
 147           retval = -1;
 148         break;
 149       }
 150
 151       if(!fd_can_read(ctl->handle, 15)) {
 152         mp_msg(MSGT_OPEN,MSGL_ERR, "[ftp] read timed out\n");
 153         retval = -1;
 154         break;
 155       }
 156
 157       if ((x = recv(ctl->handle,ctl->cput,ctl->cleft,0)) == -1) {
 158         mp_msg(MSGT_STREAM,MSGL_ERR, "[ftp] read error: %s\n",strerror(errno));
 159         retval = -1;
 160         break;
 161       }
 162       if (x == 0)
 163         eof = 1;
 164       ctl->cleft -= x;
 165       ctl->cavail += x;
 166       ctl->cput += x;
 167     } while (1);
 168
 169     return retval;
 170 }
 171
 172 /*
 173  * read a response from the server
 174  *
 175  * return 0 if first char doesn't match
 176  * return 1 if first char matches
 177  */
 178 static int readresp(struct stream_priv_s* ctl,char* rsp)
 179 {
 180     static char response[256];
 181     char match[5];
 182     int r;
 183
 184     if (readline(response,256,ctl) == -1)
 185       return 0;
 186
 187     r = atoi(response)/100;
 188     if(rsp) strcpy(rsp,response);
 189
 190     mp_msg(MSGT_STREAM,MSGL_V, "[ftp] < %s",response);
 191
 192     if (response[3] == '-') {
 193       strncpy(match,response,3);
 194       match[3] = ' ';
 195       match[4] = '\0';
 196       do {
 197         if (readline(response,256,ctl) == -1) {
 198           mp_msg(MSGT_OPEN,MSGL_ERR, "[ftp] Control socket read failed\n");
 199           return 0;
 200         }
 201         mp_msg(MSGT_OPEN,MSGL_V, "[ftp] < %s",response);
 202       } while (strncmp(response,match,4));
 203     }
 204     return r;
 205 }
 206
 207
 208 static int FtpSendCmd(const char *cmd, struct stream_priv_s *nControl,char* rsp)
 209 {
 210   int l = strlen(cmd);
 211   int hascrlf = cmd[l - 2] == '\r' && cmd[l - 1] == '\n';
 212
 213   if(hascrlf && l == 2) mp_msg(MSGT_STREAM,MSGL_V, "\n");
 214   else mp_msg(MSGT_STREAM,MSGL_V, "[ftp] > %s",cmd);
 215   while(l > 0) {
 216     int s = send(nControl->handle,cmd,l,DEFAULT_SEND_FLAGS);
 217
 218     if(s <= 0) {
 219       mp_msg(MSGT_OPEN,MSGL_ERR, "[ftp] write error: %s\n",strerror(errno));
 220       return 0;
 221     }
 222
 223     cmd += s;
 224     l -= s;
 225   }
 226
 227   if (hascrlf)
 228     return readresp(nControl,rsp);
 229   else
 230     return FtpSendCmd("\r\n", nControl, rsp);
 231 }
 232
 233 static int FtpOpenPort(struct stream_priv_s* p) {
 234   int resp,fd;
 235   char rsp_txt[256];
 236   char* par,str[128];
 237   int num[6];
 238
 239   resp = FtpSendCmd("PASV",p,rsp_txt);
 240   if(resp != 2) {
 241     mp_msg(MSGT_OPEN,MSGL_WARN, "[ftp] command 'PASV' failed: %s\n",rsp_txt);
 242     return 0;
 243   }
 244
 245   par = strchr(rsp_txt,'(');
 246
 247   if(!par || !par[0] || !par[1]) {
 248     mp_msg(MSGT_OPEN,MSGL_ERR, "[ftp] invalid server response: %s ??\n",rsp_txt);
 249     return 0;
 250   }
 251
 252   sscanf(par+1,"%u,%u,%u,%u,%u,%u",&num[0],&num[1],&num[2],
 253          &num[3],&num[4],&num[5]);
 254   snprintf(str,127,"%d.%d.%d.%d",num[0],num[1],num[2],num[3]);
 255   fd = connect2Server(str,(num[4]<<8)+num[5],0);
 256
 257   if(fd < 0)
 258     mp_msg(MSGT_OPEN,MSGL_ERR, "[ftp] failed to create data connection\n");
 259
 260   return fd;
 261 }
 262
 263 static int FtpOpenData(stream_t* s,off_t newpos) {
 264   struct stream_priv_s* p = s->priv;
 265   int resp;
 266   char str[256],rsp_txt[256];
 267
 268   // Open a new connection
 269   s->fd = FtpOpenPort(p);
 270
 271   if(s->fd < 0) return 0;
 272
 273   if(newpos > 0) {
 274     snprintf(str,255,"REST %"PRId64, (int64_t)newpos);
 275
 276     resp = FtpSendCmd(str,p,rsp_txt);
 277     if(resp != 3) {
 278       mp_msg(MSGT_OPEN,MSGL_WARN, "[ftp] command '%s' failed: %s\n",str,rsp_txt);
 279       newpos = 0;
 280     }
 281   }
 282
 283   // Get the file
 284   snprintf(str,255,"RETR %s",p->filename);
 285   resp = FtpSendCmd(str,p,rsp_txt);
 286
 287   if(resp != 1) {
 288     mp_msg(MSGT_OPEN,MSGL_ERR, "[ftp] command '%s' failed: %s\n",str,rsp_txt);
 289     return 0;
 290   }
 291
 292   s->pos = newpos;
 293   return 1;
 294 }
 295
 296 static int fill_buffer(stream_t *s, char* buffer, int max_len){
 297   int r;
 298
 299   if(s->fd < 0 && !FtpOpenData(s,s->pos))
 300     return -1;
 301
 302   if(!fd_can_read(s->fd, 15)) {
 303     mp_msg(MSGT_OPEN,MSGL_ERR, "[ftp] read timed out\n");
 304     return -1;
 305   }
 306
 307   r = recv(s->fd,buffer,max_len,0);
 308   return (r <= 0) ? -1 : r;
 309 }
 310
 311 static int seek(stream_t *s,off_t newpos) {
 312   struct stream_priv_s* p = s->priv;
 313   int resp;
 314   char rsp_txt[256];
 315
 316   if(s->pos > s->end_pos) {
 317     s->eof=1;
 318     return 0;
 319   }
 320
 321   // Check to see if the server did not already terminate the transfer
 322   if(fd_can_read(p->handle, 0)) {
 323     if(readresp(p,rsp_txt) != 2)
 324       mp_msg(MSGT_OPEN,MSGL_WARN, "[ftp] Warning the server didn't finished the transfer correctly: %s\n",rsp_txt);
 325     closesocket(s->fd);
 326     s->fd = -1;
 327   }
 328
 329   // Close current download
 330   if(s->fd >= 0) {
 331     static const char pre_cmd[]={TELNET_IAC,TELNET_IP,TELNET_IAC,TELNET_SYNCH};
 332     //int fl;
 333
 334     // First close the fd
 335     closesocket(s->fd);
 336     s->fd = 0;
 337
 338     // Send send the telnet sequence needed to make the server react
 339
 340     // Dunno if this is really needed, lftp have it. I let
 341     // it here in case it turn out to be needed on some other OS
 342     //fl=fcntl(p->handle,F_GETFL);
 343     //fcntl(p->handle,F_SETFL,fl&~O_NONBLOCK);
 344
 345     // send only first byte as OOB due to OOB braindamage in many unices
 346     send(p->handle,pre_cmd,1,MSG_OOB|DEFAULT_SEND_FLAGS);
 347     send(p->handle,pre_cmd+1,sizeof(pre_cmd)-1,DEFAULT_SEND_FLAGS);
 348
 349     //fcntl(p->handle,F_SETFL,fl);
 350
 351     // Get the 426 Transfer aborted
 352     // Or the 226 Transfer complete
 353     resp = readresp(p,rsp_txt);
 354     if(resp != 4 && resp != 2) {
 355       mp_msg(MSGT_OPEN,MSGL_ERR, "[ftp] Server didn't abort correctly: %s\n",rsp_txt);
 356       s->eof = 1;
 357       return 0;
 358     }
 359     // Send the ABOR command
 360     // Ignore the return code as sometimes it fail with "nothing to abort"
 361     FtpSendCmd("ABOR",p,rsp_txt);
 362   }
 363   return FtpOpenData(s,newpos);
 364 }
 365
 366
 367 static void close_f(stream_t *s) {
 368   struct stream_priv_s* p = s->priv;
 369
 370   if(!p) return;
 371
 372   if(s->fd > 0) {
 373     closesocket(s->fd);
 374     s->fd = 0;
 375   }
 376
 377   FtpSendCmd("QUIT",p,NULL);
 378
 379   if(p->handle) closesocket(p->handle);
 380   free(p->buf);
 381
 382   m_struct_free(&stream_opts,p);
 383 }
 384
 385
 386
 387 static int open_f(stream_t *stream,int mode, void* opts, int* file_format) {
 388   int resp;
 389   int64_t len = 0;
 390   struct stream_priv_s* p = (struct stream_priv_s*)opts;
 391   char str[256],rsp_txt[256];
 392
 393   if(mode != STREAM_READ) {
 394     mp_msg(MSGT_OPEN,MSGL_ERR, "[ftp] Unknown open mode %d\n",mode);
 395     m_struct_free(&stream_opts,opts);
 396     return STREAM_UNSUPPORTED;
 397   }
 398
 399   if(!p->filename || !p->host) {
 400     mp_msg(MSGT_OPEN,MSGL_ERR, "[ftp] Bad url\n");
 401     m_struct_free(&stream_opts,opts);
 402     return STREAM_ERROR;
 403   }
 404
 405   // Open the control connection
 406   p->handle = connect2Server(p->host,p->port,1);
 407
 408   if(p->handle < 0) {
 409     m_struct_free(&stream_opts,opts);
 410     return STREAM_ERROR;
 411   }
 412
 413   // We got a connection, let's start serious things
 414   stream->fd = -1;
 415   stream->priv = p;
 416   p->buf = malloc(BUFSIZE);
 417
 418   if (readresp(p, NULL) == 0) {
 419     close_f(stream);
 420     return STREAM_ERROR;
 421   }
 422
 423   // Login
 424   snprintf(str,255,"USER %s",p->user);
 425   resp = FtpSendCmd(str,p,rsp_txt);
 426
 427   // password needed
 428   if(resp == 3) {
 429     snprintf(str,255,"PASS %s",p->pass);
 430     resp = FtpSendCmd(str,p,rsp_txt);
 431     if(resp != 2) {
 432       mp_msg(MSGT_OPEN,MSGL_ERR, "[ftp] command '%s' failed: %s\n",str,rsp_txt);
 433       close_f(stream);
 434       return STREAM_ERROR;
 435     }
 436   } else if(resp != 2) {
 437     mp_msg(MSGT_OPEN,MSGL_ERR, "[ftp] command '%s' failed: %s\n",str,rsp_txt);
 438     close_f(stream);
 439     return STREAM_ERROR;
 440   }
 441
 442   // Set the transfer type
 443   resp = FtpSendCmd("TYPE I",p,rsp_txt);
 444   if(resp != 2) {
 445     mp_msg(MSGT_OPEN,MSGL_WARN, "[ftp] command 'TYPE I' failed: %s\n",rsp_txt);
 446     close_f(stream);
 447     return STREAM_ERROR;
 448   }
 449
 450   // Get the filesize
 451   snprintf(str,255,"SIZE %s",p->filename);
 452   resp = FtpSendCmd(str,p,rsp_txt);
 453   if(resp != 2) {
 454     mp_msg(MSGT_OPEN,MSGL_WARN, "[ftp] command '%s' failed: %s\n",str,rsp_txt);
 455   } else {
 456     int dummy;
 457     sscanf(rsp_txt,"%d %"SCNd64,&dummy,&len);
 458   }
 459
 460   if(len > 0) {
 461     stream->seek = seek;
 462     stream->end_pos = len;
 463   }
 464
 465   // The data connection is really opened only at the first
 466   // read/seek. This must be done when the cache is used
 467   // because the connection would stay open in the main process,
 468   // preventing correct abort with many servers.
 469   stream->fd = -1;
 470   stream->priv = p;
 471   stream->fill_buffer = fill_buffer;
 472   stream->close = close_f;
 473
 474   return STREAM_OK;
 475 }
 476
 477 const stream_info_t stream_info_ftp = {
 478   "File Transfer Protocol",
 479   "ftp",
 480   "Albeu",
 481   "reuse a bit of code from ftplib written by Thomas Pfau",
 482   open_f,
 483   { "ftp", NULL },
 484   &stream_opts,
 485   1 // Urls are an option string
 486 };