mirror of
https://github.com/esp8266/Arduino.git
synced 2025-10-30 04:26:50 +03:00
61a8a6b14e
emulation with socket: fix nasty delay
properly check all return values (read() returning 0 indicates closed peer)
ClientContext::state() triggers read() to check for closed peer
uart emulation on tty: add cmdline -b debug option to disable raw tty
hide annoying MOCK messages, add cmdline -v option to show them
uart emulation on tty: check more return values
emulation on host: tcp/udp port shifting: new cmdline -s(port Shift) option
141 lines
3.6 KiB
C++
141 lines
3.6 KiB
C++
/*
|
|
Arduino emulation - WiFiServer socket side
|
|
Copyright (c) 2018 david gauchard. All rights reserved.
|
|
|
|
Permission is hereby granted, free of charge, to any person obtaining a
|
|
copy of this software and associated documentation files (the "Software"),
|
|
to deal with the Software without restriction, including without limitation
|
|
the rights to use, copy, modify, merge, publish, distribute, sublicense,
|
|
and/or sell copies of the Software, and to permit persons to whom the
|
|
Software is furnished to do so, subject to the following conditions:
|
|
|
|
- Redistributions of source code must retain the above copyright notice,
|
|
this list of conditions and the following disclaimers.
|
|
|
|
- Redistributions in binary form must reproduce the above copyright notice,
|
|
this list of conditions and the following disclaimers in the
|
|
documentation and/or other materials provided with the distribution.
|
|
|
|
- The names of its contributors may not be used to endorse or promote
|
|
products derived from this Software without specific prior written
|
|
permission.
|
|
|
|
THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
|
|
IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
|
|
FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL
|
|
THE CONTRIBUTORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR
|
|
OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE,
|
|
ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER
|
|
DEALINGS WITH THE SOFTWARE.
|
|
*/
|
|
|
|
#include <WiFiServer.h>
|
|
|
|
#include <arpa/inet.h>
|
|
#include <sys/types.h>
|
|
#include <sys/socket.h>
|
|
#include <poll.h>
|
|
#include <unistd.h>
|
|
#include <fcntl.h>
|
|
#include <errno.h>
|
|
|
|
#define int2pcb(x) ((tcp_pcb*)(intptr_t)(x))
|
|
#define pcb2int(x) ((int)(intptr_t)(x))
|
|
|
|
// host socket internal side of WiFiServer
|
|
|
|
int serverAccept (int srvsock)
|
|
{
|
|
int clisock;
|
|
socklen_t n;
|
|
struct sockaddr_in client;
|
|
n = sizeof(client);
|
|
if ((clisock = accept(srvsock, (struct sockaddr*)&client, &n)) == -1)
|
|
{
|
|
perror(MOCK "accept()");
|
|
exit(EXIT_FAILURE);
|
|
}
|
|
return mockSockSetup(clisock);
|
|
}
|
|
|
|
void WiFiServer::begin (uint16_t port)
|
|
{
|
|
_port = port;
|
|
return begin();
|
|
}
|
|
|
|
void WiFiServer::begin ()
|
|
{
|
|
int sock;
|
|
int mockport;
|
|
struct sockaddr_in server;
|
|
|
|
mockport = _port;
|
|
if (mockport < 1024 && mock_port_shifter)
|
|
{
|
|
mockport += mock_port_shifter;
|
|
fprintf(stderr, MOCK "=====> WiFiServer port: %d shifted to %d (use option -s) <=====\n", _port, mockport);
|
|
}
|
|
else
|
|
fprintf(stderr, MOCK "=====> WiFiServer port: %d <=====\n", mockport);
|
|
|
|
if ((sock = socket(AF_INET, SOCK_STREAM, 0)) == -1)
|
|
{
|
|
perror(MOCK "socket()");
|
|
exit(EXIT_FAILURE);
|
|
}
|
|
|
|
int optval = 1;
|
|
if (setsockopt(sock, SOL_SOCKET, SO_REUSEPORT, &optval, sizeof(optval)) == -1)
|
|
{
|
|
perror(MOCK "reuseport");
|
|
exit(EXIT_FAILURE);
|
|
}
|
|
|
|
server.sin_family = AF_INET;
|
|
server.sin_port = htons(mockport);
|
|
server.sin_addr.s_addr = htonl(INADDR_ANY);
|
|
if (bind(sock, (struct sockaddr*)&server, sizeof(server)) == -1)
|
|
{
|
|
perror(MOCK "bind()");
|
|
exit(EXIT_FAILURE);
|
|
}
|
|
|
|
if (listen(sock, 1) == -1)
|
|
{
|
|
perror(MOCK "listen()");
|
|
exit(EXIT_FAILURE);
|
|
}
|
|
|
|
|
|
// store int into pointer
|
|
_pcb = int2pcb(sock);
|
|
}
|
|
|
|
bool WiFiServer::hasClient ()
|
|
{
|
|
struct pollfd p;
|
|
p.fd = pcb2int(_pcb);
|
|
p.events = POLLIN;
|
|
return poll(&p, 1, 0) && p.revents == POLLIN;
|
|
}
|
|
|
|
size_t WiFiServer::write (uint8_t c)
|
|
{
|
|
return write(&c, 1);
|
|
}
|
|
|
|
size_t WiFiServer::write (const uint8_t *buf, size_t size)
|
|
{
|
|
fprintf(stderr, MOCK "todo: WiFiServer::write(%p, %zd)\n", buf, size);
|
|
exit(EXIT_FAILURE);
|
|
return 0;
|
|
}
|
|
|
|
void WiFiServer::close ()
|
|
{
|
|
if (pcb2int(_pcb) >= 0)
|
|
::close(pcb2int(_pcb));
|
|
_pcb = int2pcb(-1);
|
|
}
|