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;