This repository has been archived on 2025-02-01. You can view files and clone it, but cannot push or open issues or pull requests.
reprapfirmware-dc42/Network.cpp
David Crocker 52b21d0eed Version 1.04c
Implemented rr_config web request for zpl's latest web interface
Implemented 7-factor auto calibration for delta printers
Fix bug with fileinfo calls corrupting memory
M563 can now be used to delete tools
2015-04-02 20:19:17 +01:00

1478 lines
35 KiB
C++

/****************************************************************************************************
RepRapFirmware - Network: RepRapPro Ormerod with Arduino Due controller
2014-04-05 Created from portions taken out of Platform.cpp by dc42
2014-04-07 Added portions of httpd.c. These portions are subject to the following copyright notice:
* Copyright (c) 2001-2003 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. The name of the author may not be used to endorse or promote products
* derived from this software without specific prior written permission.
*
* THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``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 AUTHOR 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 lwIP TCP/IP stack.
*
* Author: Adam Dunkels <adam@sics.se>
*
* (end httpd.c copyright notice)
****************************************************************************************************/
#include "RepRapFirmware.h"
#include "ethernet_sam.h"
#ifdef LWIP_STATS
#include "lwip/src/include/lwip/stats.h"
#endif
extern "C"
{
#include "lwipopts.h"
#include "lwip/src/include/lwip/tcp.h"
void RepRapNetworkSetMACAddress(const u8_t macAddress[]);
}
static tcp_pcb *http_pcb = NULL;
static tcp_pcb *ftp_main_pcb = NULL;
static tcp_pcb *ftp_pasv_pcb = NULL;
static tcp_pcb *telnet_pcb = NULL;
static bool closingDataPort = false;
static volatile bool lwipLocked = false;
static NetworkTransaction *sendingTransaction = NULL;
static char sendingWindow[TCP_WND];
static uint16_t sendingWindowSize, sentDataOutstanding;
static uint8_t sendingRetries;
static uint16_t httpPort = 80;
// Called only by LWIP to put out a message.
// May be called from C as well as C++
extern "C" void RepRapNetworkMessage(const char* s)
{
#ifdef LWIP_DEBUG
reprap.GetPlatform()->Message(DEBUG_MESSAGE, "%s", s);
#else
reprap.GetPlatform()->Message(HOST_MESSAGE, "%s", s);
#endif
}
/*-----------------------------------------------------------------------------------*/
extern "C"
{
// Lock functions for LWIP (LWIP generally isn't thread-safe)
bool LockLWIP()
{
if (lwipLocked)
return false;
lwipLocked = true;
return true;
}
void UnlockLWIP()
{
lwipLocked = false;
}
// Callback functions for the EMAC driver (called from ISR)
static void emac_read_packet(uint32_t ul_status)
{
// Because the LWIP stack can become corrupted if we work with it in parallel,
// we may have to wait for the next Spin() call to read the next packet.
if (LockLWIP())
{
do {
// read all queued packets from the RX buffer
} while (ethernet_read());
UnlockLWIP();
}
else
{
reprap.GetNetwork()->ReadPacket();
ethernet_set_rx_callback(NULL);
}
}
// Callback functions called by LWIP (may be called from ISR)
static void conn_err(void *arg, err_t err)
{
// Report the error to the monitor
reprap.GetPlatform()->Message(HOST_MESSAGE, "Network: Connection error, code %d\n", err);
ConnectionState *cs = (ConnectionState*)arg;
if (cs != NULL)
{
reprap.GetNetwork()->ConnectionClosed(cs, false); // tell the higher levels about the error
if (sendingTransaction == cs->sendingTransaction)
{
sendingTransaction = NULL;
sentDataOutstanding = 0;
}
}
}
/*-----------------------------------------------------------------------------------*/
static err_t conn_poll(void *arg, tcp_pcb *pcb)
{
ConnectionState *cs = (ConnectionState*)arg;
if (cs != NULL && sendingTransaction != NULL && cs == sendingTransaction->GetConnection())
{
// We tried to send data, but didn't receive an ACK within reasonable time.
sendingRetries++;
if (sendingRetries == 4)
{
reprap.GetPlatform()->Message(HOST_MESSAGE, "Network: Poll received error!\n");
tcp_abort(pcb);
return ERR_ABRT;
}
// Try to send the remaining data once again
err_t err = tcp_write(pcb, sendingWindow + (sendingWindowSize - sentDataOutstanding), sentDataOutstanding, 0);
if (err == ERR_OK)
{
tcp_output(pcb);
}
else
{
reprap.GetPlatform()->Message(HOST_MESSAGE, "Network: tcp_write in conn_poll failed with code %d\n", err);
tcp_abort(pcb);
return ERR_ABRT;
}
}
return ERR_OK;
}
/*-----------------------------------------------------------------------------------*/
static err_t conn_sent(void *arg, tcp_pcb *pcb, u16_t len)
{
LWIP_UNUSED_ARG(pcb);
ConnectionState *cs = (ConnectionState*)arg;
if (cs != NULL)
{
reprap.GetNetwork()->SentPacketAcknowledged(cs, len);
}
return ERR_OK;
}
/*-----------------------------------------------------------------------------------*/
static err_t conn_recv(void *arg, tcp_pcb *pcb, pbuf *p, err_t err)
{
ConnectionState *cs = (ConnectionState*)arg;
if (err == ERR_OK && cs != NULL)
{
if (cs->pcb != pcb)
{
reprap.GetPlatform()->Message(HOST_MESSAGE, "Network: Mismatched pcb!\n");
tcp_abort(pcb);
return ERR_ABRT;
}
if (p != NULL)
{
// Tell higher levels that we are receiving data
reprap.GetNetwork()->ReceiveInput(p, cs);
}
else if (cs->persistConnection)
{
// This is called when the connection has been gracefully closed, but LWIP doesn't close these
// connections automatically. That's why we must do it once all packets have been read.
reprap.GetNetwork()->ConnectionClosedGracefully(cs);
}
}
return ERR_OK;
}
/*-----------------------------------------------------------------------------------*/
static err_t conn_accept(void *arg, tcp_pcb *pcb, err_t err)
{
LWIP_UNUSED_ARG(arg);
LWIP_UNUSED_ARG(err);
tcp_setprio(pcb, TCP_PRIO_MIN);
//RepRapNetworkMessage("conn_accept\n");
/* Allocate a new ConnectionState for this connection */
ConnectionState *cs = reprap.GetNetwork()->ConnectionAccepted(pcb);
if (cs == NULL)
{
tcp_abort(pcb);
return ERR_ABRT;
}
/* Keep the listening PCBs running */
switch (pcb->local_port) // tell LWIP to accept further connections on the listening PCB
{
case ftpPort: // FTP
tcp_accepted(ftp_main_pcb);
break;
case telnetPort: // Telnet
tcp_accepted(telnet_pcb);
break;
default: // HTTP and FTP data
tcp_accepted((pcb->local_port == httpPort) ? http_pcb : ftp_pasv_pcb);
break;
}
tcp_arg(pcb, cs); // tell LWIP that this is the structure we wish to be passed for our callbacks
tcp_recv(pcb, conn_recv); // tell LWIP that we wish to be informed of incoming data by a call to the conn_recv() function
tcp_err(pcb, conn_err);
tcp_poll(pcb, conn_poll, 4);
return ERR_OK;
}
} // end extern "C"
/*-----------------------------------------------------------------------------------*/
// These functions (are) should be called only once at the start.
void httpd_init()
{
static int httpInitCount = 0;
httpInitCount++;
if (httpInitCount > 1)
{
reprap.GetPlatform()->Message(HOST_MESSAGE, "httpd_init() called more than once.\n");
}
tcp_pcb* pcb = tcp_new();
tcp_bind(pcb, IP_ADDR_ANY, httpPort);
http_pcb = tcp_listen(pcb);
tcp_accept(http_pcb, conn_accept);
}
void ftpd_init()
{
static int ftpInitCount = 0;
ftpInitCount++;
if (ftpInitCount > 1)
{
reprap.GetPlatform()->Message(HOST_MESSAGE, "ftpd_init() called more than once.\n");
}
tcp_pcb* pcb = tcp_new();
tcp_bind(pcb, IP_ADDR_ANY, ftpPort);
ftp_main_pcb = tcp_listen(pcb);
tcp_accept(ftp_main_pcb, conn_accept);
}
void telnetd_init()
{
static int telnetInitCount = 0;
telnetInitCount++;
if (telnetInitCount > 1)
{
reprap.GetPlatform()->Message(HOST_MESSAGE, "telnetd_init() called more than once.\n");
}
tcp_pcb* pcb = tcp_new();
tcp_bind(pcb, IP_ADDR_ANY, telnetPort);
telnet_pcb = tcp_listen(pcb);
tcp_accept(telnet_pcb, conn_accept);
}
//***************************************************************************************************
// Network/Ethernet class
Network::Network(Platform* p)
: platform(p), isEnabled(true), state(NetworkInactive), readingData(false),
freeTransactions(NULL), readyTransactions(NULL), writingTransactions(NULL),
dataCs(NULL), ftpCs(NULL), telnetCs(NULL), freeSendBuffers(NULL), freeConnections(NULL)
{
for (unsigned int i = 0; i < networkTransactionCount; i++)
{
freeTransactions = new NetworkTransaction(freeTransactions);
}
for (unsigned int i = 0; i < tcpOutputBufferCount; i++)
{
freeSendBuffers = new SendBuffer(freeSendBuffers);
}
for (unsigned int i = 0; i < numConnections; i++)
{
ConnectionState *cs = new ConnectionState;
cs->next = freeConnections;
freeConnections = cs;
}
ethPinsInit();
}
void Network::AppendTransaction(NetworkTransaction* volatile* list, NetworkTransaction *r)
{
r->next = NULL;
while (*list != NULL)
{
list = &((*list)->next);
}
*list = r;
}
void Network::PrependTransaction(NetworkTransaction* volatile* list, NetworkTransaction *r)
{
r->next = *list;
*list = r;
}
void Network::Init()
{
if (!isEnabled)
{
platform->Message(HOST_MESSAGE, "Attempting to start the network when it is disabled.\n");
return;
}
RepRapNetworkSetMACAddress(platform->MACAddress());
init_ethernet();
longWait = platform->Time();
state = NetworkInitializing;
}
void Network::Spin()
{
// Basically we can't do anything if we can't interact with LWIP
if (!LockLWIP())
{
platform->ClassReport(longWait);
return;
}
if (state == NetworkActive)
{
// See if we can read any packets
if (readingData)
{
readingData = false;
do {
// read all queued packets from the RX buffer
} while (ethernet_read());
ethernet_set_rx_callback(&emac_read_packet);
}
// See if we can send anything
NetworkTransaction *r = writingTransactions;
if (r != NULL && r->Send())
{
// We're done, free up this transaction
ConnectionState *cs = r->cs;
NetworkTransaction *rn = r->nextWrite;
writingTransactions = r->next;
AppendTransaction(&freeTransactions, r);
// If there is more data to write on this connection, do it next time
if (cs != NULL)
{
cs->sendingTransaction = rn;
}
if (rn != NULL)
{
PrependTransaction(&writingTransactions, rn);
}
}
}
else if (state == NetworkInitializing && establish_ethernet_link())
{
set_dhcp_hostname(reprap.GetName());
start_ethernet(platform->IPAddress(), platform->NetMask(), platform->GateWay());
httpd_init();
ftpd_init();
telnetd_init();
ethernet_set_rx_callback(&emac_read_packet);
state = NetworkActive;
}
UnlockLWIP();
platform->ClassReport(longWait);
}
void Network::Interrupt()
{
if (state != NetworkInactive && LockLWIP())
{
ethernet_timers_update();
UnlockLWIP();
}
}
void Network::Diagnostics()
{
platform->AppendMessage(BOTH_MESSAGE, "Network Diagnostics:\n");
uint8_t numFreeConnections = 0;
ConnectionState *freeConn = freeConnections;
while (freeConn != NULL)
{
numFreeConnections++;
freeConn = freeConn->next;
}
platform->AppendMessage(BOTH_MESSAGE, "Free connections: %d of %d\n", numFreeConnections, numConnections);
uint8_t numFreeTransactions = 0;
NetworkTransaction *freeTrans = freeTransactions;
while (freeTrans != NULL)
{
numFreeTransactions++;
freeTrans = freeTrans->next;
}
platform->AppendMessage(BOTH_MESSAGE, "Free transactions: %d of %d\n", numFreeTransactions, networkTransactionCount);
uint16_t numFreeSendBuffs = 0;
SendBuffer *freeSendBuff = freeSendBuffers;
while (freeSendBuff != NULL)
{
numFreeSendBuffs++;
freeSendBuff = freeSendBuff->next;
}
platform->AppendMessage(BOTH_MESSAGE, "Free send buffers: %d of %d\n", numFreeSendBuffs, tcpOutputBufferCount);
#if LWIP_STATS
// Normally we should NOT try to display LWIP stats here, because it uses debugPrintf(), which will hang the system is no USB cable is connected.
if (reprap.Debug(moduleNetwork))
{
stats_display();
}
#endif
}
void Network::Enable()
{
if (!isEnabled)
{
readingData = true;
// EMAC RX callback will be reset on next Spin calls
Init();
isEnabled = true;
}
}
void Network::Disable()
{
if (isEnabled)
{
readingData = false;
ethernet_set_rx_callback(NULL);
state = NetworkInactive;
isEnabled = false;
}
}
bool Network::IsEnabled() const
{
return isEnabled;
}
uint16_t Network::GetHttpPort() const
{
return httpPort;
}
void Network::SetHttpPort(uint16_t port)
{
if (state == NetworkActive && port != httpPort)
{
// Close old HTTP port
tcp_close(http_pcb);
// Create a new one for the new port
tcp_pcb* pcb = tcp_new();
tcp_bind(pcb, IP_ADDR_ANY, port);
http_pcb = tcp_listen(pcb);
tcp_accept(http_pcb, conn_accept);
}
httpPort = port;
}
bool Network::AllocateSendBuffer(SendBuffer *&buffer)
{
buffer = freeSendBuffers;
if (buffer == NULL)
{
platform->Message(HOST_MESSAGE, "Network: Could not allocate send buffer!\n");
return false;
}
freeSendBuffers = buffer->next;
buffer->bytesToWrite = 0;
buffer->next = NULL;
return true;
}
SendBuffer *Network::ReleaseSendBuffer(SendBuffer *buffer)
{
// If we used up all available send buffers, reset freeSendBuffers here
if (freeSendBuffers == NULL)
{
freeSendBuffers = buffer;
freeSendBuffers->next = NULL;
return NULL;
}
// Get the last item in the chain
SendBuffer *lastItem = freeSendBuffers;
while (lastItem->next != NULL)
{
lastItem = lastItem->next;
}
// Append the send buffer to be released
lastItem->next = buffer;
lastItem = buffer->next;
buffer->next = NULL;
// And return the next one of the send buffer being freed
return lastItem;
}
void Network::SentPacketAcknowledged(ConnectionState *cs, unsigned int len)
{
if (cs != NULL && sendingTransaction != NULL && cs == sendingTransaction->GetConnection())
{
if (sentDataOutstanding > len)
{
sentDataOutstanding -= len;
}
else
{
sendingTransaction = NULL;
sentDataOutstanding = 0;
}
}
// debugPrintf("Network SentPacketAcknowledged: invalid cs=%08x\n", (unsigned int)cs);
}
// This is called when a connection is being established and returns an initialised ConnectionState instance.
ConnectionState *Network::ConnectionAccepted(tcp_pcb *pcb)
{
ConnectionState *cs = freeConnections;
if (cs == NULL)
{
platform->Message(HOST_MESSAGE, "Network::ConnectionAccepted() - no free ConnectionStates!\n");
return NULL;
}
NetworkTransaction* r = freeTransactions;
if (r == NULL)
{
platform->Message(HOST_MESSAGE, "Network::ConnectionAccepted() - no free transactions!\n");
return NULL;
}
freeConnections = cs->next;
cs->Init(pcb);
r->Set(NULL, cs, connected);
freeTransactions = r->next;
AppendTransaction(&readyTransactions, r);
return cs;
}
// This is called when a connection is being closed or has gone down.
// It must set the state of any NetworkTransaction that refers to it to connection lost.
void Network::ConnectionClosed(ConnectionState* cs, bool closeConnection)
{
// make sure these connections are not reused
if (cs == dataCs)
{
dataCs = NULL;
}
if (cs == ftpCs)
{
ftpCs = NULL;
}
if (cs == telnetCs)
{
telnetCs = NULL;
}
// inform the Webserver that we are about to remove an existing connection
tcp_pcb *pcb = cs->pcb;
if (pcb != NULL)
{
reprap.GetWebserver()->ConnectionLost(cs);
if (closeConnection)
{
tcp_arg(pcb, NULL);
tcp_sent(pcb, NULL);
tcp_recv(pcb, NULL);
tcp_poll(pcb, NULL, 4);
tcp_close(pcb);
cs->pcb = NULL;
}
}
// cs points to a connection state block that the caller is about to release, so we need to stop referring to it.
// There may be one NetworkTransaction in the writing or closing list referring to it, and possibly more than one in the ready list.
for (NetworkTransaction* r = readyTransactions; r != NULL; r = r->next)
{
if (r->cs == cs)
{
r->SetConnectionLost();
}
}
if (cs->sendingTransaction != NULL)
{
cs->sendingTransaction->SetConnectionLost();
cs->sendingTransaction = NULL;
}
cs->next = freeConnections;
freeConnections = cs;
}
void Network::ConnectionClosedGracefully(ConnectionState *cs)
{
NetworkTransaction* r = freeTransactions;
if (r == NULL)
{
platform->Message(HOST_MESSAGE, "Network::ConnectionClosedGracefully() - no free transactions!\n");
return;
}
freeTransactions = r->next;
r->Set(NULL, cs, disconnected);
AppendTransaction(&readyTransactions, r);
}
bool Network::Lock()
{
return LockLWIP();
}
void Network::Unlock()
{
UnlockLWIP();
}
bool Network::InLwip() const
{
return lwipLocked;
}
void Network::ReadPacket()
{
readingData = true;
}
void Network::ReceiveInput(pbuf *pb, ConnectionState* cs)
{
NetworkTransaction* r = freeTransactions;
if (r == NULL)
{
platform->Message(HOST_MESSAGE, "Network::ReceiveInput() - no free transactions!\n");
return;
}
freeTransactions = r->next;
r->Set(pb, cs, dataReceiving);
AppendTransaction(&readyTransactions, r);
// debugPrintf("Network - input received\n");
}
// This is called by the web server to get a new received packet.
// If the connection parameter is NULL, we just return the request at the head of the ready list.
// Otherwise, we are only interested in packets received from the specified connection. If we find one than
// we move it to the head of the ready list, so that a subsequent call with a null connection parameter
// will return the same one.
NetworkTransaction *Network::GetTransaction(const ConnectionState *cs)
{
// See if there is any transaction at all
NetworkTransaction *rs = readyTransactions;
if (rs == NULL)
{
return NULL;
}
// If we're waiting for a new connection on a data port, see if there is a matching transaction available
if (cs == NULL && rs->waitingForDataConnection)
{
const uint16_t localPort = rs->GetLocalPort();
for (NetworkTransaction *rsNext = rs->next; rsNext != NULL; rsNext = rs->next)
{
if (rsNext->status == connected && rsNext->GetLocalPort() > 1023)
{
rs->next = rsNext->next; // remove rsNext from the list
rsNext->next = readyTransactions;
readyTransactions = rsNext;
return rsNext;
}
rs = rsNext;
}
return readyTransactions; // nothing found, process this transaction once again
}
// See if the first one is the transaction we're looking for
if (cs == NULL || rs->cs == cs)
{
return rs;
}
// There is at least one ready transaction, but it's not on the connection we are looking for
for (NetworkTransaction *rsNext = rs->next; rsNext != NULL; rsNext = rs->next)
{
if (rsNext->cs == cs)
{
rs->next = rsNext->next; // remove rsNext from the list
rsNext->next = readyTransactions;
readyTransactions = rsNext;
return rsNext;
}
rs = rsNext;
}
return NULL;
}
// Send the output data we already have, optionally with a file appended, then close the connection unless keepConnectionOpen is true.
// The file may be too large for our buffer, so we may have to send it in multiple transactions.
void Network::SendAndClose(FileStore *f, bool keepConnectionOpen)
{
NetworkTransaction *r = readyTransactions;
if (r == NULL)
{
return;
}
if (r->status == dataSending)
{
// This transaction is already in use for sending (e.g. a Telnet request),
// so all we have to do is to remove it from readyTransactions.
readyTransactions = r->next;
}
else
{
readyTransactions = r->next;
if (r->LostConnection())
{
if (f != NULL)
{
f->Close();
}
while (r->sendBuffer != NULL)
{
r->sendBuffer = ReleaseSendBuffer(r->sendBuffer);
}
AppendTransaction(&freeTransactions, r);
// debugPrintf("Conn lost before send\n");
}
else
{
r->FreePbuf();
r->cs->persistConnection = keepConnectionOpen;
r->fileBeingSent = f;
r->status = dataSending;
if (f != NULL && r->sendBuffer == NULL)
{
SendBuffer *buf;
if (AllocateSendBuffer(buf))
{
r->sendBuffer = buf;
r->fileBeingSent = f;
}
else
{
r->fileBeingSent = NULL;
// debugPrintf("Could not allocate send buffer for file transfer!\n");
}
}
NetworkTransaction *mySendingTransaction = r->cs->sendingTransaction;
if (mySendingTransaction == NULL)
{
r->cs->sendingTransaction = r;
AppendTransaction(&writingTransactions, r);
//debug
// r->outputBuffer[r->outputPointer] = 0;
// debugPrintf("Transaction queued for writing to network, file=%c, data=%s\n", (f ? 'Y' : 'N'), r->outputBuffer);
}
else
{
while (mySendingTransaction->nextWrite != NULL)
{
mySendingTransaction = mySendingTransaction->nextWrite;
}
mySendingTransaction->nextWrite = r;
// debugPrintf("Transaction appended to sending RS\n");
}
}
}
}
// We have no data to write and we want to keep the current connection alive if possible.
// That way we can speed up freeing the current NetworkTransaction.
void Network::CloseTransaction()
{
// Free the current NetworkTransaction's data (if any)
NetworkTransaction *r = readyTransactions;
if (r == NULL)
{
return;
}
r->FreePbuf();
// Terminate this connection if this NetworkTransaction indicates a graceful disconnect
TransactionStatus status = r->status;
if (!r->LostConnection() && status == disconnected)
{
// debugPrintf("Network: CloseRequest is closing connection cs=%08x\n", (unsigned int)locCs);
ConnectionClosed(r->cs, true);
}
// Remove the current item from readyTransactions
readyTransactions = r->next;
// Append it to freeTransactions again unless it's already on another list
if (status != dataSending)
{
AppendTransaction(&freeTransactions, r);
}
}
// The current NetworkTransaction must be processed again,
// e.g. because we're still waiting for another data connection.
void Network::WaitForDataConection()
{
NetworkTransaction *r = readyTransactions;
r->waitingForDataConnection = true;
r->inputPointer = 0; // behave as if this request hasn't been processed yet
}
void Network::OpenDataPort(uint16_t port)
{
closingDataPort = false;
tcp_pcb* pcb = tcp_new();
tcp_bind(pcb, IP_ADDR_ANY, port);
ftp_pasv_pcb = tcp_listen(pcb);
tcp_accept(ftp_pasv_pcb, conn_accept);
}
uint16_t Network::GetDataPort() const
{
return (closingDataPort || (ftp_pasv_pcb == NULL) ? 0 : ftp_pasv_pcb->local_port);
}
// Close FTP data port and purge associated PCB
void Network::CloseDataPort()
{
// See if it's already being closed
if (closingDataPort)
{
return;
}
closingDataPort = true;
// Close remote connection of our data port or do it as soon as the current transaction has finished
if (dataCs != NULL && dataCs->pcb != NULL)
{
NetworkTransaction *mySendingTransaction = dataCs->sendingTransaction;
if (mySendingTransaction != NULL)
{
mySendingTransaction->Close();
return;
}
}
// We can close it now, so do it here
if (ftp_pasv_pcb != NULL)
{
tcp_accept(ftp_pasv_pcb, NULL);
tcp_close(ftp_pasv_pcb);
ftp_pasv_pcb = NULL;
}
closingDataPort = false;
}
// These methods keep track of our connections in case we need to send to one of them
void Network::SaveDataConnection()
{
dataCs = readyTransactions->cs;
}
void Network::SaveFTPConnection()
{
ftpCs = readyTransactions->cs;
}
void Network::SaveTelnetConnection()
{
telnetCs = readyTransactions->cs;
}
// Check if there are enough resources left to allocate another NetworkTransaction for sending
bool Network::CanAcquireTransaction()
{
if (freeTransactions == NULL)
{
return false;
}
return (freeSendBuffers != NULL);
}
bool Network::AcquireFTPTransaction()
{
return AcquireTransaction(ftpCs);
}
bool Network::AcquireDataTransaction()
{
return AcquireTransaction(dataCs);
}
bool Network::AcquireTelnetTransaction()
{
return AcquireTransaction(telnetCs);
}
// Retrieves the NetworkTransaction of a sending connection to which data can be appended to,
// or prepares a released NetworkTransaction, which can easily be sent via SendAndClose.
bool Network::AcquireTransaction(ConnectionState *cs)
{
// Make sure we have a valid connection
if (cs == NULL)
{
return false;
}
// If our current transaction already belongs to cs and can be used, don't look for another one
NetworkTransaction *currentTransaction = readyTransactions;
if (currentTransaction != NULL && currentTransaction->GetConnection() == cs && currentTransaction->fileBeingSent == NULL)
{
return true;
}
// See if we're already writing on this connection
NetworkTransaction *lastTransaction = cs->sendingTransaction;
if (lastTransaction != NULL)
{
while (lastTransaction->nextWrite != NULL)
{
lastTransaction = lastTransaction->nextWrite;
}
}
// Then check if this transaction is valid and safe to use
NetworkTransaction *transactionToUse;
if (lastTransaction != NULL && sendingTransaction != lastTransaction && lastTransaction->fileBeingSent == NULL)
{
transactionToUse = lastTransaction;
}
// We cannot use it, so try to allocate a free one
else
{
transactionToUse = freeTransactions;
if (transactionToUse == NULL)
{
platform->Message(HOST_MESSAGE, "Network: Could not acquire free transaction!\n");
return false;
}
freeTransactions = transactionToUse->next;
transactionToUse->Set(NULL, cs, dataReceiving); // set it to dataReceiving as we expect a response
}
// Replace the first entry of readyTransactions with our new transaction, so it can be used by SendAndClose().
PrependTransaction(&readyTransactions, transactionToUse);
return true;
}
// Initialise a ConnectionState for a new connection
void ConnectionState::Init(tcp_pcb *p)
{
pcb = p;
next = NULL;
sendingTransaction = NULL;
persistConnection = true;
}
// Get local port from a ConnectionState
uint16_t ConnectionState::GetLocalPort() const
{
return pcb->local_port;
}
// Get remote IP from a ConnectionState
uint32_t ConnectionState::GetRemoteIP() const
{
return pcb->remote_ip.addr;
}
// Get remote port from a ConnectionState
uint16_t ConnectionState::GetRemotePort() const
{
return pcb->remote_port;
}
// NetworkTransaction class members
void NetworkTransaction::Set(pbuf *p, ConnectionState *c, TransactionStatus s)
{
cs = c;
pb = p;
bufferLength = (p == NULL) ? 0 : pb->tot_len;
status = s;
inputPointer = 0;
sendBuffer = NULL;
fileBeingSent = NULL;
closeRequested = false;
nextWrite = NULL;
lastWriteTime = NAN;
waitingForDataConnection = false;
}
// How many incoming bytes do we have to process?
uint16_t NetworkTransaction::DataLength() const
{
return (pb == NULL) ? 0 : pb->tot_len;
}
// Webserver calls this to read bytes that have come in from the network.
bool NetworkTransaction::Read(char& b)
{
if (LostConnection() || pb == NULL)
{
return false;
}
if (inputPointer == pb->len)
{
// See if there is another pbuf in the chain
if (inputPointer < pb->tot_len)
{
pb = pbuf_dechain(pb);
if (pb == NULL)
{
return false;
}
inputPointer = 0;
}
else
{
return false;
}
}
b = ((const char*)pb->payload)[inputPointer++];
return true;
}
// Read an entire pbuf from the NetworkTransaction
bool NetworkTransaction::ReadBuffer(char *&buffer, unsigned int &len)
{
if (LostConnection() || pb == NULL)
{
return false;
}
if (inputPointer == pb->len)
{
// See if there is another pbuf in the chain
if (inputPointer < pb->tot_len)
{
pb = pbuf_dechain(pb);
if (pb == NULL)
{
return false;
}
inputPointer = 0;
}
else
{
return false;
}
}
len = pb->len;
buffer = static_cast<char *>(pb->payload);
inputPointer += len;
return true;
}
// Webserver calls this to write bytes that need to go out to the network
void NetworkTransaction::Write(char b)
{
if (LostConnection() || status == disconnected) return;
if (sendBuffer == NULL)
{
if (reprap.GetNetwork()->AllocateSendBuffer(sendBuffer))
{
sendBuffer->tcpOutputBuffer[0] = b;
sendBuffer->bytesToWrite = 1;
}
else
{
// We cannot write because there are no more send buffers available.
}
}
else
{
// Get the last SendBuffer in the chain
SendBuffer *lastSendBuffer = sendBuffer;
while (lastSendBuffer->next != NULL)
{
lastSendBuffer = lastSendBuffer->next;
}
// Check if there's enough space left
if (tcpOutputBufferSize - lastSendBuffer->bytesToWrite > 0)
{
lastSendBuffer->tcpOutputBuffer[lastSendBuffer->bytesToWrite] = b;
lastSendBuffer->bytesToWrite++;
}
// No, try to allocate another one instead and append it to the list
else
{
SendBuffer *newSendBuffer;
if (reprap.GetNetwork()->AllocateSendBuffer(newSendBuffer))
{
lastSendBuffer->next = newSendBuffer;
newSendBuffer->tcpOutputBuffer[0] = b;
newSendBuffer->bytesToWrite = 1;
}
else
{
// We cannot write because there are no more send buffers available.
}
}
}
}
// These functions attempt to store a whole string for sending.
// It may be necessary to split it up into multiple SendBuffers.
void NetworkTransaction::Write(const char* s)
{
unsigned int len = strlen(s);
Write(s, len);
}
void NetworkTransaction::Write(StringRef ref)
{
Write(ref.Pointer(), ref.strlen());
}
void NetworkTransaction::Write(const char* s, size_t len)
{
if (LostConnection() || status == disconnected) return;
// Do we have a SendBuffer instance?
Network *net = reprap.GetNetwork();
if (sendBuffer == NULL)
{
if (!net->AllocateSendBuffer(sendBuffer))
{
// We cannot write because there are no more send buffers available.
return;
}
}
// Yes - retrieve the last send buffer from the chain.
SendBuffer *lastSendBuffer = sendBuffer;
while (lastSendBuffer->next != NULL)
{
lastSendBuffer = lastSendBuffer->next;
}
// Then try to store the whole string in multiple chunks if necessary.
size_t bytesStored = 0, bytesToStore = min<size_t>(len, tcpOutputBufferSize - lastSendBuffer->bytesToWrite);
do {
// Fill up current SendBuffer
memcpy(lastSendBuffer->tcpOutputBuffer + lastSendBuffer->bytesToWrite, s + bytesStored, bytesToStore);
lastSendBuffer->bytesToWrite += bytesToStore;
bytesStored += bytesToStore;
bytesToStore = min<size_t>(tcpOutputBufferSize, len - bytesStored);
// debugPrintf("%d of %d bytes stored, now storing %d bytes (%d bytes left in this chunk)\n", bytesStored, len, bytesToStore, tcpOutputBufferSize - currSendBuff->bytesToWrite);
// Allocate another SendBuffer if we cannot store the whole string in the current one
if (bytesToStore)
{
SendBuffer *newSendBuff;
if (net->AllocateSendBuffer(newSendBuff))
{
lastSendBuffer->next = newSendBuff;
lastSendBuffer = newSendBuff;
}
else
{
// We cannot write because there are no more send buffers available.
break;
}
}
} while (bytesToStore);
}
// Write formatted data to the output buffer
void NetworkTransaction::Printf(const char* fmt, ...)
{
if (LostConnection() || status == disconnected) return;
va_list p;
va_start(p, fmt);
char tempString[STRING_LENGTH];
int len = vsnprintf(tempString, STRING_LENGTH, fmt, p);
va_end(p);
Write(tempString, len);
}
// Send exactly one TCP window of data or return true if we can free up this object
bool NetworkTransaction::Send()
{
// Free up this transaction if we either lost our connection or are supposed to close it now
if (LostConnection() || closeRequested)
{
if (fileBeingSent != NULL)
{
fileBeingSent->Close();
fileBeingSent = NULL;
}
Network *net = reprap.GetNetwork();
while (sendBuffer != NULL)
{
sendBuffer = net->ReleaseSendBuffer(sendBuffer);
}
if (!LostConnection())
{
// debugPrintf("NetworkTransaction is closing connection cs=%08x\n", (unsigned int)cs);
reprap.GetNetwork()->ConnectionClosed(cs, true);
}
if (closingDataPort)
{
if (ftp_pasv_pcb != NULL)
{
tcp_accept(ftp_pasv_pcb, NULL);
tcp_close(ftp_pasv_pcb);
ftp_pasv_pcb = NULL;
}
closingDataPort = false;
}
sendingTransaction = NULL;
sentDataOutstanding = 0;
return true;
}
// We're still waiting for data to be ACK'ed, so check timeouts here
if (sentDataOutstanding)
{
if (!isnan(lastWriteTime))
{
float timeNow = reprap.GetPlatform()->Time();
if (timeNow - lastWriteTime > writeTimeout)
{
reprap.GetPlatform()->Message(HOST_MESSAGE, "Network: Timing out connection cs=%08x\n", (unsigned int)cs);
tcp_abort(cs->pcb);
cs->pcb = NULL;
}
return false;
}
}
else
{
sendingTransaction = NULL;
}
// See if we can fill up the TCP window with some data chunks from our SendBuffer instances
uint16_t bytesBeingSent = 0, bytesLeftToSend = TCP_WND;
while (sendBuffer != NULL && bytesLeftToSend >= sendBuffer->bytesToWrite)
{
memcpy(sendingWindow + bytesBeingSent, sendBuffer->tcpOutputBuffer, sendBuffer->bytesToWrite);
bytesBeingSent += sendBuffer->bytesToWrite;
bytesLeftToSend -= sendBuffer->bytesToWrite;
sendBuffer = reprap.GetNetwork()->ReleaseSendBuffer(sendBuffer);
}
// We also intend to send a file, so check if we can fill up the TCP window
if (sendBuffer == NULL)
{
int bytesRead;
size_t bytesToRead;
while (bytesLeftToSend && fileBeingSent != NULL)
{
bytesToRead = min<size_t>(256, bytesLeftToSend); // FIXME: doesn't work with higher block sizes
bytesRead = fileBeingSent->Read(sendingWindow + bytesBeingSent, bytesToRead);
if (bytesRead > 0)
{
bytesBeingSent += bytesRead;
bytesLeftToSend = TCP_WND - bytesBeingSent;
}
if (bytesRead != bytesToRead)
{
fileBeingSent->Close();
fileBeingSent = NULL;
}
}
}
if (!bytesBeingSent)
{
// If we have no data to send and fileBeingSent is NULL, we can close the connection
if (!cs->persistConnection && nextWrite == NULL)
{
Close();
return false;
}
// We want to send data from another transaction, so only free up this one
return true;
}
else
{
// The TCP window has been filled up as much as possible, so send it now. There is no need to check
// the available space in the SNDBUF queue, because we really write only one TCP window at once.
tcp_sent(cs->pcb, conn_sent);
err_t result = tcp_write(cs->pcb, sendingWindow, bytesBeingSent, 0);
if (result != ERR_OK) // Final arg - 1 means make a copy
{
reprap.GetPlatform()->Message(HOST_MESSAGE, "Network: tcp_write returned error code %d, this should never happen!\n", result);
tcp_abort(cs->pcb);
cs->pcb = NULL;
}
else
{
sendingTransaction = this;
sendingRetries = 0;
sendingWindowSize = sentDataOutstanding = bytesBeingSent;
lastWriteTime = reprap.GetPlatform()->Time();
tcp_output(cs->pcb);
}
}
return false;
}
void NetworkTransaction::SetConnectionLost()
{
cs = NULL;
FreePbuf();
for (NetworkTransaction *rs = nextWrite; rs != NULL; rs = rs->nextWrite)
{
rs->cs = NULL;
}
}
uint32_t NetworkTransaction::GetRemoteIP() const
{
return (cs != NULL) ? cs->pcb->remote_ip.addr : 0;
}
uint16_t NetworkTransaction::GetRemotePort() const
{
return (cs != NULL) ? cs->pcb->remote_port : 0;
}
uint16_t NetworkTransaction::GetLocalPort() const
{
return (cs != NULL) ? cs->pcb->local_port : 0;
}
void NetworkTransaction::Close()
{
tcp_pcb *pcb = cs->pcb;
tcp_poll(pcb, NULL, 4);
tcp_recv(pcb, NULL);
closeRequested = true;
}
void NetworkTransaction::FreePbuf()
{
// Tell LWIP that we have processed data
if (cs != NULL && bufferLength > 0 && cs->pcb != NULL)
{
tcp_recved(cs->pcb, bufferLength);
bufferLength = 0;
}
// Free pbuf (pbufs are thread-safe)
if (pb != NULL)
{
pbuf_free(pb);
pb = NULL;
}
}
// End