nes-proj/apps/shell/shell-wget.c
2010-06-09 08:36:43 +00:00

224 lines
6.1 KiB
C

/*
* Copyright (c) 2008, Swedish Institute of Computer Science.
* All rights reserved.
*
* Redistribution and use in source and binary forms, with or without
* modification, are permitted provided that the following conditions
* are met:
* 1. Redistributions of source code must retain the above copyright
* notice, this list of conditions and the following disclaimer.
* 2. Redistributions in binary form must reproduce the above copyright
* notice, this list of conditions and the following disclaimer in the
* documentation and/or other materials provided with the distribution.
* 3. Neither the name of the Institute nor the names of its contributors
* may be used to endorse or promote products derived from this software
* without specific prior written permission.
*
* THIS SOFTWARE IS PROVIDED BY THE INSTITUTE AND CONTRIBUTORS ``AS IS'' AND
* ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
* IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
* ARE DISCLAIMED. IN NO EVENT SHALL THE INSTITUTE OR CONTRIBUTORS BE LIABLE
* FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
* DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
* OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
* HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
* LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
* OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
* SUCH DAMAGE.
*
* This file is part of the Contiki operating system.
*
* $Id: shell-wget.c,v 1.3 2010/06/09 08:36:43 nifi Exp $
*/
/**
* \file
* Wget Contiki shell command
* \author
* Adam Dunkels <adam@sics.se>
*/
#include "contiki.h"
#include "shell.h"
#include "webclient.h"
#include <stdio.h>
#include <string.h>
#define MAX_URLLEN 128
static char url[MAX_URLLEN];
static uint8_t running;
/*---------------------------------------------------------------------------*/
PROCESS(shell_wget_process, "wget");
SHELL_COMMAND(wget_command,
"wget",
"wget [URL]: download a file with HTTP",
&shell_wget_process);
/*---------------------------------------------------------------------------*/
static void
open_url(char *url)
{
unsigned char i, c;
static char host[32];
char *file;
register char *urlptr;
static uip_ipaddr_t addr;
/* Trim off any spaces in the end of the url. */
urlptr = url + strlen(url) - 1;
while(*urlptr == ' ' && urlptr > url) {
*urlptr = 0;
--urlptr;
}
/* Don't even try to go further if the URL is empty. */
if(urlptr == url) {
return;
}
/* See if the URL starts with http://, otherwise prepend it. */
if(strncmp(url, http_http, 7) != 0) {
while(urlptr >= url) {
*(urlptr + 7) = *urlptr;
--urlptr;
}
strncpy(url, http_http, 7);
}
/* Find host part of the URL. */
urlptr = &url[7];
if(*urlptr == '[') {
/* IPv6 address */
c = ']';
++urlptr;
} else {
c = ':';
}
for(i = 0; i < sizeof(host); ++i) {
if(*urlptr == 0 ||
*urlptr == '/' ||
*urlptr == ' ' ||
*urlptr == c) {
host[i] = 0;
break;
}
host[i] = *urlptr;
++urlptr;
}
if(*urlptr == ']') {
/* Move past end of IPv6 host */
urlptr++;
}
/* XXX: Here we should find the port part of the URL, but this isn't
currently done because of laziness from the programmer's side
:-) */
/* Find file part of the URL. */
while(*urlptr != '/' && *urlptr != 0) {
++urlptr;
}
if(*urlptr == '/') {
file = urlptr;
} else {
file = "/";
}
/* Try to lookup the hostname. If it fails, we initiate a hostname
lookup and print out an informative message on the statusbar. */
if(uiplib_ipaddrconv(host, &addr) == 0) {
shell_output_str(&wget_command, "Not an IP address", "");
if(resolv_lookup(host) == NULL) {
shell_output_str(&wget_command, "Not resolved", "");
resolv_query(host);
return;
}
}
/* The hostname we present in the hostname table, so we send out the
initial GET request. */
shell_output_str(&wget_command, "Host ", host);
shell_output_str(&wget_command, "File ", file);
if(webclient_get(host, 80, file) == 0) {
shell_output_str(&wget_command, "Out of memory", "");
} else {
shell_output_str(&wget_command, "Connecting...", "");
}
}
/*---------------------------------------------------------------------------*/
PROCESS_THREAD(shell_wget_process, ev, data)
{
PROCESS_BEGIN();
strncpy(url, data, sizeof(url));
open_url(url);
running = 1;
while(running) {
PROCESS_WAIT_EVENT();
if(ev == tcpip_event) {
webclient_appcall(data);
} else if(ev == resolv_event_found) {
/* Either found a hostname, or not. */
if((char *)data != NULL &&
resolv_lookup((char *)data) != NULL) {
open_url(url);
} else {
shell_output_str(&wget_command, "Host not found.", "");
}
}
}
PROCESS_END();
}
/*---------------------------------------------------------------------------*/
void
shell_wget_init(void)
{
shell_register_command(&wget_command);
}
/*---------------------------------------------------------------------------*/
void
webclient_datahandler(char *data, u16_t len)
{
if(len == 0) {
running = 0;
} else {
shell_output(&wget_command, data, len, NULL, 0);
}
}
/*---------------------------------------------------------------------------*/
void
webclient_connected(void)
{
shell_output_str(&wget_command, "connected", "");
}
/*---------------------------------------------------------------------------*/
void
webclient_timedout(void)
{
shell_output_str(&wget_command, "timedout", "");
running = 0;
}
/*---------------------------------------------------------------------------*/
void
webclient_aborted(void)
{
shell_output_str(&wget_command, "aborted", "");
running = 0;
}
/*---------------------------------------------------------------------------*/
void
webclient_closed(void)
{
shell_output_str(&wget_command, "closed", "");
running = 0;
}
/*---------------------------------------------------------------------------*/