Skip to content

Commit

Permalink
WiFiS3 WiFiServer::accept()
Browse files Browse the repository at this point in the history
  • Loading branch information
JAndrassy committed Sep 22, 2023
1 parent b82e642 commit 3390295
Show file tree
Hide file tree
Showing 4 changed files with 123 additions and 1 deletion.
Original file line number Diff line number Diff line change
@@ -0,0 +1,104 @@
/*
Advanced WiFi Chat Server
A more advanced server that distributes any incoming messages
to all connected clients but the client the message comes from.
To use, telnet to your device's IP address and type.
You can see the client's input in the serial monitor as well.
*/

#include <WiFiS3.h>

#include "arduino_secrets.h"
///////please enter your sensitive data in the Secret tab/arduino_secrets.h
char ssid[] = SECRET_SSID; // your network SSID (name)
char pass[] = SECRET_PASS; // your network password (use for WPA, or use as key for WEP)

int status = WL_IDLE_STATUS;

// telnet defaults to port 23
WiFiServer server(23);

WiFiClient clients[8];

void setup() {
//Initialize serial and wait for port to open:
Serial.begin(9600);
while (!Serial) {
; // wait for serial port to connect. Needed for native USB port only
}

// check for the WiFi module:
if (WiFi.status() == WL_NO_MODULE) {
Serial.println("Communication with WiFi module failed!");
// don't continue
while (true);
}

String fv = WiFi.firmwareVersion();
if (fv < WIFI_FIRMWARE_LATEST_VERSION) {
Serial.println("Please upgrade the firmware");
}

// attempt to connect to WiFi network:
while (status != WL_CONNECTED) {
Serial.print("Attempting to connect to SSID: ");
Serial.println(ssid);
// Connect to WPA/WPA2 network. Change this line if using open or WEP network:
status = WiFi.begin(ssid, pass);

// wait 10 seconds for connection:
delay(10000);
}

// start the server:
server.begin();

Serial.print("Chat server address:");
Serial.println(WiFi.localIP());
}

void loop() {
// check for any new client connecting, and say hello (before any incoming data)
WiFiClient newClient = server.accept();
if (newClient) {
for (byte i=0; i < 8; i++) {
if (!clients[i]) {
Serial.print("We have a new client #");
Serial.println(i);
newClient.print("Hello, client number: ");
newClient.println(i);
// Once we "accept", the client is no longer tracked by WiFiServer
// so we must store it into our list of clients
clients[i] = newClient;
break;
}
}
}

// check for incoming data from all clients
for (byte i=0; i < 8; i++) {
if (clients[i] && clients[i].available() > 0) {
// read bytes from a client
byte buffer[80];
int count = clients[i].read(buffer, 80);
// write the bytes to all other connected clients
for (byte j=0; j < 8; j++) {
if (j != i && clients[j].connected()) {
clients[j].write(buffer, count);
}
}
}
}

// stop any clients which disconnect
for (byte i=0; i < 8; i++) {
if (clients[i] && !clients[i].connected()) {
Serial.print("disconnect client #");
Serial.println(i);
clients[i].stop();
}
}

}
Original file line number Diff line number Diff line change
@@ -0,0 +1,2 @@
#define SECRET_SSID ""
#define SECRET_PASS ""
17 changes: 16 additions & 1 deletion libraries/WiFiS3/src/WiFiServer.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -32,6 +32,21 @@ WiFiClient WiFiServer::available() {
return WiFiClient();
}

/* -------------------------------------------------------------------------- */
WiFiClient WiFiServer::accept() {
/* -------------------------------------------------------------------------- */
if(_sock != -1) {
string res = "";
modem.begin();
/* call the server accept on esp so that the accept is performed */
if(modem.write(string(PROMPT(_SERVERACCEPT)),res, "%s%d\r\n" , CMD_WRITE(_SERVERACCEPT), _sock)) {
int client_sock = atoi(res.c_str());
return WiFiClient(client_sock);
}
}
return WiFiClient();
}

/* -------------------------------------------------------------------------- */
void WiFiServer::begin(int port) {
/* -------------------------------------------------------------------------- */
Expand Down Expand Up @@ -87,4 +102,4 @@ void WiFiServer::end() {
bool WiFiServer::operator==(const WiFiServer& whs)
{
return _sock == whs._sock;
}
}
1 change: 1 addition & 0 deletions libraries/WiFiS3/src/WiFiServer.h
Original file line number Diff line number Diff line change
Expand Up @@ -37,6 +37,7 @@ class WiFiServer : public Server {
WiFiServer();
WiFiServer(int p);
WiFiClient available();
WiFiClient accept();
void begin(int port);
void begin();
virtual size_t write(uint8_t);
Expand Down

0 comments on commit 3390295

Please sign in to comment.