QubesOS/qubes-installer-qubes-os
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity
d0922a1cfa
qubes-installer-qubes-os/anaconda/loader/urls.c
Tomasz Sterna 25b30fa527 Moved Anaconda source to subdirectory
2011-03-31 13:40:54 +02:00

370 lines
12 KiB
C
Raw Blame History

/*
* urls.c - url handling code
*
* Copyright (C) 1997, 1998, 1999, 2000, 2001, 2002, 2009 Red Hat, Inc.
* All rights reserved.
*
* This program is free software; you can redistribute it and/or modify
* it under the terms of the GNU General Public License as published by
* the Free Software Foundation; either version 2 of the License, or
* (at your option) any later version.
*
* This program is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU General Public License for more details.
*
* You should have received a copy of the GNU General Public License
* along with this program. If not, see <http://www.gnu.org/licenses/>.
*
* Author(s): Erik Troan <ewt@redhat.com>
* Matt Wilson <msw@redhat.com>
* Michael Fulbright <msf@redhat.com>
* Jeremy Katz <katzj@redhat.com>
* Chris Lumens <clumens@redhat.com>
*/
#include <arpa/inet.h>
#include <ctype.h>
#include <fcntl.h>
#include <netinet/in.h>
#include <newt.h>
#include <regex.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <sys/types.h>
#include <unistd.h>
#include <netdb.h>
#include <errno.h>
#include <curl/curl.h>
#include "lang.h"
#include "loader.h"
#include "loadermisc.h"
#include "urls.h"
#include "log.h"
#include "windows.h"
#include "net.h"
#define NMATCH 10
/* boot flags */
extern uint64_t flags;
/* This is just a wrapper around the windows.c progress callback that accepts
* the arguments libcurl provides.
*/
int progress_cb(void *data, double dltotal, double dlnow, double ultotal, double ulnow) {
struct progressCBdata *cb_data = (struct progressCBdata *) data;
progressCallback(cb_data, dlnow, dltotal);
return 0;
}
int splitProxyParam(char *param, char **user, char **password, char **proxy) {
/* proxy=[protocol://][username[:password]@]host[:port] */
char *pattern = "([[:alpha:]]+://)?(([[:alnum:]]+)(:[^:@]+)?@)?([^:]+)(:[[:digit:]]+)?(/.*)?";
regex_t re;
regmatch_t pmatch[NMATCH];
if (regcomp(&re, pattern, REG_EXTENDED)) {
return 0;
}
if (regexec(&re, param, NMATCH, pmatch, 0) == REG_NOMATCH) {
regfree(&re);
return 0;
}
/* Match 0 is always the whole string (assuming regexec matched anything)
* so skip it. Then, these indices are just the number of the starting
* paren in pattern above. Make sure to change these whenever changing
* the pattern.
*/
if (pmatch[3].rm_so != -1)
*user = strndup(param+pmatch[3].rm_so, pmatch[3].rm_eo-pmatch[3].rm_so);
/* Skip the leading colon. */
if (pmatch[4].rm_so != -1)
*password = strndup(param+pmatch[4].rm_so+1, pmatch[4].rm_eo-pmatch[4].rm_so-1);
if (pmatch[5].rm_so != -1) {
char *portStr = "";
if (pmatch[6].rm_so != -1)
portStr = strndup(param+pmatch[6].rm_so, pmatch[6].rm_eo-pmatch[6].rm_so);
/* If no parameter was given, default to HTTP. yum will want to know
* the protocol, and curl will just ignore it if given.
*/
if (pmatch[1].rm_so != -1) {
checked_asprintf(proxy, "%.*s%.*s%s", pmatch[1].rm_eo-pmatch[1].rm_so,
param+pmatch[1].rm_so,
pmatch[5].rm_eo-pmatch[5].rm_so,
param+pmatch[5].rm_so,
portStr);
} else {
checked_asprintf(proxy, "http://%.*s%s", pmatch[5].rm_eo-pmatch[5].rm_so,
param+pmatch[5].rm_so,
portStr);
}
}
regfree(&re);
return 1;
}
int urlinstTransfer(struct loaderData_s *loaderData, struct iurlinfo *ui,
char **extraHeaders, char *dest) {
struct progressCBdata *cb_data;
CURL *curl = NULL;
CURLcode status;
struct curl_slist *headers = NULL;
char *version;
FILE *f = NULL;
logMessage(INFO, "transferring %s", ui->url);
f = fopen(dest, "w");
/* Initialize libcurl */
curl_global_init(CURL_GLOBAL_SSL);
curl = curl_easy_init();
checked_asprintf(&version, "anaconda/%s", VERSION);
curl_easy_setopt(curl, CURLOPT_USERAGENT, version);
curl_easy_setopt(curl, CURLOPT_URL, ui->url);
curl_easy_setopt(curl, CURLOPT_WRITEDATA, f);
curl_easy_setopt(curl, CURLOPT_FOLLOWLOCATION, 1);
curl_easy_setopt(curl, CURLOPT_MAXREDIRS, 10);
/* If a proxy was provided, add the options for that now. */
if (loaderData->proxy && strcmp(loaderData->proxy, "")) {
curl_easy_setopt(curl, CURLOPT_PROXY, loaderData->proxy);
if (loaderData->proxyUser && strcmp(loaderData->proxyUser, ""))
curl_easy_setopt(curl, CURLOPT_PROXYUSERNAME,
loaderData->proxyUser);
if (loaderData->proxyPassword && strcmp(loaderData->proxyPassword, ""))
curl_easy_setopt(curl, CURLOPT_PROXYPASSWORD,
loaderData->proxyPassword);
}
if (extraHeaders) {
int i;
for (i = 0; extraHeaders[i] != NULL; i++) {
headers = curl_slist_append(headers, extraHeaders[i]);
}
curl_easy_setopt(curl, CURLOPT_HTTPHEADER, headers);
}
/* Only set up the progress bar if we've got a UI to display it. */
if (FL_CMDLINE(flags)) {
printf("%s %s...\n", _("Retrieving"), ui->url);
} else {
char *filename;
filename = strrchr(ui->url, '/');
if (!filename)
filename = ui->url;
cb_data = winProgressBar(70, 5, _("Retrieving"), "%s %s...", _("Retrieving"), filename);
curl_easy_setopt(curl, CURLOPT_NOPROGRESS, 0);
curl_easy_setopt(curl, CURLOPT_PROGRESSFUNCTION, progress_cb);
curl_easy_setopt(curl, CURLOPT_PROGRESSDATA, cb_data);
}
/* Finally, do the transfer. */
status = curl_easy_perform(curl);
if (status)
logMessage(ERROR, "Error downloading %s: %s", ui->url, curl_easy_strerror(status));
if (!FL_CMDLINE(flags))
newtPopWindow();
if (headers)
curl_slist_free_all(headers);
fclose(f);
free(version);
curl_easy_cleanup(curl);
curl_global_cleanup();
return status;
}
char * addrToIp(char * hostname) {
struct in_addr ad;
struct in6_addr ad6;
char *ret;
struct hostent *host;
if ((ret = malloc(INET6_ADDRSTRLEN+1)) == NULL)
return hostname;
if (inet_ntop(AF_INET, &ad, ret, INET_ADDRSTRLEN) != NULL)
return ret;
else if (inet_ntop(AF_INET6, &ad6, ret, INET6_ADDRSTRLEN) != NULL)
return ret;
else if ((host = gethostbyname(hostname)) != NULL)
return host->h_name;
else
return NULL;
}
static void setProxySensitivity(newtComponent co, void *dptr) {
int i;
/* It's 3 because there are three entry boxes in the proxy grid. Lame. */
for (i = 0; i < 3; i++) {
newtEntrySetFlags(*((newtComponent *) dptr), NEWT_FLAG_DISABLED,
NEWT_FLAGS_TOGGLE);
dptr += sizeof(newtComponent);
}
return;
}
int urlMainSetupPanel(struct loaderData_s *loaderData, struct iurlinfo * ui) {
newtComponent form, okay, cancel, urlEntry, proxyCheckbox;
newtComponent proxyEntries[3];
newtComponent answer, text;
char enableProxy;
char *url = "", *proxy = "", *proxyUser = "", *proxyPassword = "";
char * reflowedText = NULL;
int width, height;
newtGrid buttons, grid, proxyGrid;
char * buf = NULL;
/* Populate the UI with whatever initial value we've got. */
if (ui && ui->url)
url = ui->url;
if (loaderData->proxy)
proxy = loaderData->proxy;
if (loaderData->proxyUser)
proxyUser = loaderData->proxyUser;
if (loaderData->proxyPassword)
proxyPassword = loaderData->proxyPassword;
buttons = newtButtonBar(_("OK"), &okay, _("Back"), &cancel, NULL);
checked_asprintf(&buf,
_("Please enter the URL containing the %s installation image on your server."),
getProductName());
reflowedText = newtReflowText(buf, 47, 5, 5, &width, &height);
free(buf);
text = newtTextbox(-1, -1, width, height, NEWT_TEXTBOX_WRAP);
newtTextboxSetText(text, reflowedText);
free(reflowedText);
urlEntry = newtEntry(22, 8, url, 60, (const char **) &url,
NEWT_ENTRY_SCROLL);
/* If we've been provided with proxy settings already, enable the proxy
* grid. This will make sure all the fields get filled in, too.
*/
enableProxy = loaderData->proxy != NULL && strcmp("", loaderData->proxy) ? '*' : ' ';
proxyCheckbox = newtCheckbox(-1, -1, _("Enable HTTP proxy"), enableProxy,
NULL, &enableProxy);
newtComponentAddCallback(proxyCheckbox, setProxySensitivity, &proxyEntries);
proxyEntries[0] = newtEntry(-1, -1, proxy, 35, (const char **) &proxy, NEWT_FLAG_SCROLL);
proxyEntries[1] = newtEntry(-1, -1, proxyUser, 15, (const char **) &proxyUser, NEWT_FLAG_SCROLL);
proxyEntries[2] = newtEntry(-1, -1, proxyPassword, 15, (const char **) &proxyPassword, NEWT_FLAG_SCROLL|NEWT_FLAG_PASSWORD);
/* Set the initial proxy grid sensitivity to match. */
if (enableProxy == ' ')
setProxySensitivity(proxyCheckbox, proxyEntries);
proxyGrid = newtCreateGrid(2, 3);
newtGridSetField(proxyGrid, 0, 0, NEWT_GRID_COMPONENT,
newtLabel(-1, -1, _("Proxy URL")),
0, 0, 0, 0, 0, NEWT_ANCHOR_LEFT);
newtGridSetField(proxyGrid, 1, 0, NEWT_GRID_COMPONENT, proxyEntries[0],
0, 0, 0, 0, 0, NEWT_ANCHOR_LEFT);
newtGridSetField(proxyGrid, 0, 1, NEWT_GRID_COMPONENT,
newtLabel(-1, -1, _("Username")),
0, 0, 0, 1, 0, NEWT_ANCHOR_LEFT);
newtGridSetField(proxyGrid, 1, 1, NEWT_GRID_COMPONENT, proxyEntries[1],
0, 0, 0, 1, 0, NEWT_ANCHOR_LEFT);
newtGridSetField(proxyGrid, 0, 2, NEWT_GRID_COMPONENT,
newtLabel(-1, -1, _("Password")),
0, 0, 0, 1, 0, NEWT_ANCHOR_LEFT);
newtGridSetField(proxyGrid, 1, 2, NEWT_GRID_COMPONENT, proxyEntries[2],
0, 0, 0, 1, 0, NEWT_ANCHOR_LEFT);
grid = newtCreateGrid(1, 5);
newtGridSetField(grid, 0, 0, NEWT_GRID_COMPONENT, text,
0, 0, 0, 1, 0, 0);
newtGridSetField(grid, 0, 1, NEWT_GRID_COMPONENT, urlEntry,
0, 0, 0, 1, 0, 0);
newtGridSetField(grid, 0, 2, NEWT_GRID_COMPONENT, proxyCheckbox,
0, 0, 0, 1, 0, NEWT_ANCHOR_LEFT);
newtGridSetField(grid, 0, 3, NEWT_GRID_SUBGRID, proxyGrid,
0, 0, 0, 0, 0, NEWT_GRID_FLAG_GROWX);
newtGridSetField(grid, 0, 4, NEWT_GRID_SUBGRID, buttons,
0, 0, 0, 0, 0, NEWT_GRID_FLAG_GROWX);
form = newtForm(NULL, NULL, 0);
newtGridAddComponentsToForm(grid, form, 1);
newtGridWrappedWindow(grid, _("URL Setup"));
newtGridFree(grid, 1);
do {
answer = newtRunForm(form);
if (answer != cancel) {
if (!strlen(url)) {
newtWinMessage(_("Error"), _("OK"),
_("You must enter a URL."));
continue;
}
if (strncmp(url, "http", 4) && strncmp(url, "ftp://", 6)) {
newtWinMessage(_("Error"), _("OK"),
_("URL must be either an ftp or http URL"));
continue;
}
ui->url = strdup(url);
if (enableProxy == '*') {
loaderData->proxy = strdup(proxy);
loaderData->proxyUser = strdup(proxyUser);
loaderData->proxyPassword = strdup(proxyPassword);
} else {
loaderData->proxy = "";
loaderData->proxyUser = "";
loaderData->proxyPassword = "";
}
/* FIXME: add back in hostname checking */
}
break;
} while (1);
if (answer == cancel) {
newtFormDestroy(form);
newtPopWindow();
return LOADER_BACK;
}
newtFormDestroy(form);
newtPopWindow();
return 0;
}
Reference in New Issue View Git Blame Copy Permalink