You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
145 lines
3.0 KiB
145 lines
3.0 KiB
/*
|
|
* LibUCW Utilities -- URL Handling Tool
|
|
*
|
|
* (c) 2004 Martin Mares <mj@ucw.cz>
|
|
*/
|
|
|
|
#include <ucw/lib.h>
|
|
#include <ucw/getopt.h>
|
|
#include <ucw/url.h>
|
|
#include <ucw/fastbuf.h>
|
|
|
|
#include <stdio.h>
|
|
#include <stdlib.h>
|
|
#include <string.h>
|
|
|
|
static byte *base_url;
|
|
static struct url base;
|
|
static uint opt_split = 0, opt_normalize = 0, opt_forgive = 0;
|
|
static struct fastbuf *fout;
|
|
static uint err_count;
|
|
|
|
static void
|
|
process_url(byte *url)
|
|
{
|
|
byte buf1[MAX_URL_SIZE], buf2[MAX_URL_SIZE], buf3[MAX_URL_SIZE], buf4[MAX_URL_SIZE];
|
|
int e;
|
|
struct url ur;
|
|
|
|
if ((e = url_deescape(url, buf1)) || (e = url_split(buf1, &ur, buf2)))
|
|
goto error;
|
|
if ((base_url || opt_normalize) && (e = url_normalize(&ur, &base)))
|
|
goto error;
|
|
if (opt_normalize && (e = url_canonicalize(&ur)))
|
|
goto error;
|
|
if (opt_split)
|
|
{
|
|
if (ur.protocol)
|
|
bprintf(fout, "protocol=%s\n", ur.protocol);
|
|
if (ur.user)
|
|
bprintf(fout, "user=%s\n", ur.user);
|
|
if (ur.pass)
|
|
bprintf(fout, "pass=%s\n", ur.pass);
|
|
if (ur.host)
|
|
bprintf(fout, "host=%s\n", ur.host);
|
|
if (ur.port != ~0U)
|
|
bprintf(fout, "port=%d\n", ur.port);
|
|
if (ur.rest)
|
|
bprintf(fout, "rest=%s\n", ur.rest);
|
|
bputc(fout, '\n');
|
|
}
|
|
else
|
|
{
|
|
if ((e = url_pack(&ur, buf3)) || (e = url_enescape(buf3, buf4)))
|
|
goto error;
|
|
bprintf(fout, "%s\n", buf4);
|
|
}
|
|
return;
|
|
|
|
error:
|
|
msg(L_ERROR, "%s: %s", url, url_error(e));
|
|
err_count++;
|
|
}
|
|
|
|
static char *shortopts = CF_SHORT_OPTS "b:fns";
|
|
static struct option longopts[] =
|
|
{
|
|
CF_LONG_OPTS
|
|
{ "base", 1, 0, 'b' },
|
|
{ "forgive", 0, 0, 'f' },
|
|
{ "normalize", 0, 0, 'n' },
|
|
{ "split", 0, 0, 's' },
|
|
{ NULL, 0, 0, 0 }
|
|
};
|
|
|
|
static char *help = "\
|
|
Usage: ucw-urltool [<options>] <operations> [<URL's>]\n\
|
|
\n\
|
|
Options:\n"
|
|
CF_USAGE "\
|
|
-b, --base <URL>\tInput URL's are relative to this base\n\
|
|
-f, --forgive\t\tReturn exit status 0 even if there were errors\n\
|
|
\n\
|
|
Operations:\n\
|
|
-s, --split\t\tSplit a given URL to components\n\
|
|
-n, --normalize\t\tNormalize given URL\n\
|
|
";
|
|
|
|
static void NONRET
|
|
usage(byte *msg)
|
|
{
|
|
if (msg)
|
|
{
|
|
fputs(msg, stderr);
|
|
fputc('\n', stderr);
|
|
}
|
|
fputs(help, stderr);
|
|
exit(1);
|
|
}
|
|
|
|
int
|
|
main(int argc, char **argv)
|
|
{
|
|
int opt, err;
|
|
byte *base_url = NULL;
|
|
byte basebuf1[MAX_URL_SIZE], basebuf2[MAX_URL_SIZE];
|
|
|
|
log_init(argv[0]);
|
|
while ((opt = cf_getopt(argc, argv, shortopts, longopts, NULL)) >= 0)
|
|
switch (opt)
|
|
{
|
|
case 'b':
|
|
base_url = optarg;
|
|
err = url_canon_split(base_url, basebuf1, basebuf2, &base);
|
|
if (err)
|
|
die("Invalid base URL: %s", url_error(err));
|
|
break;
|
|
case 's':
|
|
opt_split = 1;
|
|
break;
|
|
case 'n':
|
|
opt_normalize = 1;
|
|
break;
|
|
case 'f':
|
|
opt_forgive = 1;
|
|
break;
|
|
default:
|
|
usage("Invalid option");
|
|
}
|
|
|
|
fout = bfdopen_shared(1, 4096);
|
|
if (optind >= argc)
|
|
{
|
|
struct fastbuf *fin = bfdopen_shared(0, 4096);
|
|
byte url[MAX_URL_SIZE];
|
|
while (bgets(fin, url, sizeof(url)))
|
|
process_url(url);
|
|
bclose(fin);
|
|
}
|
|
else
|
|
while (optind < argc)
|
|
process_url(argv[optind++]);
|
|
bclose(fout);
|
|
|
|
return (err_count && !opt_forgive);
|
|
}
|
|
|