1 #Region "Microsoft.VisualBasic::7e3b7e103b4668634dc99b2fde490ce0, Microsoft.VisualBasic.Core\ApplicationServices\Tools\WinForm\VistaSecurity.vb"
2
3     ' Author:
4     
5     '       asuka (amethyst.asuka@gcmodeller.org)
6     '       xie (genetics@smrucc.org)
7     '       xieguigang (xie.guigang@live.com)
8     
9     ' Copyright (c) 2018 GPL3 Licensed
10     
11     
12     ' GNU GENERAL PUBLIC LICENSE (GPL3)
13     
14     
15     ' This program is free software: you can redistribute it and/or modify
16     ' it under the terms of the GNU General Public License as published by
17     ' the Free Software Foundation, either version 3 of the License, or
18     ' (at your option) any later version.
19     
20     ' This program is distributed in the hope that it will be useful,
21     ' but WITHOUT ANY WARRANTY; without even the implied warranty of
22     ' MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
23     ' GNU General Public License for more details.
24     
25     ' You should have received a copy of the GNU General Public License
26     ' along with this program. If not, see <http://www.gnu.org/licenses/>.
27
28
29
30     ' /********************************************************************************/
31
32     ' Summaries:
33
34     '     Module VistaSecurity
35     
36     '         FunctionIsAdmin, IsVistaOrHigher, SendMessage
37     
38     '         Sub: AddShieldToButton, RestartElevated
39     
40     
41     ' /********************************************************************************/
42
43 #End Region
44
45 Imports System.ComponentModel
46 Imports System.Runtime.InteropServices
47 Imports System.Security.Principal
48
49 Namespace ApplicationServices.Windows.Forms
50
51     Public Module VistaSecurity
52
53         <DllImport("user32")>
54         Public Function SendMessage(hWnd As IntPtr, msg As UInt32, wParam As UInt32, lParam As UInt32) As UInt32
55         End Function
56
57         Friend Const BCM_FIRST As Integer = &H1600
58         Friend Const BCM_SETSHIELD As Integer = (BCM_FIRST + &HC)
59
60         Public Function IsVistaOrHigher() As Boolean
61             Return Environment.OSVersion.Version.Major < 6
62         End Function
63
64         ''' <summary>
65         ''' Checks if the process is elevated
66         ''' </summary>
67         ''' <returns>If is elevated</returns>
68         Public Function IsAdmin() As Boolean
69             Dim id As WindowsIdentity = WindowsIdentity.GetCurrent()
70             Dim p As New WindowsPrincipal(id)
71             Return p.IsInRole(WindowsBuiltInRole.Administrator)
72         End Function
73
74         ''' <summary>
75         ''' Add a shield icon to a button
76         ''' </summary>
77         ''' <param name="b">The button</param>
78         Public Sub AddShieldToButton(b As Button)
79             b.FlatStyle = FlatStyle.System
80             SendMessage(b.Handle, BCM_SETSHIELD, 0, &HFFFFFFFFUI)
81         End Sub
82
83         ''' <summary>
84         ''' Restart the current process with administrator credentials.(以管理员的身份重启本应用程序)
85         ''' </summary>
86         Public Sub RestartElevated(Optional args$ = "")
87             Dim startInfo As New ProcessStartInfo()
88             startInfo.UseShellExecute = True
89             startInfo.WorkingDirectory = Environment.CurrentDirectory
90             startInfo.FileName = Application.ExecutablePath
91             startInfo.Arguments = args
92             startInfo.Verb = "runas"
93             Try
94                 Dim p As Process = Process.Start(startInfo)
95             Catch ex As Win32Exception
96                 'If cancelled, do nothing
97                 Return
98             End Try
99
100             Call App.Exit()
101         End Sub
102     End Module
103 End Namespace