master
  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