summaryrefslogtreecommitdiff
path: root/support/dktools/dk4isadm.ctr
blob: bdd757376353a18676536d6f47c6aecb97a119c8 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
%%	options

copyright owner	=	Dirk Krause
copyright year	=	2015-xxxx
license		=	bsd


%%	header

/**	@file	dk4isadm.h	Check whether process runs with
	admin privileges.
*/

#ifdef __cplusplus
extern "C" {
#endif

/**	Check whether the current process runs with administrative privileges.
	@return	1 if administrative, 0 if user process, -1 if failed to check.
*/
int
dk4isadmin(void);

#ifdef __cplusplus
}
#endif

%%	module


#include "dk4conf.h"

#if DK4_ON_WINDOWS
#ifndef WINDOWS_H_INCLUDED
#include <windows.h>
#define WINDOWS_H_INCLUDED 1
#endif
#endif

#include "dk4types.h"

#if DK4_HAVE_UNISTD_H
#ifndef UNISTD_H_INCLUDED
#include <unistd.h>
#define UNISTD_H_INCLUDED 1
#endif
#endif

#if DK4_HAVE_SYS_TYPES_H
#ifndef SYS_TYPES_H_INCLUDED
#include <sys/types.h>
#define SYS_TYPES_H_INCLUDED 1
#endif
#endif

int
dk4isadmin(void)
{
#if DK4_ON_WINDOWS
  PSID				pAdmGroup	=	NULL;
  SID_IDENTIFIER_AUTHORITY	ntAuthority	=	SECURITY_NT_AUTHORITY;
  BOOL				bres		=	FALSE;
  BOOL				isadm		=	FALSE;
#endif
  int		 back = 0;
#if DK4_ON_WINDOWS
  back = -1;
  bres = AllocateAndInitializeSid(
    &ntAuthority, 2,
    SECURITY_BUILTIN_DOMAIN_RID,	/* sub-authority 0 */
    DOMAIN_ALIAS_RID_ADMINS,		/* sub-authority 1 */
    (DWORD)0,				/* sub-authority 2 */
    (DWORD)0,				/* sub-authority 3 */
    (DWORD)0,				/* sub-authority 4 */
    (DWORD)0,				/* sub-authority 5 */
    (DWORD)0,				/* sub-authority 6 */
    (DWORD)0,				/* sub-authority 7 */
    &pAdmGroup
  );
  if (bres) {
    if (CheckTokenMembership(NULL, pAdmGroup, &isadm)) {
      if (isadm) {
        back = 1;
      } else {
        back = 0;
      }
    }
  }
  if (NULL != pAdmGroup) {
    FreeSid(pAdmGroup);
  }
#else
  if (0 == geteuid()) {
    back = 1;
  }
#endif
  return back;
}