2 * Win32 kernel functions
4 * Copyright 1995 Martin von Loewis and Cameron Heide
5 * Copyright 1999 Peter Ganten
6 * Copyright 2002 Martin Wilck
8 * This library is free software; you can redistribute it and/or
9 * modify it under the terms of the GNU Lesser General Public
10 * License as published by the Free Software Foundation; either
11 * version 2.1 of the License, or (at your option) any later version.
13 * This library is distributed in the hope that it will be useful,
14 * but WITHOUT ANY WARRANTY; without even the implied warranty of
15 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
16 * Lesser General Public License for more details.
18 * You should have received a copy of the GNU Lesser General Public
19 * License along with this library; if not, write to the Free Software
20 * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
24 #include "wine/port.h"
38 #define WIN32_NO_STATUS
44 #include "wine/unicode.h"
45 #include "wine/exception.h"
46 #include "wine/debug.h"
48 #include "kernel_private.h"
50 WINE_DEFAULT_DEBUG_CHANNEL(computername);
52 /* Registry key and value names */
53 static const WCHAR ComputerW[] = {'M','a','c','h','i','n','e','\\',
54 'S','y','s','t','e','m','\\',
55 'C','u','r','r','e','n','t','C','o','n','t','r','o','l','S','e','t','\\',
56 'C','o','n','t','r','o','l','\\',
57 'C','o','m','p','u','t','e','r','N','a','m','e',0};
58 static const WCHAR ActiveComputerNameW[] = {'A','c','t','i','v','e','C','o','m','p','u','t','e','r','N','a','m','e',0};
59 static const WCHAR ComputerNameW[] = {'C','o','m','p','u','t','e','r','N','a','m','e',0};
61 static const char default_ComputerName[] = "WINE";
63 #define IS_OPTION_TRUE(ch) ((ch) == 'y' || (ch) == 'Y' || (ch) == 't' || (ch) == 'T' || (ch) == '1')
65 /***********************************************************************
66 * dns_gethostbyname (INTERNAL)
69 * "The FQDN is the name gethostbyname(2) returns for the host name returned by gethostname(2)."
71 * Wine can use this technique only if the thread-safe gethostbyname_r is available.
73 #ifdef HAVE_LINUX_GETHOSTBYNAME_R_6
74 static BOOL dns_gethostbyname ( char *name, int *size )
76 struct hostent* host = NULL;
79 struct hostent hostentry;
80 int locerr = ENOBUFS, res = ENOMEM;
82 extrabuf = HeapAlloc( GetProcessHeap(), 0, ebufsize ) ;
86 res = gethostbyname_r ( name, &hostentry, extrabuf, ebufsize, &host, &locerr );
87 if( res != ERANGE ) break;
89 extrabuf = HeapReAlloc( GetProcessHeap(), 0, extrabuf, ebufsize ) ;
93 WARN ("Error in gethostbyname_r %d (%d)\n", res, locerr);
96 WARN ("gethostbyname_r returned NULL host, locerr = %d\n", locerr);
101 int len = strlen ( host->h_name );
104 strcpy ( name, host->h_name );
109 memcpy ( name, host->h_name, *size );
111 SetLastError ( ERROR_MORE_DATA );
116 HeapFree( GetProcessHeap(), 0, extrabuf );
120 # define dns_gethostbyname(name,size) 0
123 /***********************************************************************
124 * dns_fqdn (INTERNAL)
126 static BOOL dns_fqdn ( char *name, int *size )
128 if ( gethostname ( name, *size + 1 ) )
133 SetLastError ( ERROR_MORE_DATA );
135 SetLastError ( ERROR_INVALID_PARAMETER );
140 if ( !dns_gethostbyname ( name, size ) )
141 *size = strlen ( name );
146 /***********************************************************************
147 * dns_hostname (INTERNAL)
149 static BOOL dns_hostname ( char *name, int *size )
152 if ( ! dns_fqdn ( name, size ) ) return FALSE;
153 c = strchr ( name, '.' );
162 /***********************************************************************
163 * dns_domainname (INTERNAL)
165 static BOOL dns_domainname ( char *name, int *size )
168 if ( ! dns_fqdn ( name, size ) ) return FALSE;
169 c = strchr ( name, '.' );
174 memmove ( name, c, *size + 1 );
179 /***********************************************************************
180 * _init_attr (INTERNAL)
182 static inline void _init_attr ( OBJECT_ATTRIBUTES *attr, UNICODE_STRING *name )
184 attr->Length = sizeof (OBJECT_ATTRIBUTES);
185 attr->RootDirectory = 0;
186 attr->ObjectName = name;
187 attr->Attributes = 0;
188 attr->SecurityDescriptor = NULL;
189 attr->SecurityQualityOfService = NULL;
192 /***********************************************************************
195 static BOOL get_use_dns_option(void)
197 static const WCHAR NetworkW[] = {'S','o','f','t','w','a','r','e','\\',
198 'W','i','n','e','\\','N','e','t','w','o','r','k',0};
199 static const WCHAR UseDNSW[] = {'U','s','e','D','n','s','C','o','m','p','u','t','e','r','N','a','m','e',0};
204 OBJECT_ATTRIBUTES attr;
205 UNICODE_STRING nameW;
208 _init_attr( &attr, &nameW );
209 RtlOpenCurrentUser( KEY_READ, &root );
210 attr.RootDirectory = root;
211 RtlInitUnicodeString( &nameW, NetworkW );
213 /* @@ Wine registry key: HKCU\Software\Wine\Network */
214 if (!NtOpenKey( &hkey, KEY_READ, &attr ))
216 RtlInitUnicodeString( &nameW, UseDNSW );
217 if (!NtQueryValueKey( hkey, &nameW, KeyValuePartialInformation, tmp, sizeof(tmp), &dummy ))
219 WCHAR *str = (WCHAR *)((KEY_VALUE_PARTIAL_INFORMATION *)tmp)->Data;
220 ret = IS_OPTION_TRUE( str[0] );
229 /***********************************************************************
230 * COMPUTERNAME_Init (INTERNAL)
232 void COMPUTERNAME_Init (void)
234 HANDLE hkey = INVALID_HANDLE_VALUE, hsubkey = INVALID_HANDLE_VALUE;
235 OBJECT_ATTRIBUTES attr;
236 UNICODE_STRING nameW;
237 char buf[offsetof( KEY_VALUE_PARTIAL_INFORMATION, Data ) + (MAX_COMPUTERNAME_LENGTH + 1) * sizeof( WCHAR )];
238 DWORD len = sizeof( buf );
239 LPWSTR computer_name = (LPWSTR) (buf + offsetof( KEY_VALUE_PARTIAL_INFORMATION, Data ));
240 NTSTATUS st = STATUS_INTERNAL_ERROR;
243 _init_attr ( &attr, &nameW );
245 RtlInitUnicodeString( &nameW, ComputerW );
246 if ( ( st = NtCreateKey( &hkey, KEY_ALL_ACCESS, &attr, 0, NULL, 0, NULL ) ) != STATUS_SUCCESS )
249 attr.RootDirectory = hkey;
250 RtlInitUnicodeString( &nameW, ComputerNameW );
251 if ( (st = NtCreateKey( &hsubkey, KEY_ALL_ACCESS, &attr, 0, NULL, 0, NULL ) ) != STATUS_SUCCESS )
254 st = NtQueryValueKey( hsubkey, &nameW, KeyValuePartialInformation, buf, len, &len );
256 if ( st != STATUS_SUCCESS || get_use_dns_option() )
259 int hlen = sizeof (hbuf);
261 TRACE( "retrieving Unix host name\n" );
262 if ( gethostname ( hbuf, hlen ) )
264 strcpy ( hbuf, default_ComputerName );
265 WARN( "gethostname() error: %d, using host name %s\n", errno, hbuf );
267 hbuf[MAX_COMPUTERNAME_LENGTH] = 0;
268 dot = strchr ( hbuf, '.' );
270 hlen = strlen ( hbuf );
271 len = MultiByteToWideChar( CP_UNIXCP, 0, hbuf, hlen + 1, computer_name, MAX_COMPUTERNAME_LENGTH + 1 )
273 if ( NtSetValueKey( hsubkey, &nameW, 0, REG_SZ, computer_name, len ) != STATUS_SUCCESS )
274 WARN ( "failed to set ComputerName\n" );
278 len = (len - offsetof( KEY_VALUE_PARTIAL_INFORMATION, Data ));
279 TRACE( "found in registry\n" );
283 TRACE(" ComputerName: %s (%u)\n", debugstr_w (computer_name), len);
285 RtlInitUnicodeString( &nameW, ActiveComputerNameW );
286 if ( ( st = NtCreateKey( &hsubkey, KEY_ALL_ACCESS, &attr, 0, NULL, REG_OPTION_VOLATILE, NULL ) )
290 RtlInitUnicodeString( &nameW, ComputerNameW );
291 st = NtSetValueKey( hsubkey, &nameW, 0, REG_SZ, computer_name, len );
297 if ( st == STATUS_SUCCESS )
298 TRACE( "success\n" );
301 WARN( "status trying to set ComputerName: %x\n", st );
302 SetLastError ( RtlNtStatusToDosError ( st ) );
307 /***********************************************************************
308 * GetComputerNameW (KERNEL32.@)
310 BOOL WINAPI GetComputerNameW(LPWSTR name,LPDWORD size)
312 UNICODE_STRING nameW;
313 OBJECT_ATTRIBUTES attr;
314 HANDLE hkey = INVALID_HANDLE_VALUE, hsubkey = INVALID_HANDLE_VALUE;
315 char buf[offsetof( KEY_VALUE_PARTIAL_INFORMATION, Data ) + (MAX_COMPUTERNAME_LENGTH + 1) * sizeof( WCHAR )];
316 DWORD len = sizeof( buf );
317 LPWSTR theName = (LPWSTR) (buf + offsetof( KEY_VALUE_PARTIAL_INFORMATION, Data ));
318 NTSTATUS st = STATUS_INVALID_PARAMETER;
319 DWORD err = ERROR_SUCCESS;
321 TRACE ("%p %p\n", name, size);
323 _init_attr ( &attr, &nameW );
324 RtlInitUnicodeString( &nameW, ComputerW );
325 if ( ( st = NtOpenKey( &hkey, KEY_READ, &attr ) ) != STATUS_SUCCESS )
327 err = RtlNtStatusToDosError ( st );
331 attr.RootDirectory = hkey;
332 RtlInitUnicodeString( &nameW, ActiveComputerNameW );
333 if ( ( st = NtOpenKey( &hsubkey, KEY_READ, &attr ) ) != STATUS_SUCCESS )
335 err = RtlNtStatusToDosError ( st );
339 RtlInitUnicodeString( &nameW, ComputerNameW );
340 if ( ( st = NtQueryValueKey( hsubkey, &nameW, KeyValuePartialInformation, buf, len, &len ) )
343 err = RtlNtStatusToDosError ( st );
347 len = (len -offsetof( KEY_VALUE_PARTIAL_INFORMATION, Data )) / sizeof (WCHAR) - 1;
348 TRACE ("ComputerName is %s (length %u)\n", debugstr_w ( theName ), len);
350 if ( *size < len + 1 )
353 err = ERROR_BUFFER_OVERFLOW;
357 memcpy ( name, theName, len * sizeof (WCHAR) );
366 if ( err == ERROR_SUCCESS )
370 SetLastError ( err );
371 WARN ( "Status %u reading computer name from registry\n", st );
376 /***********************************************************************
377 * GetComputerNameA (KERNEL32.@)
379 BOOL WINAPI GetComputerNameA(LPSTR name, LPDWORD size)
381 WCHAR nameW[ MAX_COMPUTERNAME_LENGTH + 1 ];
382 DWORD sizeW = MAX_COMPUTERNAME_LENGTH + 1;
386 if ( !GetComputerNameW (nameW, &sizeW) ) return FALSE;
388 len = WideCharToMultiByte ( CP_ACP, 0, nameW, -1, NULL, 0, NULL, 0 );
389 /* for compatibility with Win9x */
395 SetLastError( ERROR_BUFFER_OVERFLOW );
400 WideCharToMultiByte ( CP_ACP, 0, nameW, -1, name, len, NULL, 0 );
407 SetLastError( ERROR_INVALID_PARAMETER );
415 /***********************************************************************
416 * GetComputerNameExA (KERNEL32.@)
418 BOOL WINAPI GetComputerNameExA(COMPUTER_NAME_FORMAT type, LPSTR name, LPDWORD size)
421 int len = sizeof(buf) - 1, ret;
422 TRACE("%d, %p, %p\n", type, name, size);
425 case ComputerNameNetBIOS:
426 case ComputerNamePhysicalNetBIOS:
427 ret = GetComputerNameA (name, size);
428 if (!ret && GetLastError() == ERROR_BUFFER_OVERFLOW)
429 SetLastError( ERROR_MORE_DATA );
432 case ComputerNameDnsHostname:
433 case ComputerNamePhysicalDnsHostname:
434 ret = dns_hostname (buf, &len);
436 case ComputerNameDnsDomain:
437 case ComputerNamePhysicalDnsDomain:
438 ret = dns_domainname (buf, &len);
440 case ComputerNameDnsFullyQualified:
441 case ComputerNamePhysicalDnsFullyQualified:
442 ret = dns_fqdn (buf, &len);
445 SetLastError (ERROR_INVALID_PARAMETER);
451 TRACE ("-> %s (%d)\n", debugstr_a (buf), len);
452 if ( *size < len + 1 )
455 SetLastError( ERROR_MORE_DATA );
460 memcpy( name, buf, len );
471 /***********************************************************************
472 * GetComputerNameExW (KERNEL32.@)
474 BOOL WINAPI GetComputerNameExW( COMPUTER_NAME_FORMAT type, LPWSTR name, LPDWORD size )
477 int len = sizeof(buf) - 1, ret;
479 TRACE("%d, %p, %p\n", type, name, size);
482 case ComputerNameNetBIOS:
483 case ComputerNamePhysicalNetBIOS:
484 ret = GetComputerNameW (name, size);
485 if (!ret && GetLastError() == ERROR_BUFFER_OVERFLOW)
486 SetLastError( ERROR_MORE_DATA );
488 case ComputerNameDnsHostname:
489 case ComputerNamePhysicalDnsHostname:
490 ret = dns_hostname (buf, &len);
492 case ComputerNameDnsDomain:
493 case ComputerNamePhysicalDnsDomain:
494 ret = dns_domainname (buf, &len);
496 case ComputerNameDnsFullyQualified:
497 case ComputerNamePhysicalDnsFullyQualified:
498 ret = dns_fqdn (buf, &len);
501 SetLastError (ERROR_INVALID_PARAMETER);
509 TRACE ("-> %s (%d)\n", debugstr_a (buf), len);
511 lenW = MultiByteToWideChar( CP_ACP, 0, buf, len, NULL, 0 );
512 if ( *size < lenW + 1 )
515 SetLastError( ERROR_MORE_DATA );
520 MultiByteToWideChar( CP_ACP, 0, buf, len, name, lenW );
530 /******************************************************************************
531 * netbios_char (INTERNAL)
533 static WCHAR netbios_char ( WCHAR wc )
535 static const WCHAR special[] = {'!','@','#','$','%','^','&','\'',')','(','-','_','{','}','~'};
536 static const WCHAR deflt = '_';
539 if ( isalnumW ( wc ) ) return wc;
540 for ( i = 0; i < sizeof (special) / sizeof (WCHAR); i++ )
541 if ( wc == special[i] ) return wc;
545 /******************************************************************************
546 * SetComputerNameW [KERNEL32.@]
548 * Set a new NetBIOS name for the local computer.
551 * lpComputerName [I] Address of new computer name
557 BOOL WINAPI SetComputerNameW( LPCWSTR lpComputerName )
559 UNICODE_STRING nameW;
560 OBJECT_ATTRIBUTES attr;
561 HANDLE hkey = INVALID_HANDLE_VALUE, hsubkey = INVALID_HANDLE_VALUE;
562 int plen = strlenW ( lpComputerName );
564 NTSTATUS st = STATUS_INTERNAL_ERROR;
566 if (get_use_dns_option())
568 /* This check isn't necessary, but may help debugging problems. */
569 WARN( "Disabled by Wine Configuration.\n" );
570 WARN( "Set \"UseDnsComputerName\" = \"N\" in HKCU\\Software\\Wine\\Network to enable.\n" );
571 SetLastError ( ERROR_ACCESS_DENIED );
575 TRACE( "%s\n", debugstr_w (lpComputerName) );
577 /* Check parameter */
578 if ( plen > MAX_COMPUTERNAME_LENGTH )
581 /* This is NT behaviour. Win 95/98 would coerce characters. */
582 for ( i = 0; i < plen; i++ )
584 WCHAR wc = lpComputerName[i];
585 if ( wc != netbios_char( wc ) )
589 _init_attr ( &attr, &nameW );
591 RtlInitUnicodeString (&nameW, ComputerW);
592 if ( ( st = NtOpenKey( &hkey, KEY_ALL_ACCESS, &attr ) ) != STATUS_SUCCESS )
594 attr.RootDirectory = hkey;
595 RtlInitUnicodeString( &nameW, ComputerNameW );
596 if ( ( st = NtOpenKey( &hsubkey, KEY_ALL_ACCESS, &attr ) ) != STATUS_SUCCESS )
598 if ( ( st = NtSetValueKey( hsubkey, &nameW, 0, REG_SZ, lpComputerName, ( plen + 1) * sizeof(WCHAR) ) )
606 if ( st == STATUS_SUCCESS )
608 TRACE( "ComputerName changed\n" );
614 SetLastError ( RtlNtStatusToDosError ( st ) );
615 WARN ( "status %u\n", st );
620 /******************************************************************************
621 * SetComputerNameA [KERNEL32.@]
623 * See SetComputerNameW.
625 BOOL WINAPI SetComputerNameA( LPCSTR lpComputerName )
628 DWORD len = MultiByteToWideChar( CP_ACP, 0, lpComputerName, -1, NULL, 0 );
629 LPWSTR nameW = HeapAlloc( GetProcessHeap(), 0, len * sizeof(WCHAR) );
631 MultiByteToWideChar( CP_ACP, 0, lpComputerName, -1, nameW, len );
632 ret = SetComputerNameW( nameW );
633 HeapFree( GetProcessHeap(), 0, nameW );
637 /******************************************************************************
638 * SetComputerNameExW [KERNEL32.@]
641 BOOL WINAPI SetComputerNameExW( COMPUTER_NAME_FORMAT type, LPCWSTR lpComputerName )
643 TRACE("%d, %s\n", type, debugstr_w (lpComputerName));
646 case ComputerNameNetBIOS:
647 case ComputerNamePhysicalNetBIOS:
648 return SetComputerNameW( lpComputerName );
650 SetLastError( ERROR_ACCESS_DENIED );
655 /******************************************************************************
656 * SetComputerNameExA [KERNEL32.@]
659 BOOL WINAPI SetComputerNameExA( COMPUTER_NAME_FORMAT type, LPCSTR lpComputerName )
661 TRACE( "%d, %s\n", type, debugstr_a (lpComputerName) );
664 case ComputerNameNetBIOS:
665 case ComputerNamePhysicalNetBIOS:
666 return SetComputerNameA( lpComputerName );
668 SetLastError( ERROR_ACCESS_DENIED );
673 /***********************************************************************
674 * DnsHostnameToComputerNameA (KERNEL32.@)
676 BOOL WINAPI DnsHostnameToComputerNameA(LPCSTR hostname,
677 LPSTR computername, LPDWORD size)
681 FIXME("(%s, %p, %p): stub\n", debugstr_a(hostname), computername, size);
683 if (!hostname || !size) return FALSE;
684 len = lstrlenA(hostname);
686 if (len > MAX_COMPUTERNAME_LENGTH)
687 len = MAX_COMPUTERNAME_LENGTH;
694 if (!computername) return FALSE;
696 memcpy( computername, hostname, len );
697 computername[len + 1] = 0;
701 /***********************************************************************
702 * DnsHostnameToComputerNameW (KERNEL32.@)
704 BOOL WINAPI DnsHostnameToComputerNameW(LPCWSTR hostname,
705 LPWSTR computername, LPDWORD size)
709 FIXME("(%s, %p, %p): stub\n", debugstr_w(hostname), computername, size);
711 if (!hostname || !size) return FALSE;
712 len = lstrlenW(hostname);
714 if (len > MAX_COMPUTERNAME_LENGTH)
715 len = MAX_COMPUTERNAME_LENGTH;
722 if (!computername) return FALSE;
724 memcpy( computername, hostname, len * sizeof(WCHAR) );
725 computername[len + 1] = 0;