56 lines
1.6 KiB
Factor
56 lines
1.6 KiB
Factor
! Copyright (C) 2010 Doug Coleman.
|
|
! See http://factorcode.org/license.txt for BSD license.
|
|
USING: accessors alien alien.data alien.syntax classes.struct
|
|
continuations fry kernel libc literals locals sequences
|
|
windows.advapi32 windows.errors windows.kernel32 windows.types
|
|
alien.c-types ;
|
|
IN: windows.privileges
|
|
|
|
TYPEDEF: TOKEN_PRIVILEGES* PTOKEN_PRIVILEGES
|
|
|
|
! Security tokens
|
|
! http://msdn.microsoft.com/msdnmag/issues/05/03/TokenPrivileges/
|
|
|
|
: (open-process-token) ( handle -- handle )
|
|
flags{ TOKEN_ADJUST_PRIVILEGES TOKEN_QUERY }
|
|
{ PHANDLE }
|
|
[ OpenProcessToken win32-error=0/f ]
|
|
with-out-parameters ;
|
|
|
|
: open-process-token ( -- handle )
|
|
#! remember to CloseHandle
|
|
GetCurrentProcess (open-process-token) ;
|
|
|
|
: with-process-token ( quot -- )
|
|
#! quot: ( token-handle -- token-handle )
|
|
[ open-process-token ] dip
|
|
[ keep ] curry
|
|
[ CloseHandle drop ] [ ] cleanup ; inline
|
|
|
|
: lookup-privilege ( string -- luid )
|
|
[ f ] dip LUID <struct>
|
|
[ LookupPrivilegeValue win32-error=0/f ] keep ;
|
|
|
|
:: make-token-privileges ( name enabled? -- obj )
|
|
TOKEN_PRIVILEGES <struct>
|
|
1 >>PrivilegeCount
|
|
LUID_AND_ATTRIBUTES malloc-struct &free
|
|
enabled? [ SE_PRIVILEGE_ENABLED >>Attributes ] when
|
|
name lookup-privilege >>Luid
|
|
>>Privileges ;
|
|
|
|
: set-privilege ( name ? -- )
|
|
'[
|
|
0
|
|
_ _ make-token-privileges
|
|
dup byte-length
|
|
f
|
|
f
|
|
AdjustTokenPrivileges win32-error=0/f
|
|
] with-process-token ;
|
|
|
|
: with-privileges ( seq quot -- )
|
|
[ '[ _ [ t set-privilege ] each @ ] ]
|
|
[ drop '[ _ [ f set-privilege ] each ] ]
|
|
2bi [ ] cleanup ; inline
|