secur32: Sign-compare warning fix.
[wine] / dlls / secur32 / secur32.c
1 /* Copyright (C) 2004 Juan Lang
2  *
3  * This file implements loading of SSP DLLs.
4  *
5  * This library is free software; you can redistribute it and/or
6  * modify it under the terms of the GNU Lesser General Public
7  * License as published by the Free Software Foundation; either
8  * version 2.1 of the License, or (at your option) any later version.
9  *
10  * This library is distributed in the hope that it will be useful,
11  * but WITHOUT ANY WARRANTY; without even the implied warranty of
12  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
13  * Lesser General Public License for more details.
14  *
15  * You should have received a copy of the GNU Lesser General Public
16  * License along with this library; if not, write to the Free Software
17  * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
18  */
19 #include <assert.h>
20 #include <stdarg.h>
21
22 #include "ntstatus.h"
23 #define WIN32_NO_STATUS
24 #include "windef.h"
25 #include "winbase.h"
26 #include "winnls.h"
27 #include "winreg.h"
28 #include "winternl.h"
29 #include "shlwapi.h"
30 #include "sspi.h"
31 #include "secur32_priv.h"
32 #include "secext.h"
33 #include "ntsecapi.h"
34 #include "thunks.h"
35
36 #include "wine/list.h"
37 #include "wine/debug.h"
38
39 WINE_DEFAULT_DEBUG_CHANNEL(secur32);
40
41 /**
42  *  Type definitions
43  */
44
45 typedef struct _SecurePackageTable
46 {
47     DWORD numPackages;
48     DWORD numAllocated;
49     struct list table;
50 } SecurePackageTable;
51
52 typedef struct _SecureProviderTable
53 {
54     DWORD numProviders;
55     DWORD numAllocated;
56     struct list table;
57 } SecureProviderTable;
58
59 /**
60  *  Prototypes
61  */
62
63 /* Tries to load moduleName as a provider.  If successful, enumerates what
64  * packages it can and adds them to the package and provider tables.  Resizes
65  * tables as necessary.
66  */
67 static void _tryLoadProvider(PWSTR moduleName);
68
69 /* Initialization: read securityproviders value and attempt to open each dll
70  * there.  For each DLL, call _tryLoadProvider to see if it's really an SSP.
71  * Two undocumented functions, AddSecurityPackage(A/W) and
72  * DeleteSecurityPackage(A/W), seem suspiciously like they'd register or
73  * unregister a dll, but I'm not sure.
74  */
75 static void SECUR32_initializeProviders(void);
76
77 /* Frees all loaded packages and providers */
78 static void SECUR32_freeProviders(void);
79
80 /**
81  *  Globals
82  */
83
84 static CRITICAL_SECTION cs;
85 static SecurePackageTable *packageTable = NULL;
86 static SecureProviderTable *providerTable = NULL;
87
88 static SecurityFunctionTableA securityFunctionTableA = {
89     SECURITY_SUPPORT_PROVIDER_INTERFACE_VERSION,
90     EnumerateSecurityPackagesA,
91     QueryCredentialsAttributesA,
92     AcquireCredentialsHandleA,
93     FreeCredentialsHandle,
94     NULL, /* Reserved2 */
95     InitializeSecurityContextA,
96     AcceptSecurityContext,
97     CompleteAuthToken,
98     DeleteSecurityContext,
99     ApplyControlToken,
100     QueryContextAttributesA,
101     ImpersonateSecurityContext,
102     RevertSecurityContext,
103     MakeSignature,
104     VerifySignature,
105     FreeContextBuffer,
106     QuerySecurityPackageInfoA,
107     EncryptMessage, /* Reserved3 */
108     DecryptMessage, /* Reserved4 */
109     ExportSecurityContext,
110     ImportSecurityContextA,
111     AddCredentialsA,
112     NULL, /* Reserved8 */
113     QuerySecurityContextToken,
114     EncryptMessage,
115     DecryptMessage,
116     SetContextAttributesA
117 };
118
119 static SecurityFunctionTableW securityFunctionTableW = {
120     SECURITY_SUPPORT_PROVIDER_INTERFACE_VERSION,
121     EnumerateSecurityPackagesW,
122     QueryCredentialsAttributesW,
123     AcquireCredentialsHandleW,
124     FreeCredentialsHandle,
125     NULL, /* Reserved2 */
126     InitializeSecurityContextW,
127     AcceptSecurityContext,
128     CompleteAuthToken,
129     DeleteSecurityContext,
130     ApplyControlToken,
131     QueryContextAttributesW,
132     ImpersonateSecurityContext,
133     RevertSecurityContext,
134     MakeSignature,
135     VerifySignature,
136     FreeContextBuffer,
137     QuerySecurityPackageInfoW,
138     EncryptMessage, /* Reserved3 */
139     DecryptMessage, /* Reserved4 */
140     ExportSecurityContext,
141     ImportSecurityContextW,
142     AddCredentialsW,
143     NULL, /* Reserved8 */
144     QuerySecurityContextToken,
145     EncryptMessage,
146     DecryptMessage,
147     SetContextAttributesW
148 };
149
150 /***********************************************************************
151  *              InitSecurityInterfaceA (SECUR32.@)
152  */
153 PSecurityFunctionTableA WINAPI InitSecurityInterfaceA(void)
154 {
155     return &securityFunctionTableA;
156 }
157
158 /***********************************************************************
159  *              InitSecurityInterfaceW (SECUR32.@)
160  */
161 PSecurityFunctionTableW WINAPI InitSecurityInterfaceW(void)
162 {
163     return &securityFunctionTableW;
164 }
165
166 PWSTR SECUR32_strdupW(PCWSTR str)
167 {
168     PWSTR ret;
169
170     if (str)
171     {
172         ret = HeapAlloc(GetProcessHeap(), 0, (lstrlenW(str) + 1) * sizeof(WCHAR));
173         if (ret)
174             lstrcpyW(ret, str);
175     }
176     else
177         ret = NULL;
178     return ret;
179 }
180
181 PWSTR SECUR32_AllocWideFromMultiByte(PCSTR str)
182 {
183     PWSTR ret;
184
185     if (str)
186     {
187         int charsNeeded = MultiByteToWideChar(CP_ACP, 0, str, -1, NULL, 0);
188
189         if (charsNeeded)
190         {
191             ret = HeapAlloc(GetProcessHeap(), 0, charsNeeded * sizeof(WCHAR));
192             if (ret)
193                 MultiByteToWideChar(CP_ACP, 0, str, -1, ret, charsNeeded);
194         }
195         else
196             ret = NULL;
197     }
198     else
199         ret = NULL;
200     return ret;
201 }
202
203 PSTR SECUR32_AllocMultiByteFromWide(PCWSTR str)
204 {
205     PSTR ret;
206
207     if (str)
208     {
209         int charsNeeded = WideCharToMultiByte(CP_ACP, 0, str, -1, NULL, 0,
210          NULL, NULL);
211
212         if (charsNeeded)
213         {
214             ret = HeapAlloc(GetProcessHeap(), 0, charsNeeded);
215             if (ret)
216                 WideCharToMultiByte(CP_ACP, 0, str, -1, ret, charsNeeded,
217                  NULL, NULL);
218         }
219         else
220             ret = NULL;
221     }
222     else
223         ret = NULL;
224     return ret;
225 }
226
227 static void _makeFnTableA(PSecurityFunctionTableA fnTableA,
228  const SecurityFunctionTableA *inFnTableA,
229  const SecurityFunctionTableW *inFnTableW)
230 {
231     if (fnTableA)
232     {
233         if (inFnTableA)
234         {
235             /* The size of the version 1 table is based on platform sdk's
236              * sspi.h, though the sample ssp also provided with platform sdk
237              * implies only functions through QuerySecurityPackageInfoA are
238              * implemented (yikes)
239              */
240             size_t tableSize = inFnTableA->dwVersion == 1 ?
241              (const BYTE *)&inFnTableA->SetContextAttributesA -
242              (const BYTE *)inFnTableA : sizeof(SecurityFunctionTableA);
243
244             memcpy(fnTableA, inFnTableA, tableSize);
245             /* override this, since we can do it internally anyway */
246             fnTableA->QuerySecurityPackageInfoA =
247              QuerySecurityPackageInfoA;
248         }
249         else if (inFnTableW)
250         {
251             /* functions with thunks */
252             if (inFnTableW->AcquireCredentialsHandleW)
253                 fnTableA->AcquireCredentialsHandleA =
254                  thunk_AcquireCredentialsHandleA;
255             if (inFnTableW->InitializeSecurityContextW)
256                 fnTableA->InitializeSecurityContextA =
257                  thunk_InitializeSecurityContextA;
258             if (inFnTableW->ImportSecurityContextW)
259                 fnTableA->ImportSecurityContextA =
260                  thunk_ImportSecurityContextA;
261             if (inFnTableW->AddCredentialsW)
262                 fnTableA->AddCredentialsA =
263                  thunk_AddCredentialsA;
264             if (inFnTableW->QueryCredentialsAttributesW)
265                 fnTableA->QueryCredentialsAttributesA =
266                  thunk_QueryCredentialsAttributesA;
267             if (inFnTableW->QueryContextAttributesW)
268                 fnTableA->QueryContextAttributesA =
269                  thunk_QueryContextAttributesA;
270             if (inFnTableW->SetContextAttributesW)
271                 fnTableA->SetContextAttributesA =
272                  thunk_SetContextAttributesA;
273             /* this can't be thunked, there's no extra param to know which
274              * package to forward to */
275             fnTableA->EnumerateSecurityPackagesA = NULL;
276             /* functions with no thunks needed */
277             fnTableA->AcceptSecurityContext = inFnTableW->AcceptSecurityContext;
278             fnTableA->CompleteAuthToken = inFnTableW->CompleteAuthToken;
279             fnTableA->DeleteSecurityContext = inFnTableW->DeleteSecurityContext;
280             fnTableA->ImpersonateSecurityContext =
281              inFnTableW->ImpersonateSecurityContext;
282             fnTableA->RevertSecurityContext = inFnTableW->RevertSecurityContext;
283             fnTableA->MakeSignature = inFnTableW->MakeSignature;
284             fnTableA->VerifySignature = inFnTableW->VerifySignature;
285             fnTableA->FreeContextBuffer = inFnTableW->FreeContextBuffer;
286             fnTableA->QuerySecurityPackageInfoA =
287              QuerySecurityPackageInfoA;
288             fnTableA->ExportSecurityContext =
289              inFnTableW->ExportSecurityContext;
290             fnTableA->QuerySecurityContextToken =
291              inFnTableW->QuerySecurityContextToken;
292             fnTableA->EncryptMessage = inFnTableW->EncryptMessage;
293             fnTableA->DecryptMessage = inFnTableW->DecryptMessage;
294         }
295     }
296 }
297
298 static void _makeFnTableW(PSecurityFunctionTableW fnTableW,
299  const SecurityFunctionTableA *inFnTableA,
300  const SecurityFunctionTableW *inFnTableW)
301 {
302     if (fnTableW)
303     {
304         if (inFnTableW)
305         {
306             /* The size of the version 1 table is based on platform sdk's
307              * sspi.h, though the sample ssp also provided with platform sdk
308              * implies only functions through QuerySecurityPackageInfoA are
309              * implemented (yikes)
310              */
311             size_t tableSize = inFnTableW->dwVersion == 1 ?
312              (const BYTE *)&inFnTableW->SetContextAttributesW -
313              (const BYTE *)inFnTableW : sizeof(SecurityFunctionTableW);
314
315             memcpy(fnTableW, inFnTableW, tableSize);
316             /* override this, since we can do it internally anyway */
317             fnTableW->QuerySecurityPackageInfoW =
318              QuerySecurityPackageInfoW;
319         }
320         else if (inFnTableA)
321         {
322             /* functions with thunks */
323             if (inFnTableA->AcquireCredentialsHandleA)
324                 fnTableW->AcquireCredentialsHandleW =
325                  thunk_AcquireCredentialsHandleW;
326             if (inFnTableA->InitializeSecurityContextA)
327                 fnTableW->InitializeSecurityContextW =
328                  thunk_InitializeSecurityContextW;
329             if (inFnTableA->ImportSecurityContextA)
330                 fnTableW->ImportSecurityContextW =
331                  thunk_ImportSecurityContextW;
332             if (inFnTableA->AddCredentialsA)
333                 fnTableW->AddCredentialsW =
334                  thunk_AddCredentialsW;
335             if (inFnTableA->QueryCredentialsAttributesA)
336                 fnTableW->QueryCredentialsAttributesW =
337                  thunk_QueryCredentialsAttributesW;
338             if (inFnTableA->QueryContextAttributesA)
339                 fnTableW->QueryContextAttributesW =
340                  thunk_QueryContextAttributesW;
341             if (inFnTableA->SetContextAttributesA)
342                 fnTableW->SetContextAttributesW =
343                  thunk_SetContextAttributesW;
344             /* this can't be thunked, there's no extra param to know which
345              * package to forward to */
346             fnTableW->EnumerateSecurityPackagesW = NULL;
347             /* functions with no thunks needed */
348             fnTableW->AcceptSecurityContext = inFnTableA->AcceptSecurityContext;
349             fnTableW->CompleteAuthToken = inFnTableA->CompleteAuthToken;
350             fnTableW->DeleteSecurityContext = inFnTableA->DeleteSecurityContext;
351             fnTableW->ImpersonateSecurityContext =
352              inFnTableA->ImpersonateSecurityContext;
353             fnTableW->RevertSecurityContext = inFnTableA->RevertSecurityContext;
354             fnTableW->MakeSignature = inFnTableA->MakeSignature;
355             fnTableW->VerifySignature = inFnTableA->VerifySignature;
356             fnTableW->FreeContextBuffer = inFnTableA->FreeContextBuffer;
357             fnTableW->QuerySecurityPackageInfoW =
358              QuerySecurityPackageInfoW;
359             fnTableW->ExportSecurityContext =
360              inFnTableA->ExportSecurityContext;
361             fnTableW->QuerySecurityContextToken =
362              inFnTableA->QuerySecurityContextToken;
363             fnTableW->EncryptMessage = inFnTableA->EncryptMessage;
364             fnTableW->DecryptMessage = inFnTableA->DecryptMessage;
365         }
366     }
367 }
368
369 static void _copyPackageInfo(PSecPkgInfoW info, const SecPkgInfoA *inInfoA,
370  const SecPkgInfoW *inInfoW)
371 {
372     if (info && (inInfoA || inInfoW))
373     {
374         /* odd, I know, but up until Name and Comment the structures are
375          * identical
376          */
377         memcpy(info, inInfoW ? inInfoW : (const SecPkgInfoW *)inInfoA, sizeof(*info));
378         if (inInfoW)
379         {
380             info->Name = SECUR32_strdupW(inInfoW->Name);
381             info->Comment = SECUR32_strdupW(inInfoW->Comment);
382         }
383         else
384         {
385             info->Name = SECUR32_AllocWideFromMultiByte(inInfoA->Name);
386             info->Comment = SECUR32_AllocWideFromMultiByte(inInfoA->Comment);
387         }
388     }
389 }
390
391 SecureProvider *SECUR32_addProvider(const SecurityFunctionTableA *fnTableA,
392  const SecurityFunctionTableW *fnTableW, PCWSTR moduleName)
393 {
394     SecureProvider *ret;
395
396     EnterCriticalSection(&cs);
397
398     if (!providerTable)
399     {
400         providerTable = HeapAlloc(GetProcessHeap(), 0, sizeof(SecureProviderTable));
401         if (!providerTable)
402         {
403             LeaveCriticalSection(&cs);
404             return NULL;
405         }
406
407         list_init(&providerTable->table);
408     }
409
410     ret = HeapAlloc(GetProcessHeap(), 0, sizeof(SecureProvider));
411     if (!ret)
412     {
413         LeaveCriticalSection(&cs);
414         return NULL;
415     }
416
417     list_add_tail(&providerTable->table, &ret->entry);
418     ret->lib = NULL;
419
420     if (fnTableA || fnTableW)
421     {
422         ret->moduleName = moduleName ? SECUR32_strdupW(moduleName) : NULL;
423         _makeFnTableA(&ret->fnTableA, fnTableA, fnTableW);
424         _makeFnTableW(&ret->fnTableW, fnTableA, fnTableW);
425         ret->loaded = moduleName ? FALSE : TRUE;
426     }
427     else
428     {
429         ret->moduleName = SECUR32_strdupW(moduleName);
430         ret->loaded = FALSE;
431     }
432     
433     LeaveCriticalSection(&cs);
434     return ret;
435 }
436
437 void SECUR32_addPackages(SecureProvider *provider, ULONG toAdd,
438  const SecPkgInfoA *infoA, const SecPkgInfoW *infoW)
439 {
440     ULONG i;
441
442     assert(provider);
443     assert(infoA || infoW);
444
445     EnterCriticalSection(&cs);
446
447     if (!packageTable)
448     {
449         packageTable = HeapAlloc(GetProcessHeap(), 0, sizeof(SecurePackageTable));
450         if (!packageTable)
451         {
452             LeaveCriticalSection(&cs);
453             return;
454         }
455
456         packageTable->numPackages = 0;
457         list_init(&packageTable->table);
458     }
459         
460     for (i = 0; i < toAdd; i++)
461     {
462         SecurePackage *package = HeapAlloc(GetProcessHeap(), 0, sizeof(SecurePackage));
463         if (!package)
464             continue;
465
466         list_add_tail(&packageTable->table, &package->entry);
467
468         package->provider = provider;
469         _copyPackageInfo(&package->infoW,
470             infoA ? &infoA[i] : NULL,
471             infoW ? &infoW[i] : NULL);
472     }
473     packageTable->numPackages += toAdd;
474
475     LeaveCriticalSection(&cs);
476 }
477
478 static void _tryLoadProvider(PWSTR moduleName)
479 {
480     HMODULE lib = LoadLibraryW(moduleName);
481
482     if (lib)
483     {
484         INIT_SECURITY_INTERFACE_W pInitSecurityInterfaceW =
485          (INIT_SECURITY_INTERFACE_W)GetProcAddress(lib,
486          SECURITY_ENTRYPOINT_ANSIW);
487         INIT_SECURITY_INTERFACE_A pInitSecurityInterfaceA =
488          (INIT_SECURITY_INTERFACE_A)GetProcAddress(lib,
489          SECURITY_ENTRYPOINT_ANSIA);
490
491         TRACE("loaded %s, InitSecurityInterfaceA is %p, InitSecurityInterfaceW is %p\n",
492          debugstr_w(moduleName), pInitSecurityInterfaceA,
493          pInitSecurityInterfaceW);
494         if (pInitSecurityInterfaceW || pInitSecurityInterfaceA)
495         {
496             PSecurityFunctionTableA fnTableA = NULL;
497             PSecurityFunctionTableW fnTableW = NULL;
498             ULONG toAdd = 0;
499             PSecPkgInfoA infoA = NULL;
500             PSecPkgInfoW infoW = NULL;
501             SECURITY_STATUS ret = SEC_E_OK;
502
503             if (pInitSecurityInterfaceA)
504                 fnTableA = pInitSecurityInterfaceA();
505             if (pInitSecurityInterfaceW)
506                 fnTableW = pInitSecurityInterfaceW();
507             if (fnTableW && fnTableW->EnumerateSecurityPackagesW)
508             {
509                 if (fnTableW != &securityFunctionTableW)
510                     ret = fnTableW->EnumerateSecurityPackagesW(&toAdd, &infoW);
511                 else
512                     TRACE("%s has built-in providers, skip adding\n", debugstr_w(moduleName));
513             }
514             else if (fnTableA && fnTableA->EnumerateSecurityPackagesA)
515             {
516                 if (fnTableA != &securityFunctionTableA)
517                     ret = fnTableA->EnumerateSecurityPackagesA(&toAdd, &infoA);
518                 else
519                     TRACE("%s has built-in providers, skip adding\n", debugstr_w(moduleName));
520             }
521             if (ret == SEC_E_OK && toAdd > 0 && (infoW || infoA))
522             {
523                 SecureProvider *provider = SECUR32_addProvider(NULL, NULL,
524                  moduleName);
525
526                 if (provider)
527                     SECUR32_addPackages(provider, toAdd, infoA, infoW);
528                 if (infoW)
529                     fnTableW->FreeContextBuffer(infoW);
530                 else
531                     fnTableA->FreeContextBuffer(infoA);
532             }
533         }
534         FreeLibrary(lib);
535     }
536     else
537         WARN("failed to load %s\n", debugstr_w(moduleName));
538 }
539
540 static const WCHAR securityProvidersKeyW[] = {
541  'S','Y','S','T','E','M','\\','C','u','r','r','e','n','t','C','o','n','t','r',
542  'o','l','S','e','t','\\','C','o','n','t','r','o','l','\\','S','e','c','u','r',
543  'i','t','y','P','r','o','v','i','d','e','r','s','\0'
544  };
545 static const WCHAR securityProvidersW[] = {
546  'S','e','c','u','r','i','t','y','P','r','o','v','i','d','e','r','s',0
547  };
548  
549 static void SECUR32_initializeProviders(void)
550 {
551     HKEY key;
552     long apiRet;
553
554     TRACE("\n");
555     InitializeCriticalSection(&cs);
556     cs.DebugInfo->Spare[0] = (DWORD_PTR)(__FILE__ ": cs");
557     /* First load built-in providers */
558     SECUR32_initSchannelSP();
559     /* Do not load Negotiate yet. This breaks for some user on the wine-users
560      * mailing list as of 2006-09-12. Without Negotiate, applications should
561      * fall back to NTLM and that should work.*/
562 #if 0
563     SECUR32_initNegotiateSP();
564 #endif
565     SECUR32_initNTLMSP();
566     /* Now load providers from registry */
567     apiRet = RegOpenKeyExW(HKEY_LOCAL_MACHINE, securityProvidersKeyW, 0,
568      KEY_READ, &key);
569     if (apiRet == ERROR_SUCCESS)
570     {
571         WCHAR securityPkgNames[MAX_PATH]; /* arbitrary len */
572         DWORD size = sizeof(securityPkgNames) / sizeof(WCHAR), type;
573
574         apiRet = RegQueryValueExW(key, securityProvidersW, NULL, &type,
575          (PBYTE)securityPkgNames, &size);
576         if (apiRet == ERROR_SUCCESS && type == REG_SZ)
577         {
578             WCHAR *ptr;
579
580             size = size / sizeof(WCHAR);
581             for (ptr = securityPkgNames;
582               ptr < securityPkgNames + size; )
583             {
584                 WCHAR *comma;
585
586                 for (comma = ptr; *comma && *comma != ','; comma++)
587                     ;
588                 if (*comma == ',')
589                     *comma = '\0';
590                 for (; *ptr && isspace(*ptr) && ptr < securityPkgNames + size;
591                  ptr++)
592                     ;
593                 if (*ptr)
594                     _tryLoadProvider(ptr);
595                 ptr += lstrlenW(ptr) + 1;
596             }
597         }
598         RegCloseKey(key);
599     }
600 }
601
602 SecurePackage *SECUR32_findPackageW(PCWSTR packageName)
603 {
604     SecurePackage *ret = NULL;
605     BOOL matched = FALSE;
606
607     if (packageTable && packageName)
608     {
609         LIST_FOR_EACH_ENTRY(ret, &packageTable->table, SecurePackage, entry)
610         {
611             matched = !lstrcmpiW(ret->infoW.Name, packageName);
612             if (matched)
613                 break;
614         }
615         
616         if (!matched)
617                 return NULL;
618
619         if (ret->provider && !ret->provider->loaded)
620         {
621             ret->provider->lib = LoadLibraryW(ret->provider->moduleName);
622             if (ret->provider->lib)
623             {
624                 INIT_SECURITY_INTERFACE_W pInitSecurityInterfaceW =
625                  (INIT_SECURITY_INTERFACE_W)GetProcAddress(ret->provider->lib,
626                  SECURITY_ENTRYPOINT_ANSIW);
627                 INIT_SECURITY_INTERFACE_A pInitSecurityInterfaceA =
628                  (INIT_SECURITY_INTERFACE_A)GetProcAddress(ret->provider->lib,
629                  SECURITY_ENTRYPOINT_ANSIA);
630                 PSecurityFunctionTableA fnTableA = NULL;
631                 PSecurityFunctionTableW fnTableW = NULL;
632
633                 if (pInitSecurityInterfaceA)
634                     fnTableA = pInitSecurityInterfaceA();
635                 if (pInitSecurityInterfaceW)
636                     fnTableW = pInitSecurityInterfaceW();
637                 /* don't update built-in SecurityFunctionTable */
638                 if (fnTableA != &securityFunctionTableA)
639                     _makeFnTableA(&ret->provider->fnTableA, fnTableA, fnTableW);
640                 if (fnTableW != &securityFunctionTableW)
641                     _makeFnTableW(&ret->provider->fnTableW, fnTableA, fnTableW);
642                 ret->provider->loaded = TRUE;
643             }
644             else
645                 ret = NULL;
646         }
647     }
648     return ret;
649 }
650
651 SecurePackage *SECUR32_findPackageA(PCSTR packageName)
652 {
653     SecurePackage *ret;
654
655     if (packageTable && packageName)
656     {
657         UNICODE_STRING package;
658
659         RtlCreateUnicodeStringFromAsciiz(&package, packageName);
660         ret = SECUR32_findPackageW(package.Buffer);
661         RtlFreeUnicodeString(&package);
662     }
663     else
664         ret = NULL;
665     return ret;
666 }
667
668 static void SECUR32_freeProviders(void)
669 {
670     SecurePackage *package;
671     SecureProvider *provider;
672
673     TRACE("\n");
674     EnterCriticalSection(&cs);
675
676     SECUR32_deinitSchannelSP();
677
678     if (packageTable)
679     {
680         LIST_FOR_EACH_ENTRY(package, &packageTable->table, SecurePackage, entry)
681         {
682             HeapFree(GetProcessHeap(), 0, package->infoW.Name);
683             HeapFree(GetProcessHeap(), 0, package->infoW.Comment);
684         }
685
686         HeapFree(GetProcessHeap(), 0, packageTable);
687         packageTable = NULL;
688     }
689
690     if (providerTable)
691     {
692         LIST_FOR_EACH_ENTRY(provider, &providerTable->table, SecureProvider, entry)
693         {
694             HeapFree(GetProcessHeap(), 0, provider->moduleName);
695             if (provider->lib)
696                 FreeLibrary(provider->lib);
697         }
698
699         HeapFree(GetProcessHeap(), 0, providerTable);
700         providerTable = NULL;
701     }
702
703     LeaveCriticalSection(&cs);
704     cs.DebugInfo->Spare[0] = 0;
705     DeleteCriticalSection(&cs);
706 }
707
708 /***********************************************************************
709  *              FreeContextBuffer (SECUR32.@)
710  *
711  * Doh--if pv was allocated by a crypto package, this may not be correct.
712  * The sample ssp seems to use LocalAlloc/LocalFee, but there doesn't seem to
713  * be any guarantee, nor is there an alloc function in secur32.
714  */
715 SECURITY_STATUS WINAPI FreeContextBuffer(PVOID pv)
716 {
717     HeapFree(GetProcessHeap(), 0, pv);
718
719     return SEC_E_OK;
720 }
721
722 /***********************************************************************
723  *              EnumerateSecurityPackagesW (SECUR32.@)
724  */
725 SECURITY_STATUS WINAPI EnumerateSecurityPackagesW(PULONG pcPackages,
726  PSecPkgInfoW *ppPackageInfo)
727 {
728     SECURITY_STATUS ret = SEC_E_OK;
729
730     TRACE("(%p, %p)\n", pcPackages, ppPackageInfo);
731
732     /* windows just crashes if pcPackages or ppPackageInfo is NULL, so will I */
733     *pcPackages = 0;
734     EnterCriticalSection(&cs);
735     if (packageTable)
736     {
737         SecurePackage *package;
738         size_t bytesNeeded;
739
740         bytesNeeded = packageTable->numPackages * sizeof(SecPkgInfoW);
741         LIST_FOR_EACH_ENTRY(package, &packageTable->table, SecurePackage, entry)
742         {
743             if (package->infoW.Name)
744                 bytesNeeded += (lstrlenW(package->infoW.Name) + 1) * sizeof(WCHAR);
745             if (package->infoW.Comment)
746                 bytesNeeded += (lstrlenW(package->infoW.Comment) + 1) * sizeof(WCHAR);
747         }
748         if (bytesNeeded)
749         {
750             *ppPackageInfo = HeapAlloc(GetProcessHeap(), 0, bytesNeeded);
751             if (*ppPackageInfo)
752             {
753                 ULONG i = 0;
754                 PWSTR nextString;
755
756                 *pcPackages = packageTable->numPackages;
757                 nextString = (PWSTR)((PBYTE)*ppPackageInfo +
758                  packageTable->numPackages * sizeof(SecPkgInfoW));
759                 LIST_FOR_EACH_ENTRY(package, &packageTable->table, SecurePackage, entry)
760                 {
761                     PSecPkgInfoW pkgInfo = *ppPackageInfo + i++;
762
763                     *pkgInfo = package->infoW;
764                     if (package->infoW.Name)
765                     {
766                         TRACE("Name[%d] = %s\n", i - 1, debugstr_w(package->infoW.Name));
767                         pkgInfo->Name = nextString;
768                         lstrcpyW(nextString, package->infoW.Name);
769                         nextString += lstrlenW(nextString) + 1;
770                     }
771                     else
772                         pkgInfo->Name = NULL;
773                     if (package->infoW.Comment)
774                     {
775                         TRACE("Comment[%d] = %s\n", i - 1, debugstr_w(package->infoW.Comment));
776                         pkgInfo->Comment = nextString;
777                         lstrcpyW(nextString, package->infoW.Comment);
778                         nextString += lstrlenW(nextString) + 1;
779                     }
780                     else
781                         pkgInfo->Comment = NULL;
782                 }
783             }
784             else
785                 ret = SEC_E_INSUFFICIENT_MEMORY;
786         }
787     }
788     LeaveCriticalSection(&cs);
789     TRACE("<-- 0x%08x\n", ret);
790     return ret;
791 }
792
793 /* Converts info (which is assumed to be an array of cPackages SecPkgInfoW
794  * structures) into an array of SecPkgInfoA structures, which it returns.
795  */
796 static PSecPkgInfoA thunk_PSecPkgInfoWToA(ULONG cPackages,
797  const SecPkgInfoW *info)
798 {
799     PSecPkgInfoA ret;
800
801     if (info)
802     {
803         size_t bytesNeeded = cPackages * sizeof(SecPkgInfoA);
804         ULONG i;
805
806         for (i = 0; i < cPackages; i++)
807         {
808             if (info[i].Name)
809                 bytesNeeded += WideCharToMultiByte(CP_ACP, 0, info[i].Name,
810                  -1, NULL, 0, NULL, NULL);
811             if (info[i].Comment)
812                 bytesNeeded += WideCharToMultiByte(CP_ACP, 0, info[i].Comment,
813                  -1, NULL, 0, NULL, NULL);
814         }
815         ret = HeapAlloc(GetProcessHeap(), 0, bytesNeeded);
816         if (ret)
817         {
818             PSTR nextString;
819
820             nextString = (PSTR)((PBYTE)ret + cPackages * sizeof(SecPkgInfoA));
821             for (i = 0; i < cPackages; i++)
822             {
823                 PSecPkgInfoA pkgInfo = ret + i;
824                 int bytes;
825
826                 memcpy(pkgInfo, &info[i], sizeof(SecPkgInfoA));
827                 if (info[i].Name)
828                 {
829                     pkgInfo->Name = nextString;
830                     /* just repeat back to WideCharToMultiByte how many bytes
831                      * it requires, since we asked it earlier
832                      */
833                     bytes = WideCharToMultiByte(CP_ACP, 0, info[i].Name, -1,
834                      NULL, 0, NULL, NULL);
835                     WideCharToMultiByte(CP_ACP, 0, info[i].Name, -1,
836                      pkgInfo->Name, bytes, NULL, NULL);
837                     nextString += lstrlenA(nextString) + 1;
838                 }
839                 else
840                     pkgInfo->Name = NULL;
841                 if (info[i].Comment)
842                 {
843                     pkgInfo->Comment = nextString;
844                     /* just repeat back to WideCharToMultiByte how many bytes
845                      * it requires, since we asked it earlier
846                      */
847                     bytes = WideCharToMultiByte(CP_ACP, 0, info[i].Comment, -1,
848                      NULL, 0, NULL, NULL);
849                     WideCharToMultiByte(CP_ACP, 0, info[i].Comment, -1,
850                      pkgInfo->Comment, bytes, NULL, NULL);
851                     nextString += lstrlenA(nextString) + 1;
852                 }
853                 else
854                     pkgInfo->Comment = NULL;
855             }
856         }
857     }
858     else
859         ret = NULL;
860     return ret;
861 }
862
863 /***********************************************************************
864  *              EnumerateSecurityPackagesA (SECUR32.@)
865  */
866 SECURITY_STATUS WINAPI EnumerateSecurityPackagesA(PULONG pcPackages,
867  PSecPkgInfoA *ppPackageInfo)
868 {
869     SECURITY_STATUS ret;
870     PSecPkgInfoW info;
871
872     ret = EnumerateSecurityPackagesW(pcPackages, &info);
873     if (ret == SEC_E_OK && *pcPackages && info)
874     {
875         *ppPackageInfo = thunk_PSecPkgInfoWToA(*pcPackages, info);
876         if (*pcPackages && !*ppPackageInfo)
877         {
878             *pcPackages = 0;
879             ret = SEC_E_INSUFFICIENT_MEMORY;
880         }
881         FreeContextBuffer(info);
882     }
883     return ret;
884 }
885
886 /***********************************************************************
887  *              GetComputerObjectNameA (SECUR32.@)
888  *
889  * Get the local computer's name using the format specified.
890  *
891  * PARAMS
892  *  NameFormat   [I] The format for the name.
893  *  lpNameBuffer [O] Pointer to buffer to receive the name.
894  *  nSize        [I/O] Size in characters of buffer.
895  *
896  * RETURNS
897  *  TRUE  If the name was written to lpNameBuffer.
898  *  FALSE If the name couldn't be written.
899  *
900  * NOTES
901  *  If lpNameBuffer is NULL, then the size of the buffer needed to hold the
902  *  name will be returned in *nSize.
903  *
904  *  nSize returns the number of characters written when lpNameBuffer is not
905  *  NULL or the size of the buffer needed to hold the name when the buffer
906  *  is too short or lpNameBuffer is NULL.
907  * 
908  *  It the buffer is too short, ERROR_INSUFFICIENT_BUFFER error will be set.
909  */
910 BOOLEAN WINAPI GetComputerObjectNameA(
911   EXTENDED_NAME_FORMAT NameFormat, LPSTR lpNameBuffer, PULONG nSize)
912 {
913     BOOLEAN rc;
914     LPWSTR bufferW = NULL;
915     ULONG sizeW = *nSize;
916     TRACE("(%d %p %p)\n", NameFormat, lpNameBuffer, nSize);
917     if (lpNameBuffer) {
918         bufferW = HeapAlloc(GetProcessHeap(), 0, sizeW * sizeof(WCHAR));
919         if (bufferW == NULL) {
920             SetLastError(ERROR_NOT_ENOUGH_MEMORY);
921             return FALSE;
922         }
923     }
924     rc = GetComputerObjectNameW(NameFormat, bufferW, &sizeW);
925     if (rc && bufferW) {
926         ULONG len = WideCharToMultiByte(CP_ACP, 0, bufferW, -1, NULL, 0, NULL, NULL);
927         WideCharToMultiByte(CP_ACP, 0, bufferW, -1, lpNameBuffer, *nSize, NULL, NULL);
928         *nSize = len;
929     }
930     else
931         *nSize = sizeW;
932     HeapFree(GetProcessHeap(), 0, bufferW);
933     return rc;
934 }
935
936 /***********************************************************************
937  *              GetComputerObjectNameW (SECUR32.@)
938  */
939 BOOLEAN WINAPI GetComputerObjectNameW(
940   EXTENDED_NAME_FORMAT NameFormat, LPWSTR lpNameBuffer, PULONG nSize)
941 {
942     LSA_HANDLE policyHandle;
943     LSA_OBJECT_ATTRIBUTES objectAttributes;
944     PPOLICY_DNS_DOMAIN_INFO domainInfo;
945     NTSTATUS ntStatus;
946     BOOLEAN status;
947     TRACE("(%d %p %p)\n", NameFormat, lpNameBuffer, nSize);
948
949     if (NameFormat == NameUnknown)
950     {
951         SetLastError(ERROR_INVALID_PARAMETER);
952         return FALSE;
953     }
954
955     ZeroMemory(&objectAttributes, sizeof(objectAttributes));
956     objectAttributes.Length = sizeof(objectAttributes);
957
958     ntStatus = LsaOpenPolicy(NULL, &objectAttributes,
959                              POLICY_VIEW_LOCAL_INFORMATION,
960                              &policyHandle);
961     if (ntStatus != STATUS_SUCCESS)
962     {
963         SetLastError(LsaNtStatusToWinError(ntStatus));
964         WARN("LsaOpenPolicy failed with NT status %u\n", GetLastError());
965         return FALSE;
966     }
967
968     ntStatus = LsaQueryInformationPolicy(policyHandle,
969                                          PolicyDnsDomainInformation,
970                                          (PVOID *)&domainInfo);
971     if (ntStatus != STATUS_SUCCESS)
972     {
973         SetLastError(LsaNtStatusToWinError(ntStatus));
974         WARN("LsaQueryInformationPolicy failed with NT status %u\n",
975              GetLastError());
976         LsaClose(policyHandle);
977         return FALSE;
978     }
979
980     if (domainInfo->Sid)
981     {
982         switch (NameFormat)
983         {
984         case NameSamCompatible:
985             {
986                 WCHAR name[MAX_COMPUTERNAME_LENGTH + 1];
987                 DWORD size = sizeof(name)/sizeof(name[0]);
988                 if (GetComputerNameW(name, &size))
989                 {
990                     DWORD len = domainInfo->Name.Length + size + 3;
991                     if (lpNameBuffer)
992                     {
993                         if (*nSize < len)
994                         {
995                             *nSize = len;
996                             SetLastError(ERROR_INSUFFICIENT_BUFFER);
997                             status = FALSE;
998                         }
999                         else
1000                         {
1001                             WCHAR bs[] = { '\\', 0 };
1002                             WCHAR ds[] = { '$', 0 };
1003                             lstrcpyW(lpNameBuffer, domainInfo->Name.Buffer);
1004                             lstrcatW(lpNameBuffer, bs);
1005                             lstrcatW(lpNameBuffer, name);
1006                             lstrcatW(lpNameBuffer, ds);
1007                             status = TRUE;
1008                         }
1009                     }
1010                     else        /* just requesting length required */
1011                     {
1012                         *nSize = len;
1013                         status = TRUE;
1014                     }
1015                 }
1016                 else
1017                 {
1018                     SetLastError(ERROR_INTERNAL_ERROR);
1019                     status = FALSE;
1020                 }
1021             }
1022             break;
1023         case NameFullyQualifiedDN:
1024         case NameDisplay:
1025         case NameUniqueId:
1026         case NameCanonical:
1027         case NameUserPrincipal:
1028         case NameCanonicalEx:
1029         case NameServicePrincipal:
1030         case NameDnsDomain:
1031             FIXME("NameFormat %d not implemented\n", NameFormat);
1032             SetLastError(ERROR_CANT_ACCESS_DOMAIN_INFO);
1033             status = FALSE;
1034             break;
1035         default:
1036             SetLastError(ERROR_INVALID_PARAMETER);
1037             status = FALSE;
1038         }
1039     }
1040     else
1041     {
1042         SetLastError(ERROR_CANT_ACCESS_DOMAIN_INFO);
1043         status = FALSE;
1044     }
1045
1046     LsaFreeMemory(domainInfo);
1047     LsaClose(policyHandle);
1048
1049     return status;
1050 }
1051
1052 /***********************************************************************
1053  *              GetUserNameExA (SECUR32.@)
1054  */
1055 BOOLEAN WINAPI GetUserNameExA(
1056   EXTENDED_NAME_FORMAT NameFormat, LPSTR lpNameBuffer, PULONG nSize)
1057 {
1058     FIXME("%d %p %p\n", NameFormat, lpNameBuffer, nSize);
1059     return FALSE;
1060 }
1061
1062 BOOLEAN WINAPI GetUserNameExW(
1063   EXTENDED_NAME_FORMAT NameFormat, LPWSTR lpNameBuffer, PULONG nSize)
1064 {
1065     FIXME("%d %p %p\n", NameFormat, lpNameBuffer, nSize);
1066     return FALSE;
1067 }
1068
1069 BOOLEAN WINAPI TranslateNameA(
1070   LPCSTR lpAccountName, EXTENDED_NAME_FORMAT AccountNameFormat,
1071   EXTENDED_NAME_FORMAT DesiredNameFormat, LPSTR lpTranslatedName,
1072   PULONG nSize)
1073 {
1074     FIXME("%p %d %d %p %p\n", lpAccountName, AccountNameFormat,
1075           DesiredNameFormat, lpTranslatedName, nSize);
1076     return FALSE;
1077 }
1078
1079 BOOLEAN WINAPI TranslateNameW(
1080   LPCWSTR lpAccountName, EXTENDED_NAME_FORMAT AccountNameFormat,
1081   EXTENDED_NAME_FORMAT DesiredNameFormat, LPWSTR lpTranslatedName,
1082   PULONG nSize)
1083 {
1084     FIXME("%p %d %d %p %p\n", lpAccountName, AccountNameFormat,
1085           DesiredNameFormat, lpTranslatedName, nSize);
1086     return FALSE;
1087 }
1088
1089 /***********************************************************************
1090  *              DllMain (SECUR32.0)
1091  */
1092 BOOL WINAPI DllMain(HINSTANCE hinstDLL, DWORD fdwReason, LPVOID lpvReserved)
1093 {
1094     if (fdwReason == DLL_PROCESS_ATTACH)
1095     {
1096         DisableThreadLibraryCalls(hinstDLL);
1097         SECUR32_initializeProviders();
1098     }
1099     else if (fdwReason == DLL_PROCESS_DETACH)
1100     {
1101         SECUR32_freeProviders();
1102     }
1103
1104     return TRUE;
1105 }