321 lines
8.7 KiB
C
321 lines
8.7 KiB
C
|
/*
|
||
|
|
* Copyright (c) 2006-2022, RT-Thread Development Team
|
||
|
|
*
|
||
|
|
* SPDX-License-Identifier: Apache-2.0
|
||
|
|
*
|
||
|
|
* Change Logs:
|
||
|
|
* Date Author Notes
|
||
|
|
* 2022-05-26 xiangxistu first version
|
||
|
|
*/
|
||
|
|
|
||
|
|
/* Do NOT include sys/time.h in this file */
|
||
|
|
#include <WinSock2.h>
|
||
|
|
#include <WS2tcpip.h>
|
||
|
|
#include <rtthread.h>
|
||
|
|
#include <stdint.h>
|
||
|
|
|
||
|
|
#define DBG_TAG "sal.winsock"
|
||
|
|
#define DBG_LVL DBG_INFO
|
||
|
|
#include <rtdbg.h>
|
||
|
|
|
||
|
|
#include "rtt_winsock.h"
|
||
|
|
|
||
|
|
static struct sal_type_hostent sal_hostent;
|
||
|
|
static sal_type_ip_addr_t hostent_addr = { 0 };
|
||
|
|
static sal_type_ip_addr_t* hostent_addr_p = RT_NULL;
|
||
|
|
|
||
|
|
/* Important, must reserve */
|
||
|
|
#pragma comment( lib, "ws2_32.lib" )
|
||
|
|
|
||
|
|
/*
|
||
|
|
* reserve for select / poll
|
||
|
|
*
|
||
|
|
* #define WINDOWS_STACK_SIZE 2048
|
||
|
|
* #define WINDOWS_PRIORITY 2048
|
||
|
|
* #define WINDOWS_TICK 10
|
||
|
|
*
|
||
|
|
* struct poll_thread_list
|
||
|
|
* {
|
||
|
|
* int socket;
|
||
|
|
* rt_slist_t slist;
|
||
|
|
* };
|
||
|
|
* static rt_slist_t poll_pending_list = RT_SLIST_OBJECT_INIT(poll_pending_list);
|
||
|
|
*
|
||
|
|
**/
|
||
|
|
|
||
|
|
#define WINDOWS_PERVIOS_COMAMND \
|
||
|
|
do \
|
||
|
|
{ \
|
||
|
|
int iResult; \
|
||
|
|
WSADATA wsa_data; \
|
||
|
|
\
|
||
|
|
iResult = WSAStartup(MAKEWORD(2, 2), &wsa_data); \
|
||
|
|
if (iResult != 0) \
|
||
|
|
{ \
|
||
|
|
LOG_E("WSAStartup failed: %d\n", iResult); \
|
||
|
|
return -RT_ERROR; \
|
||
|
|
} \
|
||
|
|
}while(0)
|
||
|
|
|
||
|
|
static void data_sync_sockaddr(struct sal_type_sockaddr* sal, struct sockaddr** window)
|
||
|
|
{
|
||
|
|
/*
|
||
|
|
* ws2def.h
|
||
|
|
* typedef struct sockaddr {
|
||
|
|
*
|
||
|
|
* #if (_WIN32_WINNT < 0x0600)
|
||
|
|
* u_short sa_family;
|
||
|
|
* #else
|
||
|
|
* ADDRESS_FAMILY sa_family; // Address family.
|
||
|
|
* #endif //(_WIN32_WINNT < 0x0600)
|
||
|
|
*
|
||
|
|
* CHAR sa_data[14]; // Up to 14 bytes of direct address.
|
||
|
|
* } SOCKADDR, *PSOCKADDR, FAR *LPSOCKADDR;
|
||
|
|
*
|
||
|
|
***********************************
|
||
|
|
* sal_socket.h
|
||
|
|
* struct sockaddr
|
||
|
|
* {
|
||
|
|
* uint8_t sa_len;
|
||
|
|
* sa_family_t sa_family;
|
||
|
|
* char sa_data[14];
|
||
|
|
* };
|
||
|
|
*/
|
||
|
|
ADDRESS_FAMILY family = 0;
|
||
|
|
family = sal->sa_family;
|
||
|
|
|
||
|
|
*window = sal;
|
||
|
|
(* window)->sa_family = family;
|
||
|
|
}
|
||
|
|
|
||
|
|
int win_socket(int domain, int type, int protocol)
|
||
|
|
{
|
||
|
|
WINDOWS_PERVIOS_COMAMND;
|
||
|
|
|
||
|
|
return (int)socket(domain, type, protocol);
|
||
|
|
}
|
||
|
|
|
||
|
|
int win_closesocket(int s)
|
||
|
|
{
|
||
|
|
return (int)closesocket(s);
|
||
|
|
}
|
||
|
|
|
||
|
|
int win_bind(int s, const struct sal_type_sockaddr *name, sal_type_socklen_t namelen)
|
||
|
|
{
|
||
|
|
return (int)bind(s, name, namelen);
|
||
|
|
}
|
||
|
|
|
||
|
|
int win_listen(int s, int backlog)
|
||
|
|
{
|
||
|
|
return (int)listen(s, backlog);
|
||
|
|
}
|
||
|
|
|
||
|
|
int win_connect(int s, const struct sal_type_sockaddr *name, sal_type_socklen_t namelen)
|
||
|
|
{
|
||
|
|
struct sockaddr *windows_addr = RT_NULL;
|
||
|
|
data_sync_sockaddr(name, &windows_addr);
|
||
|
|
|
||
|
|
return (int)connect(s, windows_addr, namelen);
|
||
|
|
}
|
||
|
|
|
||
|
|
int win_accept(int s, struct sal_type_sockaddr *addr, sal_type_socklen_t *addrlen)
|
||
|
|
{
|
||
|
|
WINDOWS_PERVIOS_COMAMND;
|
||
|
|
|
||
|
|
return (int)accept(s, addr, addrlen);
|
||
|
|
}
|
||
|
|
|
||
|
|
int win_sendto(int s, const void *data, size_t size, int flags, const struct sal_type_sockaddr *to, sal_type_socklen_t tolen)
|
||
|
|
{
|
||
|
|
return (int)sendto(s, data, size, flags, to, tolen);
|
||
|
|
}
|
||
|
|
|
||
|
|
int win_recvfrom(int s, void *mem, size_t len, int flags, struct sal_type_sockaddr *from, sal_type_socklen_t *fromlen)
|
||
|
|
{
|
||
|
|
return (int)recvfrom(s, mem, len, flags, from, fromlen);
|
||
|
|
}
|
||
|
|
|
||
|
|
int win_getsockopt(int s, int level, int optname, void *optval, sal_type_socklen_t *optlen)
|
||
|
|
{
|
||
|
|
return (int)getsockopt(s, level, optname, optval, optlen);
|
||
|
|
}
|
||
|
|
|
||
|
|
int win_setsockopt(int s, int level, int optname, const void *optval, sal_type_socklen_t optlen)
|
||
|
|
{
|
||
|
|
return (int)setsockopt(s, level, optname, optval, optlen);
|
||
|
|
}
|
||
|
|
|
||
|
|
int win_shutdown(int s, int how)
|
||
|
|
{
|
||
|
|
return (int)shutdown(s, how);
|
||
|
|
}
|
||
|
|
|
||
|
|
int win_getpeername(int s, struct sal_type_sockaddr *name, sal_type_socklen_t *namelen)
|
||
|
|
{
|
||
|
|
return (int)getpeername(s, name, namelen);
|
||
|
|
}
|
||
|
|
|
||
|
|
int win_getsockname(int s, struct sal_type_sockaddr *name, sal_type_socklen_t *namelen)
|
||
|
|
{
|
||
|
|
return (int)getsockname(s, name, namelen);
|
||
|
|
}
|
||
|
|
|
||
|
|
int win_ioctlsocket(int s, long cmd, void *arg)
|
||
|
|
{
|
||
|
|
return (int)ioctlsocket(s, cmd, arg);
|
||
|
|
}
|
||
|
|
|
||
|
|
#ifdef SAL_USING_POSIX
|
||
|
|
/* use windows poll, but not wait */
|
||
|
|
int inet_poll(struct dfs_fd* file, struct rt_pollreq* req)
|
||
|
|
{
|
||
|
|
/*WSAPOLLFD winpoll;
|
||
|
|
struct sal_socket* sal_sock;
|
||
|
|
int mask = 0;
|
||
|
|
int poll_event = 0;
|
||
|
|
|
||
|
|
sal_sock = sal_get_socket((int)file->data);
|
||
|
|
if (!sal_sock)
|
||
|
|
{
|
||
|
|
return -1;
|
||
|
|
}
|
||
|
|
|
||
|
|
winpoll.fd = sal_sock->socket;
|
||
|
|
poll_event &= POLLERR | POLLHUP;
|
||
|
|
winpoll.events = req->_key & (~poll_event);
|
||
|
|
poll_event = 0;
|
||
|
|
|
||
|
|
mask = WSAPoll(&winpoll, 1, 0);
|
||
|
|
if (mask == 0)
|
||
|
|
return 0;
|
||
|
|
|
||
|
|
return winpoll.revents;*/
|
||
|
|
}
|
||
|
|
#endif /* SAL_USING_POSIX */
|
||
|
|
|
||
|
|
struct sal_type_hostent* win_gethostbyname(const char* name)
|
||
|
|
{
|
||
|
|
struct hostent *windows_hostent;
|
||
|
|
char** h_addr_list;
|
||
|
|
|
||
|
|
WINDOWS_PERVIOS_COMAMND;
|
||
|
|
|
||
|
|
windows_hostent = gethostbyname(name);
|
||
|
|
|
||
|
|
sal_hostent.h_name = windows_hostent->h_name;
|
||
|
|
sal_hostent.h_aliases = windows_hostent->h_aliases;
|
||
|
|
sal_hostent.h_addrtype = windows_hostent->h_addrtype;
|
||
|
|
sal_hostent.h_length = windows_hostent->h_length;
|
||
|
|
sal_hostent.h_addr_list = windows_hostent->h_addr_list;
|
||
|
|
hostent_addr.addr = ((struct in_addr*)windows_hostent->h_addr)->s_addr;
|
||
|
|
hostent_addr_p = &hostent_addr;
|
||
|
|
sal_hostent.h_addr_list = &hostent_addr_p;
|
||
|
|
|
||
|
|
return &sal_hostent;
|
||
|
|
}
|
||
|
|
|
||
|
|
int win_getaddrinfo(const char* nodename, const char* servname, const struct sal_type_addrinfo* hints, struct sal_type_addrinfo** res)
|
||
|
|
{
|
||
|
|
int port_nr = 0;
|
||
|
|
sal_type_ip_addr_t addr = { 0 };
|
||
|
|
struct sal_type_addrinfo* ai;
|
||
|
|
struct sal_type_sockaddr_storage* sa;
|
||
|
|
size_t total_size = 0;
|
||
|
|
size_t namelen = 0;
|
||
|
|
char* change_point;
|
||
|
|
|
||
|
|
if (res == RT_NULL)
|
||
|
|
return EAI_FAIL;
|
||
|
|
*res = RT_NULL;
|
||
|
|
|
||
|
|
if ((nodename == RT_NULL) && (servname == RT_NULL))
|
||
|
|
return EAI_NONAME;
|
||
|
|
|
||
|
|
if ((hints != RT_NULL)
|
||
|
|
&& (hints->ai_family != AF_INET && hints->ai_family != AF_UNSPEC))
|
||
|
|
return EAI_FAMILY;
|
||
|
|
|
||
|
|
if (servname != RT_NULL)
|
||
|
|
{
|
||
|
|
/* service name specified: convert to port number */
|
||
|
|
port_nr = atoi(servname);
|
||
|
|
if ((port_nr <= 0) || (port_nr > 0xffff))
|
||
|
|
{
|
||
|
|
return EAI_SERVICE;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
if (nodename != RT_NULL)
|
||
|
|
{
|
||
|
|
WINDOWS_PERVIOS_COMAMND;
|
||
|
|
struct hostent* hostent = gethostbyname(nodename);
|
||
|
|
if (hostent == RT_NULL)
|
||
|
|
return EAI_FAIL;
|
||
|
|
addr.addr = ((struct in_addr*)hostent->h_addr)->s_addr;
|
||
|
|
}
|
||
|
|
|
||
|
|
total_size = sizeof(struct sal_type_addrinfo) + sizeof(struct sal_type_sockaddr_storage);
|
||
|
|
if (nodename != RT_NULL)
|
||
|
|
{
|
||
|
|
namelen = strlen(nodename);
|
||
|
|
if (namelen > DNS_MAX_NAME_LENGTH)
|
||
|
|
{
|
||
|
|
/* invalid name length */
|
||
|
|
return EAI_FAIL;
|
||
|
|
}
|
||
|
|
RT_ASSERT(total_size + namelen + 1 > total_size);
|
||
|
|
total_size += namelen + 1;
|
||
|
|
}
|
||
|
|
|
||
|
|
RT_ASSERT(total_size <= sizeof(struct sal_type_addrinfo) + sizeof(struct sal_type_sockaddr_storage) + DNS_MAX_NAME_LENGTH + 1);
|
||
|
|
ai = (struct sal_type_addrinfo*)rt_malloc(total_size);
|
||
|
|
if (ai == RT_NULL)
|
||
|
|
{
|
||
|
|
return EAI_MEMORY;
|
||
|
|
}
|
||
|
|
rt_memset(ai, 0, total_size);
|
||
|
|
|
||
|
|
/* cast through void* to get rid of alignment warnings */
|
||
|
|
sa = (struct sal_type_sockaddr_storage*)(void*)((uint8_t*)ai + sizeof(struct sal_type_addrinfo));
|
||
|
|
struct sal_type_sockaddr_in* sa4 = (struct sal_type_sockaddr_in*)sa;
|
||
|
|
|
||
|
|
sa4->sin_family = AF_INET;
|
||
|
|
sa4->sin_addr.sal_type_s_addr = addr.addr;
|
||
|
|
sa4->sin_port = htons((uint16_t)port_nr);
|
||
|
|
|
||
|
|
/* set up addrinfo */
|
||
|
|
if (hints != RT_NULL)
|
||
|
|
{
|
||
|
|
/* copy socktype & protocol from hints if specified */
|
||
|
|
ai->ai_socktype = hints->ai_socktype;
|
||
|
|
ai->ai_protocol = hints->ai_protocol;
|
||
|
|
}
|
||
|
|
if (nodename != RT_NULL)
|
||
|
|
{
|
||
|
|
/* copy nodename to canonname if specified */
|
||
|
|
ai->ai_canonname = ((char*)ai + sizeof(struct sal_type_addrinfo) + sizeof(struct sal_type_sockaddr_storage));
|
||
|
|
rt_memcpy(ai->ai_canonname, nodename, namelen);
|
||
|
|
ai->ai_canonname[namelen] = 0;
|
||
|
|
}
|
||
|
|
|
||
|
|
ai->ai_addrlen = sizeof(struct sal_type_sockaddr_storage);
|
||
|
|
ai->ai_addr = (struct sal_type_sockaddr*)sa;
|
||
|
|
ai->ai_family = sa4->sin_family;
|
||
|
|
|
||
|
|
*res = ai;
|
||
|
|
|
||
|
|
return 0;
|
||
|
|
}
|
||
|
|
|
||
|
|
void win_freeaddrinfo(struct sal_type_addrinfo* ai)
|
||
|
|
{
|
||
|
|
struct sal_type_addrinfo* next;
|
||
|
|
|
||
|
|
while (ai != NULL)
|
||
|
|
{
|
||
|
|
next = ai->ai_next;
|
||
|
|
rt_free(ai);
|
||
|
|
ai = next;
|
||
|
|
}
|
||
|
|
}
|