minix/commands/urlget/urlget.c
2009-11-09 10:26:00 +00:00

629 lines
12 KiB
C

/* urlget.c Copyright 2000 by Michael Temari All Rights Reserved */
/* 04/05/2000 Michael Temari <Michael@TemWare.Com> */
#include <sys/types.h>
#include <sys/ioctl.h>
#include <sys/wait.h>
#include <stdio.h>
#include <ctype.h>
#include <stdlib.h>
#include <string.h>
#include <errno.h>
#include <fcntl.h>
#include <signal.h>
#include <unistd.h>
#include <time.h>
#include "net.h"
_PROTOTYPE(char *unesc, (char *s));
_PROTOTYPE(void encode64, (char **pp, char *s));
_PROTOTYPE(char *auth, (char *user, char *pass));
_PROTOTYPE(int skipit, (char *buf, int len, int *skip));
_PROTOTYPE(int httpget, (char *host, int port, char *user, char *pass, char *path, int headers, int discard, int post));
_PROTOTYPE(void ftppasv, (char *reply));
_PROTOTYPE(int ftpreply, (FILE *fpr));
_PROTOTYPE(int ftpcmd, (FILE *fpw, FILE *fpr, char *cmd, char *arg));
_PROTOTYPE(int ftpget, (char *host, int port, char *user, char *pass, char *path, int type));
_PROTOTYPE(int tcpget, (char *host, int port, char *user, char *pass, char *path));
_PROTOTYPE(int main, (int argc, char *argv[]));
char ftpphost[15+1];
unsigned int ftppport;
#define SCHEME_HTTP 1
#define SCHEME_FTP 2
#define SCHEME_TCP 3
#define SCHEME_NNTP 4
char buffer[16000];
#if 0
_PROTOTYPE(int strncasecmp, (const char *s1, const char *s2, size_t len));
int
strncasecmp(s1, s2, len)
const char *s1, *s2;
size_t len;
{
int c1, c2;
do {
if (len == 0)
return 0;
len--;
} while (c1= toupper(*s1++), c2= toupper(*s2++), c1 == c2 && (c1 & c2))
;
if (c1 & c2)
return c1 < c2 ? -1 : 1;
return c1 ? 1 : (c2 ? -1 : 0);
}
#endif
char *unesc(s)
char *s;
{
char *p;
char *p2;
unsigned char c;
p = s;
p2 = s;
while(*p) {
if(*p != '%') {
*p2++ = *p++;
continue;
}
p++;
if(*p == '%') {
*p2++ = *p++;
continue;
}
if(*p >= '0' && *p <= '9') c = *p++ - '0'; else
if(*p >= 'a' && *p <= 'f') c = *p++ - 'a' + 10; else
if(*p >= 'A' && *p <= 'F') c = *p++ - 'A' + 10; else
break;
if(*p >= '0' && *p <= '9') c = c << 4 | (*p++ - '0'); else
if(*p >= 'a' && *p <= 'f') c = c << 4 | (*p++ - 'a') + 10; else
if(*p >= 'A' && *p <= 'F') c = c << 4 | (*p++ - 'A') + 10; else
break;
*p2++ = c;
}
*p2 = '\0';
return(s);
}
void encode64(pp, s)
char **pp;
char *s;
{
char *p;
char c[3];
int i;
int len;
static char e64[64] = {
'A','B','C','D','E','F','G','H','I','J','K','L','M',
'N','O','P','Q','R','S','T','U','V','W','X','Y','Z',
'a','b','c','d','e','f','g','h','i','j','k','l','m',
'n','o','p','q','r','s','t','u','v','w','x','y','z',
'0','1','2','3','4','5','6','7','8','9','+','/' };
p = *pp;
len = strlen(s);
for(i=0; i < len; i += 3) {
c[0] = *s++;
c[1] = *s++;
c[2] = *s++;
*p++ = e64[ c[0] >> 2];
*p++ = e64[((c[0] << 4) & 0x30) | ((c[1] >> 4) & 0x0f)];
*p++ = e64[((c[1] << 2) & 0x3c) | ((c[2] >> 6) & 0x03)];
*p++ = e64[ c[2] & 0x3f];
}
if(i == len+1)
p[-1] = '=';
else
if(i == len+2) {
p[-1] = '=';
p[-2] = '=';
}
*p = '\0';
*pp = p;
return;
}
char *auth(user, pass)
char *user;
char *pass;
{
static char a[128];
char up[128];
char *p;
strcpy(a, "BASIC ");
p = a + 6;
sprintf(up, "%s:%s", user, pass);
encode64(&p, up);
return(a);
}
int skipit(buf, len, skip)
char *buf;
int len;
int *skip;
{
static int lf = 0;
static int crlf = 0;
char *p;
p = buf;
while(--len >= 0) {
if((crlf == 0 || crlf == 2) && *p == '\r')
crlf++;
else
if((crlf == 1 || crlf == 3) && *p == '\n')
crlf++;
else
crlf = 0;
if(*p == '\n')
lf++;
else
lf = 0;
if(crlf == 4 || lf == 2) {
*skip = 0;
return(len);
}
p++;
}
return(0);
}
int httpget(host, port, user, pass, path, headers, discard, post)
char *host;
int port;
char *user;
char *pass;
char *path;
int headers;
int discard;
int post;
{
int fd;
int skip;
int s;
int s2;
char *a;
char *qs;
int len;
if(port == 0)
port = 80;
fd = connect(host, port);
if(fd < 0) {
fprintf(stderr, "httpget: Could not connect to %s:%d\n", host, port);
return(-1);
}
if(post) {
qs = strrchr(path, '?');
if(qs != (char *)NULL) {
*qs++ = '\0';
len = strlen(qs);
} else
len = 0;
}
if(post && len > 0)
write(fd, "POST ", 5);
else
write(fd, "GET ", 4);
write(fd, path, strlen(path));
write(fd, " HTTP/1.0\r\n", 11);
write(fd, "User-Agent: urlget\r\n", 20);
write(fd, "Connection: Close\r\n", 19);
if(*user) {
write(fd, "Authorization: ", 15);
a = auth(user, pass);
write(fd, a, strlen(a));
write(fd, "\r\n", 2);
}
if(post && len > 0) {
sprintf(buffer, "Content-Length: %u\r\n", len);
write(fd, buffer, strlen(buffer));
}
write(fd, "Host: ", 6);
write(fd, host, strlen(host));
write(fd, "\r\n", 2);
write(fd, "\r\n", 2);
if(post && len > 0)
write(fd, qs, len);
skip = 1;
while((s = read(fd, buffer, sizeof(buffer)-1)) > 0) {
buffer[s] = '\0';
if(skip) {
static int firstline = 1;
if(firstline) {
static char linebuf[1000];
int l = 0;
int c, v1, v2, e;
if(s >= sizeof(linebuf)-l)
c = sizeof(linebuf)-1-l;
else c = s;
memcpy(linebuf+l, buffer, c);
linebuf[l+c] = '\0';
if(strchr(buffer, '\n') || strchr(buffer, '\r'))
firstline = 0;
if(sscanf(linebuf, "HTTP/%d.%d %d ", &v1, &v2, &e) == 3
&& e != 200) {
fprintf(stderr, "HTTP error %d\n", e);
return -1;
}
}
s2 = skipit(buffer, s, &skip);
if(headers)
write(1, buffer, s - s2);
} else
s2 = s;
if(s2 && !discard)
if(write(1, &buffer[s - s2], s2) != s2) {
perror("write");
return(-1);
}
}
if(s < 0) {
fprintf(stderr, "httpget: Read error\n");
return(-1);
}
close(fd);
return(0);
}
void ftppasv(reply)
char *reply;
{
char *p;
unsigned char n[6];
int i;
ftppport = 0;
p = reply;
while(*p && *p != '(') p++;
if(!*p) return;
p++;
i = 0;
while(1) {
n[i++] = atoi(p);
if(i == 6) break;
p = strchr(p, ',');
if(p == (char *)NULL) return;
p++;
}
sprintf(ftpphost, "%d.%d.%d.%d", n[0], n[1], n[2], n[3]);
ftppport = n[4] * 256 + n[5];
return;
}
int ftpreply(fpr)
FILE *fpr;
{
static char reply[256];
int s;
char code[4];
int ft;
do {
ft = 1;
do {
if(fgets(reply, sizeof(reply), fpr) == (char *)NULL)
return(-1);
if(ft) {
ft = 0;
strncpy(code, reply, 3);
code[3] = '\0';
}
} while(strncmp(reply, code, 3) || reply[3] == '-');
s = atoi(code);
} while(s < 200 && s != 125 && s != 150);
if(s == 227) ftppasv(reply);
return(s);
}
int ftpcmd(fpw, fpr, cmd, arg)
FILE *fpw;
FILE *fpr;
char *cmd;
char *arg;
{
fprintf(fpw, "%s%s%s\r\n", cmd, *arg ? " " : "", arg);
fflush(fpw);
return(ftpreply(fpr));
}
int ftpget(host, port, user, pass, path, type)
char *host;
int port;
char *user;
char *pass;
char *path;
int type;
{
int fd;
int fd2;
FILE *fpr;
FILE *fpw;
int s;
int s2;
char *p;
char *p2;
char typec[2];
if(port == 0)
port = 21;
if(type == '\0')
type = 'i';
fd = connect(host, port);
if(fd < 0) {
fprintf(stderr, "ftpget: Could not connect to %s:%d\n", host, port);
return(-1);
}
fpr = fdopen(fd, "r");
fpw = fdopen(fd, "w");
s = ftpreply(fpr);
if(s / 100 != 2) goto error;
s = ftpcmd(fpw, fpr, "USER", *user ? user : "ftp");
if(s / 100 == 3)
s = ftpcmd(fpw, fpr, "PASS", *pass ? pass : "urlget@");
if(s / 100 != 2) goto error;
p = path;
if(*p == '/') p++;
while((p2 = strchr(p, '/')) != (char *)NULL) {
*p2++ = '\0';
s = ftpcmd(fpw, fpr, "CWD", unesc(p));
p = p2;
}
sprintf(typec, "%c", type == 'd' ? 'A' : type);
s = ftpcmd(fpw, fpr, "TYPE", typec);
if(s / 100 != 2) goto error;
s = ftpcmd(fpw, fpr, "PASV", "");
if(s != 227) goto error;
fd2 = connect(ftpphost, ftppport);
if(fd2 < 0) goto error;
s = ftpcmd(fpw, fpr, type == 'd' ? "NLST" : "RETR", unesc(p));
if(s / 100 != 1) goto error;
while((s = read(fd2, buffer, sizeof(buffer))) > 0) {
s2 = write(1, buffer, s);
if(s2 != s) break;
}
if(s2 != s && s != 0) s = -1;
close(fd2);
s = ftpreply(fpr);
if(s / 100 == 2) s = 0;
error:
(void) ftpcmd(fpw, fpr, "QUIT", "");
fclose(fpr);
fclose(fpw);
close(fd);
return(s == 0 ? 0 : -1);
}
int tcpget(host, port, user, pass, path)
char *host;
int port;
char *user;
char *pass;
char *path;
{
int fd;
int s;
int s2;
if(port == 0) {
fprintf(stderr, "tcpget: No port specified\n");
return(-1);
}
fd = connect(host, port);
if(fd < 0) {
fprintf(stderr, "httpget: Could not connect to %s:%d\n", host, port);
return(-1);
}
if(*path == '\/')
path++;
write(fd, path, strlen(path));
write(fd, "\n", 1);
while((s = read(fd, buffer, sizeof(buffer))) > 0) {
s2 = write(1, buffer, s);
if(s2 != s) break;
}
close(fd);
return(0);
}
int main(argc, argv)
int argc;
char *argv[];
{
char *prog;
char *url;
char scheme;
char user[64];
char pass[64];
char host[64];
int port;
char *path;
int type;
char *ps;
char *p;
char *at;
int s, c;
int opt_h = 0;
int opt_d = 0;
int opt_p = 0;
prog = strrchr(*argv, '/');
if(prog == (char *)NULL)
prog = *argv;
argv++;
argc--;
while(argc && argv[0][0] == '-') {
char *opt = *argv++ + 1;
argc--;
if (opt[0] == '-' && opt[1] == 0) break;
while (*opt) switch (*opt++) {
case 'h': opt_h = -1; break;
case 'd': opt_d = -1; break;
case 'p': opt_p = -1; break;
default: argc = 0; break;
}
}
if(strcmp(prog, "ftpget") == 0) {
if(argc < 2 || argc > 4) {
fprintf(stderr, "Usage: %s host path [user [pass]]\n", prog);
return(-1);
}
strncpy(host, *argv++, sizeof(host));
port = 21;
path = *argv++;
if(argc) {
strncpy(user, *argv++, sizeof(user));
argc++;
} else
*user = '\0';
if(argc) {
strncpy(pass, *argv++, sizeof(pass));
argc++;
} else
*pass = '\0';
s = ftpget(host, port, user, pass, path, 'i');
return(s);
}
if(strcmp(prog, "httpget") == 0) {
if(argc != 2) {
fprintf(stderr, "Usage: %s [-h] [-d] [-p] host path\n", prog);
return(-1);
}
strncpy(host, *argv++, sizeof(host));
port = 80;
path = *argv++;
s = httpget(host, port, user, path, path, opt_h, opt_d, opt_p);
return(s);
}
if(argc != 1) {
usage:
fprintf(stderr, "Usage: %s [-h] [-p] url\n", prog);
return(-1);
}
url = *argv++;
argc--;
if(strncasecmp(url, "http://", 7) == 0) {
scheme = SCHEME_HTTP;
ps = url + 7;
} else
if(strncasecmp(url, "ftp://", 6) == 0) {
scheme = SCHEME_FTP;
ps = url + 6;
} else
if(strncasecmp(url, "tcp://", 6) == 0) {
scheme = SCHEME_TCP;
ps = url + 6;
} else {
fprintf(stderr, "%s: I do not handle this scheme\n", prog);
return(-1);
}
user[0] = '\0';
pass[0] = '\0';
host[0] = '\0';
port = 0;
p = ps;
while(*p && *p != '/') p++;
path = p;
c = *path;
*path = '\0';
at = strchr(ps, '@');
if(at != (char *)NULL) {
*at = '\0';
p = ps;
while(*p && *p != ':') p++;
if(*p)
*p++ = '\0';
strcpy(user, ps);
strcpy(pass, p);
ps = at + 1;
}
*path = c;
p = ps;
while(*p && *p != '/' && *p != ':') p++;
strncpy(host, ps, p - ps);
host[p - ps] = '\0';
if(*p == ':') {
p++;
ps = p;
while(*p && *p != '/')
port = port * 10 + (*p++ - '0');
}
if(*p == '/')
path = p;
else
path = "/";
if(scheme == SCHEME_FTP) {
p = path;
while(*p && *p != ';') p++;
if(*p) {
*p++ = '\0';
if(strncasecmp(p, "type=", 5) == 0) {
p += 5;
type = tolower(*p);
}
}
}
#if 0
fprintf(stderr, "Host: %s\n", host);
fprintf(stderr, "Port: %d\n", port);
fprintf(stderr, "User: %s\n", user);
fprintf(stderr, "Pass: %s\n", pass);
fprintf(stderr, "Path: %s\n", path);
fprintf(stderr, "Type: %c\n", type);
#endif
switch(scheme) {
case SCHEME_HTTP:
s = httpget(host, port, user, pass, path, opt_h, opt_d, opt_p);
break;
case SCHEME_FTP:
s = ftpget(host, port, user, pass, path, type);
break;
case SCHEME_TCP:
s = tcpget(host, port, user, pass, path);
break;
}
return(s);
}