95 lines
2.3 KiB
C
95 lines
2.3 KiB
C
/**
|
|
* This test connects to an access point and connects to a server on port 23.
|
|
* It waits for a server data and responds with another data.
|
|
*/
|
|
#include "espressif/esp_common.h"
|
|
#include "esp/uart.h"
|
|
|
|
#include <string.h>
|
|
|
|
#include "FreeRTOS.h"
|
|
#include "task.h"
|
|
|
|
#include "lwip/err.h"
|
|
#include "lwip/sockets.h"
|
|
#include "lwip/sys.h"
|
|
#include "lwip/netdb.h"
|
|
#include "lwip/dns.h"
|
|
|
|
#include "test/test.h"
|
|
|
|
#define AP_SSID "esp-open-rtos-ap"
|
|
#define AP_PSK "esp-open-rtos"
|
|
#define SERVER "172.16.0.1"
|
|
#define PORT 23
|
|
|
|
void connect_task(void *pvParameters)
|
|
{
|
|
printf("HTTP get task starting...\r\n");
|
|
|
|
while(1) {
|
|
struct sockaddr_in serv_addr;
|
|
uint8_t buf[256];
|
|
|
|
// wait for wifi connection
|
|
while (sdk_wifi_station_get_connect_status() != STATION_GOT_IP) {
|
|
vTaskDelay(100 / portTICK_RATE_MS);
|
|
}
|
|
|
|
int s = socket(AF_INET, SOCK_STREAM, 0);
|
|
TEST_ASSERT(s > 0, "failed to allocate socket.");
|
|
|
|
bzero(&serv_addr, sizeof(serv_addr));
|
|
serv_addr.sin_port = htons(PORT);
|
|
serv_addr.sin_family = AF_INET;
|
|
|
|
TEST_ASSERT(inet_aton(SERVER, &serv_addr.sin_addr.s_addr),
|
|
"failed to set IP address\n");
|
|
|
|
printf("allocated socket\n");
|
|
|
|
TEST_ASSERT(connect(s, (struct sockaddr*)&serv_addr, sizeof(serv_addr)),
|
|
"socket connect failed");
|
|
|
|
printf("connected\n");
|
|
|
|
bzero(buf, 256);
|
|
|
|
int r = 0;
|
|
for (int i = 0; i < 10; i++) {
|
|
r = read(s, buf, 256);
|
|
if (r > 0) {
|
|
break;
|
|
}
|
|
vTaskDelay(1000 / portTICK_RATE_MS);
|
|
}
|
|
TEST_CHECK(r > 0, "no data received");
|
|
|
|
TEST_CHECK(strcmp((const char*)buf, "test ping\r\n") == 0,
|
|
"received wrong data");
|
|
|
|
close(s);
|
|
|
|
TEST_FINISH();
|
|
break;
|
|
}
|
|
}
|
|
|
|
void user_init(void)
|
|
{
|
|
TEST_INIT(30);
|
|
|
|
printf("SDK version:%s\n", sdk_system_get_sdk_version());
|
|
|
|
struct sdk_station_config config = {
|
|
.ssid = AP_SSID,
|
|
.password = AP_PSK,
|
|
};
|
|
|
|
/* required to call wifi_set_opmode before station_set_config */
|
|
sdk_wifi_set_opmode(STATION_MODE);
|
|
sdk_wifi_station_set_config(&config);
|
|
|
|
xTaskCreate(&connect_task, (signed char *)"connect_task", 256, NULL, 2, NULL);
|
|
}
|
|
|