ref: afd1e85aa0dd7c71d7cd7f2cdaf18abd4e6513ae
dir: /main.c/
#include <u.h>
#include <libc.h>
#include <libsec.h>
#include <bio.h>
#include <ctype.h>
#include <plumb.h>
typedef struct Url Url;
typedef struct Response Response;
struct Url {
char *url;
char *server;
char *port;
};
struct Response {
Url *url;
char *mime;
char *prompt;
int status;
Biobuf body;
int fd;
};
#pragma varargck type "E" char*
Url *
parseurl(char *url)
{
char *server, *port, *s, *e;
Url *u;
url = strdup(url);
if((s = strpbrk(url, ":/")) != nil && s[0] == ':' && s[1] == '/' && s[2] == '/'){
server = s + 3;
}else{
s = smprint("gemini://%s", url);
free(url);
url = s;
server = s + 9;
}
port = strdup("1965");
if((e = strpbrk(server, ":/")) != nil){
s = mallocz(e-server+1, 1);
memmove(s, server, e-server);
server = s;
if(*e == ':'){
port = strdup(e+1);
if((e = strchr(port, '/')) != nil)
*e = 0;
}
}else{
server = strdup(server);
}
u = calloc(1, sizeof(*u));
u->url = url;
u->server = server;
u->port = port;
return u;
}
void
freeurl(Url *u)
{
if(u != nil){
free(u->url);
free(u->server);
free(u->port);
free(u);
}
}
void
freeresponse(Response *r)
{
if(r != nil){
close(r->fd);
freeurl(r->url);
free(r->mime);
free(r->prompt);
free(r);
}
}
Response *
request(char *url)
{
Thumbprint *th;
Response *r;
char *s;
TLSconn conn;
int ok, len, oldfd;
r = calloc(1, sizeof(*r));
r->fd = -1;
if((r->url = parseurl(url)) == nil)
goto err;
if((r->fd = dial(netmkaddr(r->url->server, "tcp", r->url->port), nil, nil, nil)) < 0)
goto err;
th = initThumbprints("/sys/lib/ssl/gemini", nil, "x509");
memset(&conn, 0, sizeof(conn));
conn.serverName = r->url->server;
oldfd = r->fd;
r->fd = tlsClient(oldfd, &conn);
close(oldfd);
if(r->fd < 0)
goto err;
/* FIXME find a way to trust on the first run */
if(th != nil){
ok = okCertificate(conn.cert, conn.certlen, th);
freeThumbprints(th);
if(!ok){
//fprint(2, "echo 'x509 %r server=%s' >>/sys/lib/ssl/gemini\n", r->url->server);
//werrstr("untrusted cert");
//goto err;
}
}
fprint(r->fd, "%s\r\n", r->url->url);
Binit(&r->body, r->fd, OREAD);
if((s = Brdstr(&r->body, '\n', 1)) == nil){
werrstr("EOF");
goto err;
}
if((len = Blinelen(&r->body)) > 0)
s[--len] = 0;
for(len--; len >= 0 && (s[len] == '\r' || s[len] == '\n'); len--)
s[len] = 0;
if(s[0] < '0' || s[0] > '9' || s[1] < '0' || s[1] > '9'){
werrstr("invalid status");
goto err;
}
r->status = 10*(int)(s[0]-'0') + s[1] - '0';
s += 2;
while(isspace(*s))
s++;
if(r->status >= 10 && r->status < 20){ /* input */
r->prompt = strdup(s);
}else if(r->status >= 20 && r->status < 30){ /* success */
r->mime = strdup(s[0] ? s : "text/gemini");
}else if(r->status >= 30 && r->status < 40){ /* redirect */
freeresponse(r);
r = request(s);
}else if(r->status >= 40 && r->status < 50){
werrstr("temporary failure: %s", s);
goto err;
}else if(r->status >= 50 && r->status < 60){
werrstr("permanent failure: %s", s);
goto err;
}else if(r->status >= 60 && r->status < 70){
werrstr("client cert required: %s", s);
goto err;
}
return r;
err:
if(r != nil && r->url != nil)
werrstr("%q: %r", r->url->url);
freeresponse(r);
return nil;
}
int
Efmt(Fmt *f)
{
char *s;
s = va_arg(f->args, char*);
for(; *s; s++){
if(*s == '%' && isxdigit(s[1]) && isxdigit(s[2])){
fmtprint(f, "%%%c%c", toupper(s[1]), toupper(s[2]));
s += 2;
}else if(isalnum(*s) || strchr(".-_~!$&'()*,;=/:@ \n", *s) == nil){
fmtprint(f, "%c", *s);
}else{
fmtprint(f, "%%%.2X", *s & 0xff);
}
}
return 0;
}
char *
readall(int fd)
{
char *s;
int n, sz, bufsz;
bufsz = 1023;
s = nil;
for(sz = 0;; sz += n){
if(bufsz-sz < 1024){
bufsz *= 2;
s = realloc(s, bufsz);
}
if((n = read(fd, s+sz, bufsz-sz-1)) < 1)
break;
}
s[sz] = 0;
if(sz > 1 && s[sz-1] == '\n')
s[sz-1] = 0;
return s;
}
void
main(int argc, char **argv)
{
Response *r;
char *s, *t, *u, *url, *prevurl;
int len, wait, pl, fd;
Plumbmsg *m;
Biobuf out;
wait = 0;
ARGBEGIN{
case 'w':
wait = 1;
break;
}ARGEND;
if(!wait && argc < 1){
fprint(2, "usage: gemnine [-w] [URL]\n");
exits("usage");
}
fmtinstall('E', Efmt);
quotefmtinstall();
Binit(&out, 1, OWRITE);
pl = -1;
prevurl = nil;
nexturl:
url = nil;
if(wait){
if(pl >= 0 || (pl = plumbopen("gemini", OREAD)) >= 0){
if((m = plumbrecv(pl)) != nil){
url = strdup(m->data);
plumbfree(m);
close(open("/dev/text", OWRITE|OTRUNC));
}else{
exits(nil);
}
}else{
sysfatal("plumbopen: %r");
}
}else{
url = strdup(argv[0]);
}
nextreq:
if((r = request(url)) != nil){
if(r->mime != nil && strncmp(r->mime, "text/", 5) != 0){
/* FIXME handle in a better way */
if(r->mime != nil)
fprint(2, "MIME %s\n", r->mime);
}else if(r->prompt != nil){
if((fd = open("/dev/consctl", OWRITE)) >= 0){
write(fd, "holdon", 6);
print("%s\n", r->prompt);
s = readall(0);
free(url);
url = smprint("%s?%E", r->url->url, s);
free(s);
freeresponse(r);
close(fd);
goto nextreq;
}else{
fprint(2, "INPUT %s\n", r->prompt);
}
}else{
while((s = Brdstr(&r->body, '\n', 1)) != nil){
if((len = Blinelen(&r->body)) > 0)
s[len] = 0;
if(s[0] == '=' && s[1] == '>'){
t = s + 2;
while(isspace(*t))
t++;
u = t;
if((t = strpbrk(t, " :/")) == nil || t[0] != ':' || t[1] != '/' || t[2] != '/'){ /* no scheme */
if(*u == '/'){ /* absolute */
Bprint(&out, "=> gemini://%s:%s/%s\n", r->url->server, r->url->port, u+1);
}else{
len = strlen(r->url->url);
Bprint(&out, "=> %s%s%s\n", r->url->url, r->url->url[len-1] != '/' ? "/" : "", u);
}
}else{
Bprint(&out, "%s\n", s);
}
}else{
Bprint(&out, "%s\n", s);
}
free(s);
}
}
freeresponse(r);
}else{
fprint(2, "%r\n");
if(!wait)
exits("failed");
}
if(prevurl != nil){
Bprint(&out, "\nPrevious: %s\n", prevurl);
free(prevurl);
}
Bflush(&out);
prevurl = url;
if(wait)
goto nexturl;
exits(nil);
}