s938
/
webclient
6
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

add webclient package

Browse Source
master
1521761801@qq.com 8 years ago
parent 1ec75df4a9
commit ecc4d8a3fe
5 changed files with 1443 additions and 0 deletions
Show Stats Download Patch File Download Diff File

9
SConscript
Unescape Escape View File

@ -0,0 +1,9 @@
from building import *
cwd = GetCurrentDir()
src = Glob('*.c')
CPPPATH = [cwd]
group = DefineGroup('WebClient', src, depend = ['RT_USING_LWIP'], CPPPATH = CPPPATH)
Return('group')

1088
webclient.c
View File

File diff suppressed because it is too large Load Diff

110
webclient.h
Unescape Escape View File

@ -0,0 +1,110 @@
/*
* File : webclient.h
* COPYRIGHT (C) 2012-2013, Shanghai Real-Thread Technology Co., Ltd
*
* Change Logs:
* Date Author Notes
* 2013-05-05 Bernard the first version
* 2015-11-14 aozima add content_length_remainder.
*/
#ifndef __WEBCLIENT_H__
#define __WEBCLIENT_H__
#include <stddef.h>
#include <rtthread.h>
#define WEBCLIENT_HEADER_BUFSZ 4096
#define WEBCLIENT_RESPONSE_BUFSZ 4096
//typedef unsigned int size_t;
enum WEBCLIENT_STATUS
{
WEBCLIENT_OK,
WEBCLIENT_NOMEM,
WEBCLIENT_NOSOCKET,
WEBCLIENT_NOBUFFER,
WEBCLIENT_CONNECT_FAILED,
WEBCLIENT_DISCONNECT,
WEBCLIENT_FILE_ERROR,
WEBCLIENT_TIMEOUT,
};
enum WEBCLIENT_METHOD
{
WEBCLIENT_USER_METHOD,
WEBCLIENT_GET,
WEBCLIENT_POST,
};
struct webclient_session
{
/* the session socket */
int socket;
/* the response code of HTTP request */
int response;
/* transfer encoding */
char *transfer_encoding;
int chunk_sz;
int chunk_offset;
/* content_type of HTTP response */
char* content_type;
/* content_length of HTTP response */
int content_length;
/* last modified timestamp of resource */
char* last_modified;
/* location */
char* location;
/* server host */
char* host;
/* HTTP request */
char* request;
/* private for webclient session. */
/* position of reading */
unsigned int position;
/* remainder of content reading */
size_t content_length_remainder;
};
struct webclient_session* webclient_open(const char* URI);
struct webclient_session* webclient_open_position(const char* URI, int position);
struct webclient_session* webclient_open_header(const char* URI, int method, const char* header, size_t header_sz);
int webclient_close(struct webclient_session* session);
int webclient_set_timeout(struct webclient_session* session, int millisecond);
int webclient_read (struct webclient_session* session, unsigned char *buffer, size_t size);
int webclient_write(struct webclient_session* session, const unsigned char *buffer, size_t size);
int webclient_send_header(struct webclient_session* session, int method,
const char *header, size_t header_sz);
int webclient_connect(struct webclient_session* session, const char *URI);
int webclient_handle_response(struct webclient_session* session);
/* hight level APIs for HTTP client */
int webclient_response(struct webclient_session* session, void **response);
struct webclient_session* webclient_open_custom(const char* URI, int method,
const char* header, size_t header_sz,
const char* data, size_t data_sz);
int webclient_transfer(const char* URI, const char* header, size_t header_sz,
const char* data, size_t data_sz,
char *result, size_t result_sz);
#ifdef RT_USING_DFS
int webclient_get_file(const char* URI, const char* filename);
int webclient_post_file(const char* URI, const char* filename, const char* form_data);
#endif
#endif

216
webclient_file.c
Unescape Escape View File

