1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
|
/****************************************************************************************************
RepRapFirmware - Network: RepRapPro Ormerod with Duet controller
Separated out from Platform.h by dc42 and extended by chrishamm
****************************************************************************************************/
#ifndef NETWORK_H
#define NETWORK_H
#include "NetworkDefs.h"
#include "RepRapFirmware.h"
#include "MessageType.h"
#include "Socket.h"
#include "MessageFormats.h"
class NetworkResponder;
class HttpResponder;
class FtpResponder;
class TelnetResponder;
class WifiFirmwareUploader;
const unsigned int NumHttpResponders = 4; // the number of concurrent HTTP requests we can process
// Class to allow us to receive some data allowing for some extra bytes being stored by the DMAC
template<class T> class Receiver
{
public:
void *DmaPointer() { return &object; }
size_t Size() const { return sizeof(T); }
T& Value() { return object; }
private:
T object;
uint32_t padding;
};
// The main network class that drives the network.
class Network
{
public:
Network(Platform& p);
void Init();
void Activate();
void Exit();
void Spin(bool full);
void Diagnostics(MessageType mtype);
void Start();
void Stop();
void EnableProtocol(int protocol, int port, int secure, StringRef& reply);
void DisableProtocol(int protocol, StringRef& reply);
void ReportProtocols(StringRef& reply) const;
void Enable(int mode, const StringRef& ssid, StringRef& reply); // enable or disable the network
bool GetNetworkState(StringRef& reply);
int EnableState() const;
void SetHostname(const char *name);
bool FindResponder(Socket *skt, Port localPort);
const uint8_t *GetIPAddress() const { return ipAddress; }
void OpenDataPort(Port port);
void TerminateDataPort();
void HandleHttpGCodeReply(const char *msg);
void HandleTelnetGCodeReply(const char *msg);
void HandleHttpGCodeReply(OutputBuffer *buf);
void HandleTelnetGCodeReply(OutputBuffer *buf);
uint32_t GetHttpReplySeq();
// The remaining functions are specific to the WiFi version
WifiFirmwareUploader& GetWifiUploader() { return *uploader; }
void ResetWiFi();
static void ResetWiFiForUpload(bool external);
const char *GetWiFiServerVersion() const { return wiFiServerVersion; }
int32_t SendCommand(NetworkCommand cmd, SocketNumber socket, uint8_t flags, const void *dataOut, size_t dataOutLength, void* dataIn, size_t dataInLength);
template<class T> int32_t SendCommand(NetworkCommand cmd, SocketNumber socket, uint8_t flags, const void *dataOut, size_t dataOutLength, Receiver<T>& recvr)
{
return SendCommand(cmd, socket, flags, dataOut, dataOutLength, recvr.DmaPointer(), recvr.Size());
}
const char* TranslateNetworkState() const;
static const char* TranslateWiFiState(WiFiState w);
void SpiInterrupt();
void EspRequestsTransfer();
void UpdateSocketStatus(uint16_t connectedSockets, uint16_t otherEndClosedSockets);
private:
enum class NetworkState
{
disabled, // WiFi module disabled
starting1, // starting up
starting2, // starting up
active, // running, but not necessarily in the requested mode
changingMode, // running and in the process of switching between modes
};
void InitSockets();
void TerminateSockets();
void TerminateSockets(Port port);
void StartProtocol(Protocol protocol)
pre(protocol < NumProtocols);
void ShutdownProtocol(Protocol protocol)
pre(protocol < NumProtocols);
void ReportOneProtocol(Protocol protocol, StringRef& reply) const
pre(protocol < NumProtocols);
void SetIPAddress(const uint8_t ipAddress[], const uint8_t netmask[], const uint8_t gateway[]);
void SetupSpi();
void SendListenCommand(Port port, Protocol protocol, unsigned int maxConnections);
void GetNewStatus();
static const char* TranslateEspResetReason(uint32_t reason);
Platform& platform;
NetworkResponder *responders;
NetworkResponder *nextResponderToPoll;
FtpResponder *ftpResponder;
TelnetResponder *telnetResponder;
uint32_t longWait;
uint32_t lastTickMillis;
WifiFirmwareUploader *uploader;
Socket sockets[NumTcpSockets];
size_t currentSocket;
Port portNumbers[NumProtocols]; // port number used for each protocol
Port ftpDataPort;
bool protocolEnabled[NumProtocols]; // whether each protocol is enabled
NetworkState state;
WiFiState requestedMode;
WiFiState currentMode;
bool activated;
volatile bool espStatusChanged;
uint8_t ipAddress[4];
uint8_t netmask[4];
uint8_t gateway[4];
char hostname[16]; // Limit DHCP hostname to 15 characters + terminating 0
char requestedSsid[SsidLength + 1];
char actualSsid[SsidLength + 1];
unsigned int spiTxUnderruns;
unsigned int spiRxOverruns;
unsigned int reconnectCount;
char wiFiServerVersion[16];
};
#endif
|
