mirror of https://github.com/ArduPilot/ardupilot
78 lines
2.1 KiB
C
78 lines
2.1 KiB
C
|
/*
|
||
|
|
* This file is free software: you can redistribute it and/or modify it
|
||
|
|
* under the terms of the GNU General Public License as published by the
|
||
|
|
* Free Software Foundation, either version 3 of the License, or
|
||
|
|
* (at your option) any later version.
|
||
|
|
*
|
||
|
|
* This file is distributed in the hope that it will be useful, but
|
||
|
|
* WITHOUT ANY WARRANTY; without even the implied warranty of
|
||
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.
|
||
|
|
* See the GNU General Public License for more details.
|
||
|
|
*
|
||
|
|
* You should have received a copy of the GNU General Public License along
|
||
|
|
* with this program. If not, see <http://www.gnu.org/licenses/>.
|
||
|
|
*/
|
||
|
|
|
||
|
|
#pragma once
|
||
|
|
|
||
|
|
#include <AP_HAL/UARTDriver.h>
|
||
|
|
#include <AP_HAL/utility/RingBuffer.h>
|
||
|
|
#include <AP_HAL_ESP32/AP_HAL_ESP32.h>
|
||
|
|
#include <AP_HAL_ESP32/Semaphores.h>
|
||
|
|
#include "lwip/sockets.h"
|
||
|
|
#include "esp_event.h"
|
||
|
|
|
||
|
|
|
||
|
|
class ESP32::WiFiUdpDriver : public AP_HAL::UARTDriver
|
||
|
|
{
|
||
|
|
public:
|
||
|
|
WiFiUdpDriver();
|
||
|
|
|
||
|
|
void begin(uint32_t b) override;
|
||
|
|
void begin(uint32_t b, uint16_t rxS, uint16_t txS) override;
|
||
|
|
void end() override;
|
||
|
|
void flush() override;
|
||
|
|
bool is_initialized() override;
|
||
|
|
void set_blocking_writes(bool blocking) override;
|
||
|
|
bool tx_pending() override;
|
||
|
|
|
||
|
|
uint32_t available() override;
|
||
|
|
uint32_t txspace() override;
|
||
|
|
int16_t read() override;
|
||
|
|
|
||
|
|
size_t write(uint8_t c) override;
|
||
|
|
size_t write(const uint8_t *buffer, size_t size) override;
|
||
|
|
|
||
|
|
uint32_t bw_in_kilobytes_per_second() const override
|
||
|
|
{
|
||
|
|
return 1000;
|
||
|
|
}
|
||
|
|
|
||
|
|
|
||
|
|
bool discard_input() override;
|
||
|
|
private:
|
||
|
|
enum ConnectionState {
|
||
|
|
NOT_INITIALIZED,
|
||
|
|
INITIALIZED,
|
||
|
|
CONNECTED
|
||
|
|
};
|
||
|
|
const size_t TX_BUF_SIZE = 1024;
|
||
|
|
const size_t RX_BUF_SIZE = 1024;
|
||
|
|
uint8_t _buffer[32];
|
||
|
|
ByteBuffer _readbuf{0};
|
||
|
|
ByteBuffer _writebuf{0};
|
||
|
|
Semaphore _write_mutex;
|
||
|
|
Semaphore _read_mutex;
|
||
|
|
ConnectionState _state;
|
||
|
|
|
||
|
|
int accept_socket;
|
||
|
|
|
||
|
|
void *_wifi_task_handle;
|
||
|
|
void initialize_wifi();
|
||
|
|
bool read_all();
|
||
|
|
bool write_data();
|
||
|
|
bool start_listen();
|
||
|
|
bool try_accept();
|
||
|
|
static void _wifi_thread(void* arg);
|
||
|
|
};
|