#include <config.h>
#include <stdio.h>
#include <stdlib.h>
#ifdef HAVE_STRING_H
# include <string.h>
#else
# include <strings.h>
#endif
#ifdef HAVE_UNISTD_H
# include <unistd.h>
#endif
#include <errno.h>
#include <assert.h>
#include <ctype.h>
#include <sys/types.h>
#include "wget.h"
#include "url.h"
#include "recur.h"
#include "utils.h"
#include "retr.h"
#include "ftp.h"
#include "fnmatch.h"
#include "host.h"
extern char *version_string;
#define ROBOTS_FILENAME "robots.txt"
static urlpos *urls_downloaded;
static slist *urls_html;
static slist *ulist;
static char **forbidden = NULL;
static int depth;
static char *base_dir;
static char *robots_host;
static int first_time = 1;
static struct urlinfo *robots_url PARAMS ((const char *, const char *));
static uerr_t retrieve_robots PARAMS ((const char *, const char *));
static char **parse_robots PARAMS ((const char *));
static int robots_match PARAMS ((struct urlinfo *, char **));
void
recursive_cleanup (void)
{
free_slist (ulist);
ulist = NULL;
free_vec (forbidden);
forbidden = NULL;
free_slist (urls_html);
urls_html = NULL;
free_urlpos (urls_downloaded);
urls_downloaded = NULL;
FREE_MAYBE (base_dir);
FREE_MAYBE (robots_host);
first_time = 1;
}
void
recursive_reset (void)
{
first_time = 1;
}
uerr_t
recursive_retrieve (const char *file, const char *this_url)
{
char *constr, *filename, *newloc;
char *canon_this_url = NULL;
int dt, inl;
int this_url_ftp;
uerr_t err;
struct urlinfo *rurl;
urlpos *url_list, *cur_url;
char *rfile;
struct urlinfo *u;
assert (this_url != NULL);
assert (file != NULL);
if (opt.quota && (opt.downloaded > opt.quota))
return QUOTEXC;
if (first_time)
{
ulist = add_slist (ulist, this_url, 0);
urls_downloaded = NULL;
urls_html = NULL;
u = newurl ();
err = parseurl (this_url, u, 0);
if (err == URLOK)
{
ulist = add_slist (ulist, u->url, 0);
urls_downloaded = add_url (urls_downloaded, u->url, file);
urls_html = add_slist (urls_html, file, NOSORT);
if (opt.no_parent)
base_dir = xstrdup (u->dir);
canon_this_url = xstrdup (u->url);
}
else
{
DEBUGP (("Double yuck! The *base* URL is broken.\n"));
base_dir = NULL;
}
freeurl (u, 1);
depth = 1;
robots_host = NULL;
forbidden = NULL;
first_time = 0;
}
else
++depth;
if ((opt.reclevel != 0) && (depth > opt.reclevel))
{
DEBUGP (("Recursion depth %d exceeded max. depth %d.\n",
depth, opt.reclevel));
--depth;
return RECLEVELEXC;
}
this_url_ftp = (urlproto (this_url) == URLFTP);
url_list = get_urls_html (file,
canon_this_url ? canon_this_url : this_url, 0);
for (cur_url = url_list; cur_url; cur_url = cur_url->next)
{
if (opt.quota && (opt.downloaded > opt.quota))
break;
u = newurl ();
if (parseurl (cur_url->url, u, 0) != URLOK)
{
DEBUGP (("Yuck! A bad URL.\n"));
freeurl (u, 1);
continue;
}
if (u->proto == URLFILE)
{
DEBUGP (("Nothing to do with file:// around here.\n"));
freeurl (u, 1);
continue;
}
assert (u->url != NULL);
constr = xstrdup (u->url);
inl = in_slist (ulist, constr);
if (!inl)
if (u->proto == URLFTP && !opt.follow_ftp && !this_url_ftp)
{
DEBUGP (("Uh, it is FTP but i'm not in the mood to follow FTP.\n"));
ulist = add_slist (ulist, constr, 0);
inl = 1;
}
if (!inl && u->proto != URLFTP)
if (opt.relative_only && !(cur_url->flags & URELATIVE))
{
DEBUGP (("It doesn't really look like a relative link.\n"));
ulist = add_slist (ulist, constr, 0);
inl = 1;
}
if (!inl)
if (!accept_domain (u))
{
DEBUGP (("I don't like the smell of that domain.\n"));
ulist = add_slist (ulist, constr, 0);
inl = 1;
}
if (!inl && opt.no_parent
&& !(!this_url_ftp && u->proto == URLFTP))
{
if (!(base_dir && frontcmp (base_dir, u->dir)))
{
struct urlinfo *ut = newurl ();
if (parseurl (this_url, ut, 0) != URLOK)
DEBUGP (("Double yuck! The *base* URL is broken.\n"));
else if (!frontcmp (ut->dir, u->dir))
{
DEBUGP (("Trying to escape parental guidance with no_parent on.\n"));
ulist = add_slist (ulist, constr, 0);
inl = 1;
}
freeurl (ut, 1);
}
}
if (!inl && (opt.includes || opt.excludes))
{
if (!accdir (u->dir, ALLABS))
{
DEBUGP (("%s (%s) is excluded/not-included.\n", constr, u->dir));
ulist = add_slist (ulist, constr, 0);
inl = 1;
}
}
if (!inl)
{
char *suf = NULL;
if (!
(!*u->file
|| (((suf = suffix (constr)) != NULL)
&& (!strcmp (suf, "html") || !strcmp (suf, "htm")))))
{
if (!acceptable (u->file))
{
DEBUGP (("%s (%s) does not match acc/rej rules.\n",
constr, u->file));
ulist = add_slist (ulist, constr, 0);
inl = 1;
}
}
FREE_MAYBE (suf);
}
if (!inl)
{
if (!opt.simple_check)
opt_url (u);
else
{
char *p;
for (p = u->host; *p; p++)
*p = tolower (*p);
free (u->url);
u->url = str_url (u, 0);
}
free (constr);
constr = xstrdup (u->url);
inl = in_slist (ulist, constr);
if (!inl && !((u->proto == URLFTP) && !this_url_ftp))
if (!opt.spanhost && this_url && !same_host (this_url, constr))
{
DEBUGP (("This is not the same hostname as the parent's.\n"));
ulist = add_slist (ulist, constr, 0);
inl = 1;
}
}
if (!inl && opt.use_robots && u->proto == URLHTTP)
{
if (!robots_host || !same_host (robots_host, u->host))
{
FREE_MAYBE (robots_host);
robots_host = xstrdup (u->host);
free_vec (forbidden);
forbidden = NULL;
err = retrieve_robots (constr, ROBOTS_FILENAME);
if (err == ROBOTSOK)
{
rurl = robots_url (constr, ROBOTS_FILENAME);
rfile = url_filename (rurl);
forbidden = parse_robots (rfile);
freeurl (rurl, 1);
free (rfile);
}
}
if (!robots_match (u, forbidden))
{
DEBUGP (("Stuffing %s because %s forbids it.\n", this_url,
ROBOTS_FILENAME));
ulist = add_slist (ulist, constr, 0);
inl = 1;
}
}
filename = NULL;
if (!inl)
{
DEBUGP (("I've decided to load it -> "));
ulist = add_slist (ulist, constr, 0);
if (u->proto == URLFTP)
{
opt.recursive = 0;
}
dt = 0;
retrieve_url (constr, &filename, &newloc,
canon_this_url ? canon_this_url : this_url, &dt);
if (u->proto == URLFTP)
{
opt.recursive = 1;
}
if (newloc)
{
free (constr);
constr = newloc;
}
if (opt.convert_links && filename)
{
if (dt & RETROKF)
{
urls_downloaded = add_url (urls_downloaded, constr, filename);
if (dt & TEXTHTML)
urls_html = add_slist (urls_html, filename, NOSORT);
}
}
if (dt & TEXTHTML)
{
if (dt & RETROKF)
recursive_retrieve (filename, constr);
}
else
DEBUGP (("%s is not text/html so we don't chase.\n",
filename ? filename: "(null)"));
if (opt.delete_after || (filename && !acceptable (filename)))
{
logprintf (LOG_VERBOSE,
(opt.delete_after ? _("Removing %s.\n")
: _("Removing %s since it should be rejected.\n")),
filename);
if (unlink (filename))
logprintf (LOG_NOTQUIET, "unlink: %s\n", strerror (errno));
dt &= ~RETROKF;
}
if (opt.convert_links && (dt & RETROKF) && (filename != NULL))
{
cur_url->flags |= UABS2REL;
cur_url->local_name = xstrdup (filename);
}
}
DEBUGP (("%s already in list, so we don't load.\n", constr));
FREE_MAYBE (filename);
FREE_MAYBE (constr);
freeurl (u, 1);
}
if (opt.convert_links)
convert_links (file, url_list);
free_urlpos (url_list);
FREE_MAYBE (canon_this_url);
--depth;
if (opt.quota && (opt.downloaded > opt.quota))
return QUOTEXC;
else
return RETROK;
}
void
convert_all_links (void)
{
uerr_t res;
urlpos *l1, *l2, *urls;
struct urlinfo *u;
slist *html;
urlpos *urlhtml;
for (html = urls_html; html; html = html->next)
{
DEBUGP (("Rescanning %s\n", html->string));
for (urlhtml = urls_downloaded; urlhtml; urlhtml = urlhtml->next)
if (!strcmp (urlhtml->local_name, html->string))
break;
if (urlhtml)
DEBUGP (("It should correspond to %s.\n", urlhtml->url));
else
DEBUGP (("I cannot find the corresponding URL.\n"));
urls = get_urls_html (html->string, urlhtml ? urlhtml->url : NULL, 1);
if (!urls)
continue;
for (l1 = urls; l1; l1 = l1->next)
{
u = newurl ();
res = parseurl (l1->url, u, 0);
if (res != URLOK)
{
freeurl (u, 1);
continue;
}
for (l2 = urls_downloaded; l2; l2 = l2->next)
if (!strcmp (l2->url, u->url))
{
DEBUGP (("%s flagged for conversion, local %s\n",
l2->url, l2->local_name));
break;
}
l1->flags &= ~ (UABS2REL | UREL2ABS);
if (l2)
{
l1->flags |= UABS2REL;
l1->local_name = xstrdup (l2->local_name);
}
else
{
l1->flags |= UREL2ABS;
l1->local_name = NULL;
}
freeurl (u, 1);
}
convert_links (html->string, urls);
free_urlpos (urls);
}
}
static struct urlinfo *
robots_url (const char *url, const char *robots_filename)
{
struct urlinfo *u = newurl ();
uerr_t err;
err = parseurl (url, u, 0);
assert (err == URLOK && u->proto == URLHTTP);
free (u->file);
free (u->dir);
free (u->url);
u->dir = xstrdup ("");
u->file = xstrdup (robots_filename);
u->url = str_url (u, 0);
return u;
}
static uerr_t
retrieve_robots (const char *url, const char *robots_filename)
{
int dt;
uerr_t err;
struct urlinfo *u;
u = robots_url (url, robots_filename);
logputs (LOG_VERBOSE, _("Loading robots.txt; please ignore errors.\n"));
err = retrieve_url (u->url, NULL, NULL, NULL, &dt);
freeurl (u, 1);
if (err == RETROK)
return ROBOTSOK;
else
return NOROBOTS;
}
static char **
parse_robots (const char *robots_filename)
{
FILE *fp;
char **entries;
char *line, *cmd, *str, *p;
char *base_version, *version;
int len, num, i;
int wget_matched;
entries = NULL;
num = 0;
fp = fopen (robots_filename, "rb");
if (!fp)
return NULL;
if (opt.useragent)
{
STRDUP_ALLOCA (base_version, opt.useragent);
STRDUP_ALLOCA (version, opt.useragent);
}
else
{
int len = 10 + strlen (version_string);
base_version = (char *)alloca (len);
sprintf (base_version, "Wget/%s", version_string);
version = (char *)alloca (len);
sprintf (version, "Wget/%s", version_string);
}
for (p = version; *p; p++)
*p = tolower (*p);
for (p = base_version; *p && *p != '/'; p++)
*p = tolower (*p);
*p = '\0';
wget_matched = 1;
while ((line = read_whole_line (fp)))
{
len = strlen (line);
if (len && line[len - 1] == '\r')
line[len - 1] = '\0';
DEBUGP (("Line: %s\n", line));
for (cmd = line; *cmd && ISSPACE (*cmd); cmd++);
if (!*cmd)
{
free (line);
DEBUGP (("(chucked out)\n"));
continue;
}
for (str = cmd; *str && *str != ':'; str++);
if (!*str)
{
free (line);
DEBUGP (("(chucked out)\n"));
continue;
}
*str++ = '\0';
for (; *str && ISSPACE (*str); str++);
for (p = str; *p; p++)
if (*p && ISSPACE (*p) && *(p + 1) == '#')
{
while (p > str && ISSPACE (*p))
--p;
if (p == str)
*p = '\0';
else
*(p + 1) = '\0';
break;
}
if (!strcasecmp (cmd, "User-agent"))
{
int match = 0;
for (p = str; *p; p++)
*p = tolower (*p);
if (*str == '*' && !*(str + 1))
match = 1;
else
{
if (has_wildcards_p (str))
{
if (strchr (str, '/'))
match = !fnmatch (str, version, 0);
else
match = !fnmatch (str, base_version, 0);
}
else
{
if (strstr (version, str))
match = 1;
else
match = 0;
}
}
wget_matched = match;
}
else if (!wget_matched)
{
free (line);
DEBUGP (("(chucking out since it is not applicable for Wget)\n"));
continue;
}
else if (!strcasecmp (cmd, "Disallow"))
{
if (!*str)
{
free_vec (entries);
entries = (char **)xmalloc (sizeof (char *));
*entries = NULL;
num = 0;
}
else
{
entries = (char **)xrealloc (entries, (num + 2)* sizeof (char *));
entries[num] = xstrdup (str);
entries[++num] = NULL;
for (i = strlen (str); i >= 0 && ISSPACE (str[i]); i--)
if (ISSPACE (str[i]))
str[i] = '\0';
}
}
else
{
DEBUGP (("(chucked out)\n"));
}
free (line);
}
fclose (fp);
return entries;
}
static int
robots_match (struct urlinfo *u, char **forbidden)
{
int l;
if (!forbidden)
return 1;
DEBUGP (("Matching %s against: ", u->path));
for (; *forbidden; forbidden++)
{
DEBUGP (("%s ", *forbidden));
l = strlen (*forbidden);
if (strncmp (u->path, *forbidden, l) == 0)
{
DEBUGP (("matched.\n"));
return 0;
}
}
DEBUGP (("not matched.\n"));
return 1;
}