1/**
2 * This file has no copyright assigned and is placed in the Public Domain.
3 * This file is part of the mingw-w64 runtime package.
4 * No warranty is given; refer to the file DISCLAIMER.PD within this package.
5 */
6#ifndef _WSPIAPI_H_
7#define _WSPIAPI_H_
8
9#include <stdio.h>
10#include <stdlib.h>
11#include <malloc.h>
12#include <string.h>
13#include <ws2tcpip.h>
14
15#define _WSPIAPI_STRCPY_S(_Dst,_Size,_Src) strcpy((_Dst),(_Src))
16#define _WSPIAPI_STRCAT_S(_Dst,_Size,_Src) strcat((_Dst),(_Src))
17#define _WSPIAPI_STRNCPY_S(_Dst,_Size,_Src,_Count) strncpy((_Dst),(_Src),(_Count)); (_Dst)[(_Size) - 1] = 0
18#define _WSPIAPI_SPRINTF_S_1(_Dst,_Size,_Format,_Arg1) sprintf((_Dst),(_Format),(_Arg1))
19
20#ifndef _WSPIAPI_COUNTOF
21#ifndef __cplusplus
22#define _WSPIAPI_COUNTOF(_Array) (sizeof(_Array) / sizeof(_Array[0]))
23#else
24template <typename __CountofType,size_t __wspiapi_countof_helper_N> char (&__wspiapi_countof_helper(__CountofType (&_Array)[__wspiapi_countof_helper_N]))[__wspiapi_countof_helper_N];
25#define _WSPIAPI_COUNTOF(_Array) sizeof(__wspiapi_countof_helper(_Array))
26#endif
27#endif
28
29#define WspiapiMalloc(tSize) calloc(1,(tSize))
30#define WspiapiFree(p) free(p)
31#define WspiapiSwap(a,b,c) { (c) = (a); (a) = (b); (b) = (c); }
32#define getaddrinfo WspiapiGetAddrInfo
33#define getnameinfo WspiapiGetNameInfo
34#define freeaddrinfo WspiapiFreeAddrInfo
35
36typedef int (WINAPI *WSPIAPI_PGETADDRINFO)(const char *nodename,const char *servname,const struct addrinfo *hints,struct addrinfo **res);
37typedef int (WINAPI *WSPIAPI_PGETNAMEINFO)(const struct sockaddr *sa,socklen_t salen,char *host,size_t hostlen,char *serv,size_t servlen,int flags);
38typedef void (WINAPI *WSPIAPI_PFREEADDRINFO)(struct addrinfo *ai);
39
40#ifdef __cplusplus
41extern "C" {
42#endif
43 typedef struct {
44 char const *pszName;
45 FARPROC pfAddress;
46 } WSPIAPI_FUNCTION;
47
48#define WSPIAPI_FUNCTION_ARRAY { { "getaddrinfo",(FARPROC)(void(*)(void)) WspiapiLegacyGetAddrInfo }, \
49 { "getnameinfo",(FARPROC)(void(*)(void)) WspiapiLegacyGetNameInfo }, \
50 { "freeaddrinfo",(FARPROC)(void(*)(void)) WspiapiLegacyFreeAddrInfo } }
51
52 char *WINAPI WspiapiStrdup (const char *pszString);
53 WINBOOL WINAPI WspiapiParseV4Address (const char *pszAddress,PDWORD pdwAddress);
54 struct addrinfo * WINAPI WspiapiNewAddrInfo (int iSocketType,int iProtocol,WORD wPort,DWORD dwAddress);
55 int WINAPI WspiapiQueryDNS (const char *pszNodeName,int iSocketType,int iProtocol,WORD wPort,char pszAlias[NI_MAXHOST],struct addrinfo **pptResult);
56 int WINAPI WspiapiLookupNode (const char *pszNodeName,int iSocketType,int iProtocol,WORD wPort,WINBOOL bAI_CANONNAME,struct addrinfo **pptResult);
57 int WINAPI WspiapiClone (WORD wPort,struct addrinfo *ptResult);
58 void WINAPI WspiapiLegacyFreeAddrInfo (struct addrinfo *ptHead);
59 int WINAPI WspiapiLegacyGetAddrInfo(const char *pszNodeName,const char *pszServiceName,const struct addrinfo *ptHints,struct addrinfo **pptResult);
60 int WINAPI WspiapiLegacyGetNameInfo(const struct sockaddr *ptSocketAddress,socklen_t tSocketLength,char *pszNodeName,size_t tNodeLength,char *pszServiceName,size_t tServiceLength,int iFlags);
61 FARPROC WINAPI WspiapiLoad(WORD wFunction);
62 int WINAPI WspiapiGetAddrInfo(const char *nodename,const char *servname,const struct addrinfo *hints,struct addrinfo **res);
63 int WINAPI WspiapiGetNameInfo (const struct sockaddr *sa,socklen_t salen,char *host,size_t hostlen,char *serv,size_t servlen,int flags);
64 void WINAPI WspiapiFreeAddrInfo (struct addrinfo *ai);
65
66#ifndef __CRT__NO_INLINE
67 __CRT_INLINE char * WINAPI
68 WspiapiStrdup (const char *pszString)
69 {
70 char *rstr;
71 size_t szlen;
72
73 if(!pszString)
74 return NULL;
75 szlen = strlen(pszString) + 1;
76 rstr = (char *) WspiapiMalloc (szlen);
77 if (!rstr)
78 return NULL;
79 strcpy (rstr, pszString);
80 return rstr;
81 }
82
83 __CRT_INLINE WINBOOL WINAPI
84 WspiapiParseV4Address (const char *pszAddress, PDWORD pdwAddress)
85 {
86 DWORD dwAddress = 0;
87 const char *h = NULL;
88 int cnt;
89
90 for (cnt = 0,h = pszAddress; *h != 0; h++)
91 if (h[0] == '.')
92 cnt++;
93 if (cnt != 3)
94 return FALSE;
95 dwAddress = inet_addr (pszAddress);
96 if (dwAddress == INADDR_NONE)
97 return FALSE;
98 *pdwAddress = dwAddress;
99 return TRUE;
100 }
101
102 __CRT_INLINE struct addrinfo * WINAPI
103 WspiapiNewAddrInfo (int iSocketType,int iProtocol, WORD wPort,DWORD dwAddress)
104 {
105 struct addrinfo *n;
106 struct sockaddr_in *pa;
107
108 if ((n = (struct addrinfo *) WspiapiMalloc (sizeof (struct addrinfo))) == NULL)
109 return NULL;
110 if ((pa = (struct sockaddr_in *) WspiapiMalloc (sizeof(struct sockaddr_in))) == NULL)
111 {
112 WspiapiFree(n);
113 return NULL;
114 }
115 pa->sin_family = AF_INET;
116 pa->sin_port = wPort;
117 pa->sin_addr.s_addr = dwAddress;
118 n->ai_family = PF_INET;
119 n->ai_socktype = iSocketType;
120 n->ai_protocol = iProtocol;
121 n->ai_addrlen = sizeof (struct sockaddr_in);
122 n->ai_addr = (struct sockaddr *) pa;
123 return n;
124 }
125
126 __CRT_INLINE int WINAPI
127 WspiapiLookupNode (const char *pszNodeName, int iSocketType, int iProtocol, WORD wPort,
128 WINBOOL bAI_CANONNAME, struct addrinfo **pptResult)
129 {
130 int err = 0, cntAlias = 0;
131 char name[NI_MAXHOST] = "";
132 char alias[NI_MAXHOST] = "";
133 char *pname = name, *palias = alias, *tmp = NULL;
134
135 strncpy (pname, pszNodeName, NI_MAXHOST - 1);
136 pname[NI_MAXHOST - 1] = 0;
137 for (;;)
138 {
139 err = WspiapiQueryDNS (pszNodeName, iSocketType, iProtocol, wPort, palias, pptResult);
140 if (err)
141 break;
142 if (*pptResult)
143 break;
144 ++cntAlias;
145 if (strlen (palias) == 0 || !strcmp (pname, palias) || cntAlias == 16)
146 {
147 err = EAI_FAIL;
148 break;
149 }
150 WspiapiSwap(pname, palias, tmp);
151 }
152 if (!err && bAI_CANONNAME)
153 {
154 (*pptResult)->ai_canonname = WspiapiStrdup (palias);
155 if (!(*pptResult)->ai_canonname)
156 err = EAI_MEMORY;
157 }
158 return err;
159 }
160
161 __CRT_INLINE int WINAPI
162 WspiapiClone (WORD wPort,struct addrinfo *ptResult)
163 {
164 struct addrinfo *p = NULL;
165 struct addrinfo *n = NULL;
166
167 for (p = ptResult; p != NULL;)
168 {
169 n = WspiapiNewAddrInfo (SOCK_DGRAM, p->ai_protocol, wPort,
170 ((struct sockaddr_in *) p->ai_addr)->sin_addr.s_addr);
171 if (!n)
172 break;
173 n->ai_next = p->ai_next;
174 p->ai_next = n;
175 p = n->ai_next;
176 }
177 if (p != NULL)
178 return EAI_MEMORY;
179 return 0;
180 }
181
182 __CRT_INLINE void WINAPI
183 WspiapiLegacyFreeAddrInfo (struct addrinfo *ptHead)
184 {
185 struct addrinfo *p;
186
187 for (p = ptHead; p != NULL; p = ptHead)
188 {
189 if (p->ai_canonname)
190 WspiapiFree (p->ai_canonname);
191 if (p->ai_addr)
192 WspiapiFree (p->ai_addr);
193 ptHead = p->ai_next;
194 WspiapiFree (p);
195 }
196 }
197#endif /* !__CRT__NO_INLINE */
198
199#ifdef __cplusplus
200}
201#endif
202
203#endif