@ -0,0 +1,216 @@
#include <stdint.h>
#include <stdlib.h>
#include <rtthread.h>
#include "webclient.h"
#include "webclient_internal.h"
#ifdef RT_USING_FINSH
#include <finsh.h>
#endif
#ifdef RT_USING_DFS
#include <dfs_posix.h>
int webclient_get_file(const char* URI, const char* filename)
{
int fd = -1;
size_t offset;
size_t length, total_length = 0;
rt_uint8_t* ptr = NULL;
struct webclient_session* session = NULL;
session = webclient_open(URI);
if (session == NULL)
{
rt_kprintf("open website failed.\n");
goto __exit;
}
if (session->response != 200)
{
rt_kprintf("wrong response: %d\n", session->response);
goto __exit;
}
fd = open(filename, O_WRONLY | O_CREAT | O_TRUNC, 0);
if (fd < 0)
{
rt_kprintf("open file failed\n");
goto __exit;
}
ptr = web_malloc(WEBCLIENT_RESPONSE_BUFSZ);
if (ptr == NULL)
{
rt_kprintf("out of memory\n");
goto __exit;
}
if (session->content_length == 0)
{
while (1)
{
length = webclient_read(session, ptr, WEBCLIENT_RESPONSE_BUFSZ);
if (length > 0)
{
write(fd, ptr, length);
total_length += length;
rt_kprintf(">");
}
else
break;
}
}
else
{
for (offset = 0; offset < session->content_length;)
{
length = webclient_read(session, ptr,
session->content_length - offset > WEBCLIENT_RESPONSE_BUFSZ ?
WEBCLIENT_RESPONSE_BUFSZ : session->content_length - offset);
if (length > 0)
{
write(fd, ptr, length);
total_length += length;
rt_kprintf(">");
}
else
break;
offset += length;
}
}
if (total_length)
{
rt_kprintf("\nSave %d bytes\n", total_length);
}
__exit:
if (fd >= 0) close(fd);
if (session != NULL) webclient_close(session);
if (ptr != NULL) web_free(ptr);
return 0;
}
int webclient_post_file(const char* URI, const char* filename,
const char* form_data)
{
size_t length;
char boundary[60];
int fd = -1, rc = WEBCLIENT_OK;
char *header = NULL, *header_ptr;
unsigned char *buffer = NULL, *buffer_ptr;
struct webclient_session* session = NULL;
fd = open(filename, O_RDONLY, 0);
if (fd < 0)
{
rc = -WEBCLIENT_FILE_ERROR;
goto __exit;
}
/* get the size of file */
length = lseek(fd, 0, SEEK_END);
lseek(fd, 0, SEEK_SET);
buffer = web_malloc(WEBCLIENT_RESPONSE_BUFSZ);
if (buffer == NULL)
{
rc = -WEBCLIENT_NOMEM;
goto __exit;
}
session = (struct webclient_session*) web_malloc(sizeof(struct webclient_session));
if (!session)
{
rc = -WEBCLIENT_NOMEM;
goto __exit;
}
memset(session, 0x0, sizeof(struct webclient_session));
rc = webclient_connect(session, URI);
if (rc < 0)
goto __exit;
header = (char*) web_malloc(WEBCLIENT_HEADER_BUFSZ);
if (header == NULL)
{
rc = -WEBCLIENT_NOMEM;
goto __exit;
}
header_ptr = header;
/* build boundary */
rt_snprintf(boundary, sizeof(boundary), "----------------------------%012d",
rt_tick_get());
/* build encapsulated mime_multipart information*/
buffer_ptr = buffer;
/* first boundary */
buffer_ptr += rt_snprintf((char*) buffer_ptr,
WEBCLIENT_RESPONSE_BUFSZ - (buffer_ptr - buffer), "--%s\r\n", boundary);
buffer_ptr += rt_snprintf((char*) buffer_ptr,
WEBCLIENT_RESPONSE_BUFSZ - (buffer_ptr - buffer),
"Content-Disposition: form-data; %s\r\n", form_data);
buffer_ptr += rt_snprintf((char*) buffer_ptr,
WEBCLIENT_RESPONSE_BUFSZ - (buffer_ptr - buffer),
"Content-Type: application/octet-stream\r\n\r\n");
/* calculate content-length */
length += buffer_ptr - buffer;
length += strlen(boundary) + 6; /* add the last boundary */
/* build header for upload */
header_ptr += rt_snprintf(header_ptr,
WEBCLIENT_HEADER_BUFSZ - (header_ptr - header),
"Content-Length: %d\r\n", length);
header_ptr += rt_snprintf(header_ptr,
WEBCLIENT_HEADER_BUFSZ - (header_ptr - header),
"Content-Type: multipart/form-data; boundary=%s\r\n", boundary);
/* send header */
rc = webclient_send_header(session, WEBCLIENT_POST, header,
header_ptr - header);
if (rc < 0)
goto __exit;
/* send mime_multipart */
webclient_write(session, buffer, buffer_ptr - buffer);
/* send file data */
while (1)
{
length = read(fd, buffer, WEBCLIENT_RESPONSE_BUFSZ);
if (length <= 0)
break;
webclient_write(session, buffer, length);
}
/* send last boundary */
rt_snprintf((char*) buffer, WEBCLIENT_RESPONSE_BUFSZ, "\r\n--%s--\r\n", boundary);
webclient_write(session, buffer, strlen(boundary) + 6);
__exit:
if (fd >= 0) close(fd);
if (session != NULL) webclient_close(session);
if (buffer != NULL) web_free(buffer);
if (header != NULL) web_free(header);
return 0;
}
int wget(int argc, char** argv)
{
if (argc != 3)
{
rt_kprintf("wget URI filename\n");
return 0;
}
webclient_get_file(argv[1], argv[2]);
return 0;
}
MSH_CMD_EXPORT(wget, web download file);
#endif

20
webclient_internal.h
Unescape Escape View File

@ -0,0 +1,20 @@
#ifndef WEBCLIENT_INTERNAL_H__
#define WEBCLIENT_INTERNAL_H__
#include <rtthread.h>
#ifdef RT_USING_ESP_PSRAM
#include <drv_sdram.h>
#define web_malloc sdram_malloc
#define web_free sdram_free
#define web_realloc sdram_realloc
#define web_calloc sdram_calloc
#else
#define web_malloc malloc
#define web_free free
#define web_realloc realloc
#define web_calloc calloc
#endif
#endif
Write Preview
Loading…
Cancel
Save