include/wine/unixlib.h

   1 /*
   2  * Definitions for Unix libraries
   3  *
   4  * Copyright (C) 2021 Alexandre Julliard
   5  *
   6  * This library is free software; you can redistribute it and/or
   7  * modify it under the terms of the GNU Lesser General Public
   8  * License as published by the Free Software Foundation; either
   9  * version 2.1 of the License, or (at your option) any later version.
  10  *
  11  * This library is distributed in the hope that it will be useful,
  12  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  14  * Lesser General Public License for more details.
  15  *
  16  * You should have received a copy of the GNU Lesser General Public
  17  * License along with this library; if not, write to the Free Software
  18  * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
  19  */
  20
  21 #ifndef __WINE_WINE_UNIXLIB_H
  22 #define __WINE_WINE_UNIXLIB_H
  23
  24 #include "winternl.h"
  25
  26 typedef UINT64 unixlib_handle_t;
  27
  28 #ifdef WINE_UNIX_LIB
  29
  30 typedef NTSTATUS (*unixlib_entry_t)( void *args );
  31
  32 extern DECLSPEC_EXPORT const unixlib_entry_t __wine_unix_call_funcs[];
  33 extern DECLSPEC_EXPORT const unixlib_entry_t __wine_unix_call_wow64_funcs[];
  34
  35 /* some useful helpers from ntdll */
  36 NTSYSAPI const char *ntdll_get_build_dir(void);
  37 NTSYSAPI const char *ntdll_get_data_dir(void);
  38 NTSYSAPI DWORD ntdll_umbstowcs( const char *src, DWORD srclen, WCHAR *dst, DWORD dstlen );
  39 NTSYSAPI int ntdll_wcstoumbs( const WCHAR *src, DWORD srclen, char *dst, DWORD dstlen, BOOL strict );
  40 NTSYSAPI int ntdll_wcsicmp( const WCHAR *str1, const WCHAR *str2 );
  41 NTSYSAPI int ntdll_wcsnicmp( const WCHAR *str1, const WCHAR *str2, int n );
  42
  43 /* exception handling */
  44
  45 #include <setjmp.h>
  46
  47 NTSYSAPI void ntdll_set_exception_jmp_buf( jmp_buf jmp );
  48
  49 #define __TRY \
  50     do { jmp_buf __jmp; \
  51          int __first = 1; \
  52          for (;;) if (!__first) \
  53          { \
  54              do {
  55
  56 #define __EXCEPT \
  57              } while(0); \
  58              ntdll_set_exception_jmp_buf( NULL ); \
  59              break; \
  60          } else { \
  61              if (setjmp( __jmp )) { \
  62                  do {
  63
  64 #define __ENDTRY \
  65                  } while (0); \
  66                  break; \
  67              } \
  68              ntdll_set_exception_jmp_buf( __jmp ); \
  69              __first = 0; \
  70          } \
  71     } while (0);
  72
  73 NTSYSAPI BOOLEAN KeAddSystemServiceTable( ULONG_PTR *funcs, ULONG_PTR *counters, ULONG limit,
  74                                           BYTE *arguments, ULONG index );
  75 NTSYSAPI NTSTATUS KeUserModeCallback( ULONG id, const void *args, ULONG len, void **ret_ptr, ULONG *ret_len );
  76
  77 /* wide char string functions */
  78
  79 static inline int ntdll_iswspace( WCHAR wc )
  80 {
  81     return ('\t' <= wc && wc <= '\r') || wc == ' ' || wc == 0xa0;
  82 }
  83
  84 static inline size_t ntdll_wcslen( const WCHAR *str )
  85 {
  86     const WCHAR *s = str;
  87     while (*s) s++;
  88     return s - str;
  89 }
  90
  91 static inline WCHAR *ntdll_wcscpy( WCHAR *dst, const WCHAR *src )
  92 {
  93     WCHAR *p = dst;
  94     while ((*p++ = *src++));
  95     return dst;
  96 }
  97
  98 static inline WCHAR *ntdll_wcscat( WCHAR *dst, const WCHAR *src )
  99 {
 100     ntdll_wcscpy( dst + ntdll_wcslen(dst), src );
 101     return dst;
 102 }
 103
 104 static inline int ntdll_wcscmp( const WCHAR *str1, const WCHAR *str2 )
 105 {
 106     while (*str1 && (*str1 == *str2)) { str1++; str2++; }
 107     return *str1 - *str2;
 108 }
 109
 110 static inline int ntdll_wcsncmp( const WCHAR *str1, const WCHAR *str2, int n )
 111 {
 112     if (n <= 0) return 0;
 113     while ((--n > 0) && *str1 && (*str1 == *str2)) { str1++; str2++; }
 114     return *str1 - *str2;
 115 }
 116
 117 static inline WCHAR *ntdll_wcschr( const WCHAR *str, WCHAR ch )
 118 {
 119     do { if (*str == ch) return (WCHAR *)(ULONG_PTR)str; } while (*str++);
 120     return NULL;
 121 }
 122
 123 static inline WCHAR *ntdll_wcsrchr( const WCHAR *str, WCHAR ch )
 124 {
 125     WCHAR *ret = NULL;
 126     do { if (*str == ch) ret = (WCHAR *)(ULONG_PTR)str; } while (*str++);
 127     return ret;
 128 }
 129
 130 static inline WCHAR *ntdll_wcspbrk( const WCHAR *str, const WCHAR *accept )
 131 {
 132     for ( ; *str; str++) if (ntdll_wcschr( accept, *str )) return (WCHAR *)(ULONG_PTR)str;
 133     return NULL;
 134 }
 135
 136 static inline SIZE_T ntdll_wcsspn( const WCHAR *str, const WCHAR *accept )
 137 {
 138     const WCHAR *ptr;
 139     for (ptr = str; *ptr; ptr++) if (!ntdll_wcschr( accept, *ptr )) break;
 140     return ptr - str;
 141 }
 142
 143 static inline SIZE_T ntdll_wcscspn( const WCHAR *str, const WCHAR *reject )
 144 {
 145     const WCHAR *ptr;
 146     for (ptr = str; *ptr; ptr++) if (ntdll_wcschr( reject, *ptr )) break;
 147     return ptr - str;
 148 }
 149
 150 static inline LONG ntdll_wcstol( const WCHAR *s, WCHAR **end, int base )
 151 {
 152     BOOL negative = FALSE, empty = TRUE;
 153     LONG ret = 0;
 154
 155     if (base < 0 || base == 1 || base > 36) return 0;
 156     if (end) *end = (WCHAR *)s;
 157     while (ntdll_iswspace(*s)) s++;
 158
 159     if (*s == '-')
 160     {
 161         negative = TRUE;
 162         s++;
 163     }
 164     else if (*s == '+') s++;
 165
 166     if ((base == 0 || base == 16) && s[0] == '0' && (s[1] == 'x' || s[1] == 'X'))
 167     {
 168         base = 16;
 169         s += 2;
 170     }
 171     if (base == 0) base = s[0] != '0' ? 10 : 8;
 172
 173     while (*s)
 174     {
 175         int v;
 176
 177         if ('0' <= *s && *s <= '9') v = *s - '0';
 178         else if ('A' <= *s && *s <= 'Z') v = *s - 'A' + 10;
 179         else if ('a' <= *s && *s <= 'z') v = *s - 'a' + 10;
 180         else break;
 181         if (v >= base) break;
 182         if (negative) v = -v;
 183         s++;
 184         empty = FALSE;
 185
 186         if (!negative && (ret > MAXLONG / base || ret * base > MAXLONG - v))
 187             ret = MAXLONG;
 188         else if (negative && (ret < (LONG)MINLONG / base || ret * base < (LONG)(MINLONG - v)))
 189             ret = MINLONG;
 190         else
 191             ret = ret * base + v;
 192     }
 193
 194     if (end && !empty) *end = (WCHAR *)s;
 195     return ret;
 196 }
 197
 198 static inline ULONG ntdll_wcstoul( const WCHAR *s, WCHAR **end, int base )
 199 {
 200     BOOL negative = FALSE, empty = TRUE;
 201     ULONG ret = 0;
 202
 203     if (base < 0 || base == 1 || base > 36) return 0;
 204     if (end) *end = (WCHAR *)s;
 205     while (ntdll_iswspace(*s)) s++;
 206
 207     if (*s == '-')
 208     {
 209         negative = TRUE;
 210         s++;
 211     }
 212     else if (*s == '+') s++;
 213
 214     if ((base == 0 || base == 16) && s[0] == '0' && (s[1] == 'x' || s[1] == 'X'))
 215     {
 216         base = 16;
 217         s += 2;
 218     }
 219     if (base == 0) base = s[0] != '0' ? 10 : 8;
 220
 221     while (*s)
 222     {
 223         int v;
 224
 225         if ('0' <= *s && *s <= '9') v = *s - '0';
 226         else if ('A' <= *s && *s <= 'Z') v = *s - 'A' + 10;
 227         else if ('a' <= *s && *s <= 'z') v = *s - 'a' + 10;
 228         else break;
 229         if (v >= base) break;
 230         s++;
 231         empty = FALSE;
 232
 233         if (ret > MAXDWORD / base || ret * base > MAXDWORD - v)
 234             ret = MAXDWORD;
 235         else
 236             ret = ret * base + v;
 237     }
 238
 239     if (end && !empty) *end = (WCHAR *)s;
 240     return negative ? -ret : ret;
 241 }
 242
 243 #define iswspace(ch)       ntdll_iswspace(ch)
 244 #define wcslen(str)        ntdll_wcslen(str)
 245 #define wcscpy(dst,src)    ntdll_wcscpy(dst,src)
 246 #define wcscat(dst,src)    ntdll_wcscat(dst,src)
 247 #define wcscmp(s1,s2)      ntdll_wcscmp(s1,s2)
 248 #define wcsncmp(s1,s2,n)   ntdll_wcsncmp(s1,s2,n)
 249 #define wcschr(str,ch)     ntdll_wcschr(str,ch)
 250 #define wcsrchr(str,ch)    ntdll_wcsrchr(str,ch)
 251 #define wcspbrk(str,ac)    ntdll_wcspbrk(str,ac)
 252 #define wcsspn(str,ac)     ntdll_wcsspn(str,ac)
 253 #define wcscspn(str,rej)   ntdll_wcscspn(str,rej)
 254 #define wcsicmp(s1, s2)    ntdll_wcsicmp(s1,s2)
 255 #define wcsnicmp(s1, s2,n) ntdll_wcsnicmp(s1,s2,n)
 256 #define wcstol(str,e,b)    ntdll_wcstol(str,e,b)
 257 #define wcstoul(str,e,b)   ntdll_wcstoul(str,e,b)
 258
 259 #else /* WINE_UNIX_LIB */
 260
 261 extern unixlib_handle_t __wine_unixlib_handle;
 262 extern NTSTATUS (WINAPI *__wine_unix_call_dispatcher)( unixlib_handle_t, unsigned int, void * );
 263 extern NTSTATUS WINAPI __wine_init_unix_call(void);
 264
 265 #ifdef __arm64ec__
 266 NTSTATUS __wine_unix_call_arm64ec( unixlib_handle_t handle, unsigned int code, void *args );
 267 static inline NTSTATUS __wine_unix_call( unixlib_handle_t handle, unsigned int code, void *args )
 268 {
 269     return __wine_unix_call_arm64ec( handle, code, args );
 270 }
 271 #else
 272 static inline NTSTATUS __wine_unix_call( unixlib_handle_t handle, unsigned int code, void *args )
 273 {
 274     return __wine_unix_call_dispatcher( handle, code, args );
 275 }
 276 #endif
 277
 278 #define WINE_UNIX_CALL(code,args) __wine_unix_call( __wine_unixlib_handle, (code), (args) )
 279
 280 #endif /* WINE_UNIX_LIB */
 281
 282 #endif  /* __WINE_WINE_UNIXLIB_H */