abc/lwip
1
0
Fork 0
mirror of https://git.savannah.nongnu.org/git/lwip.git synced 2026-05-30 22:43:54 +08:00
Code Issues Packages Projects Releases Wiki Activity

tftp: Add client functionality

Browse Source 
* add helper functions
* add tftp_get/tftp_put
* rename files
This commit is contained in:
Ben Wijen
2017-10-23 15:04:37 +02:00
committed by Dirk Ziegelmeier
parent c683427bfd
commit b2713601f0
5 changed files with 232 additions and 91 deletions
Download Patch File Download Diff File Expand all files Collapse all files

112
src/apps/tftp/tftp_server.c → src/apps/tftp/tftp_common.c
View File

@@ -1,6 +1,6 @@
/**
*
* @file tftp_server.c
* @file tftp_common.c
*
* @author Logan Gunthorpe <logang@deltatee.com>
* Dirk Ziegelmeier <dziegel@gmx.de>
@@ -41,13 +41,13 @@
*/
/**
* @defgroup tftp TFTP server
* @defgroup tftp TFTP client/server
* @ingroup apps
*
* This is simple TFTP server for the lwIP raw API.
* This is simple TFTP client/server for the lwIP raw API.
*/
#include "lwip/apps/tftp_server.h"
#include "lwip/apps/tftp_common.h"
#if LWIP_UDP
@@ -114,6 +114,40 @@ close_handle(void)
}
}
static struct pbuf*
init_packet(int opcode, int extra, int size)
{
struct pbuf* p = pbuf_alloc(PBUF_TRANSPORT, (u16_t)(TFTP_HEADER_LENGTH + size), PBUF_RAM);
u16_t* payload;
if(p != NULL) {
payload = (u16_t*) p->payload;
payload[0] = PP_HTONS(opcode);
payload[1] = lwip_htons(extra);
}
return p;
}
static void
send_request(const ip_addr_t *addr, u16_t port, int opcode, const char* fname, const char* mode) {
size_t fname_length = strlen(fname)+1;
size_t mode_length = strlen(mode)+1;
struct pbuf* p = init_packet(opcode, 0, -2 + fname_length + mode_length);
char* payload;
if(p == NULL) {
return;
}
payload = (char*) p->payload;
MEMCPY(payload+2, fname, fname_length);
MEMCPY(payload+2+fname_length, mode, mode_length);
udp_sendto(tftp_state.upcb, p, addr, port);
pbuf_free(p);
}
static void
send_error(const ip_addr_t *addr, u16_t port, enum tftp_error code, const char *str)
{
@@ -121,14 +155,12 @@ send_error(const ip_addr_t *addr, u16_t port, enum tftp_error code, const char *
struct pbuf *p;
u16_t *payload;
p = pbuf_alloc(PBUF_TRANSPORT, (u16_t)(TFTP_HEADER_LENGTH + str_length + 1), PBUF_RAM);
p = init_packet(TFTP_ERROR, code, str_length + 1);
if (p == NULL) {
return;
}
payload = (u16_t *) p->payload;
payload[0] = PP_HTONS(TFTP_ERROR);
payload[1] = lwip_htons(code);
MEMCPY(&payload[2], str, str_length + 1);
udp_sendto(tftp_state.upcb, p, addr, port);
@@ -136,25 +168,21 @@ send_error(const ip_addr_t *addr, u16_t port, enum tftp_error code, const char *
}
static void
send_ack(u16_t blknum)
send_ack(const ip_addr_t *addr, u16_t port, u16_t blknum)
{
struct pbuf *p;
u16_t *payload;
p = pbuf_alloc(PBUF_TRANSPORT, TFTP_HEADER_LENGTH, PBUF_RAM);
p = init_packet(TFTP_ACK, blknum, 0);
if (p == NULL) {
return;
}
payload = (u16_t *) p->payload;
payload[0] = PP_HTONS(TFTP_ACK);
payload[1] = lwip_htons(blknum);
udp_sendto(tftp_state.upcb, p, &tftp_state.addr, tftp_state.port);
udp_sendto(tftp_state.upcb, p, addr, port);
pbuf_free(p);
}
static void
resend_data(void)
resend_data(const ip_addr_t *addr, u16_t port)
{
struct pbuf *p = pbuf_alloc(PBUF_TRANSPORT, tftp_state.last_data->len, PBUF_RAM);
if (p == NULL) {
@@ -166,12 +194,12 @@ resend_data(void)
return;
}
udp_sendto(tftp_state.upcb, p, &tftp_state.addr, tftp_state.port);
udp_sendto(tftp_state.upcb, p, addr, port);
pbuf_free(p);
}
static void
send_data(void)
send_data(const ip_addr_t *addr, u16_t port)
{
u16_t *payload;
int ret;
@@ -180,24 +208,22 @@ send_data(void)
pbuf_free(tftp_state.last_data);
}
tftp_state.last_data = pbuf_alloc(PBUF_TRANSPORT, TFTP_HEADER_LENGTH + TFTP_MAX_PAYLOAD_SIZE, PBUF_RAM);
tftp_state.last_data = init_packet(TFTP_DATA, tftp_state.blknum, TFTP_MAX_PAYLOAD_SIZE);
if (tftp_state.last_data == NULL) {
return;
}
payload = (u16_t *) tftp_state.last_data->payload;
payload[0] = PP_HTONS(TFTP_DATA);
payload[1] = lwip_htons(tftp_state.blknum);
ret = tftp_state.ctx->read(tftp_state.handle, &payload[2], TFTP_MAX_PAYLOAD_SIZE);
if (ret < 0) {
send_error(&tftp_state.addr, tftp_state.port, TFTP_ERROR_ACCESS_VIOLATION, "Error occured while reading the file.");
send_error(addr, port, TFTP_ERROR_ACCESS_VIOLATION, "Error occured while reading the file.");
close_handle();
return;
}
pbuf_realloc(tftp_state.last_data, (u16_t)(TFTP_HEADER_LENGTH + ret));
resend_data();
resend_data(addr, port);
}
static void
@@ -270,10 +296,10 @@ recv(void *arg, struct udp_pcb *upcb, struct pbuf *p, const ip_addr_t *addr, u16
if (opcode == PP_HTONS(TFTP_WRQ)) {
tftp_state.mode_write = 1;
send_ack(0);
send_ack(addr, port, 0);
} else {
tftp_state.mode_write = 0;
send_data();
send_data(addr, port);
}
break;
@@ -302,7 +328,7 @@ recv(void *arg, struct udp_pcb *upcb, struct pbuf *p, const ip_addr_t *addr, u16
send_error(addr, port, TFTP_ERROR_ACCESS_VIOLATION, "error writing file");
close_handle();
} else {
send_ack(blknum);
send_ack(addr, port, blknum);
}
if (p->tot_len < TFTP_MAX_PAYLOAD_SIZE) {
@@ -312,7 +338,7 @@ recv(void *arg, struct udp_pcb *upcb, struct pbuf *p, const ip_addr_t *addr, u16
}
} else if ((u16_t)(blknum + 1) == tftp_state.blknum) {
/* retransmit of previous block, ack again (casting to u16_t to care for overflow) */
send_ack(blknum);
send_ack(addr, port, blknum);
} else {
send_error(addr, port, TFTP_ERROR_UNKNOWN_TRFR_ID, "Wrong block number");
}
@@ -347,14 +373,20 @@ recv(void *arg, struct udp_pcb *upcb, struct pbuf *p, const ip_addr_t *addr, u16
if (!lastpkt) {
tftp_state.blknum++;
send_data();
send_data(addr, port);
} else {
close_handle();
}
break;
}
case PP_HTONS(TFTP_ERROR):
if (tftp_state.handle != NULL) {
pbuf_remove_header(p, TFTP_HEADER_LENGTH);
tftp_state.ctx->error(tftp_state.handle, sbuf[1], p->payload, p->len);
close_handle();
}
break;
default:
send_error(addr, port, TFTP_ERROR_ILLEGAL_OPERATION, "Unknown operation");
break;
@@ -379,7 +411,7 @@ tftp_tmr(void *arg)
if ((tftp_state.timer - tftp_state.last_pkt) > (TFTP_TIMEOUT_MSECS / TFTP_TIMER_MSECS)) {
if ((tftp_state.last_data != NULL) && (tftp_state.retries < TFTP_MAX_RETRIES)) {
LWIP_DEBUGF(TFTP_DEBUG | LWIP_DBG_STATE, ("tftp: timeout, retrying\n"));
resend_data();
resend_data(&tftp_state.addr, tftp_state.port);
tftp_state.retries++;
} else {
LWIP_DEBUGF(TFTP_DEBUG | LWIP_DBG_STATE, ("tftp: timeout\n"));
@@ -389,7 +421,7 @@ tftp_tmr(void *arg)
}
/** @ingroup tftp
* Initialize TFTP server.
* Initialize TFTP client/server.
* @param ctx TFTP callback struct
*/
err_t
@@ -422,7 +454,7 @@ tftp_init(const struct tftp_context *ctx)
}
/** @ingroup tftp
* Deinitialize ("turn off") TFTP server.
* Deinitialize ("turn off") TFTP client/server.
*/
void tftp_cleanup(void)
{
@@ -432,4 +464,22 @@ void tftp_cleanup(void)
memset(&tftp_state, 0, sizeof(tftp_state));
}
err_t
tftp_get(void* handle, const ip_addr_t *addr, u16_t port, const char* fname, const char* mode) {
tftp_state.handle = handle;
tftp_state.blknum = 1;
tftp_state.mode_write = 1; // We want to receive data
send_request(addr, port, TFTP_RRQ, fname, mode);
return ERR_OK;
}
err_t
tftp_put(void* handle, const ip_addr_t *addr, u16_t port, const char* fname, const char* mode) {
tftp_state.handle = handle;
tftp_state.blknum = 1;
tftp_state.mode_write = 0; // We want to send data
send_request(addr, port, TFTP_WRQ, fname, mode);
return ERR_OK;
}
#endif /* LWIP_UDP */