Click here to Skip to main content
Click here to Skip to main content

Add a UAC shield to a button when elevation is required for admin tasks

By , 22 Apr 2007
Rate this:
Please Sign up or sign in to vote.
Screenshot - VistaSecurity.png

Introduction

Any application that does not always need administrator privileges should not run with them by default. However, when a user wants to perform a task that requires elevation, you need to show them that this is required by displaying the Vista shield icon. When this is clicked, your application will then need to restart with administrator privileges. Interested? Then read on...

Making the VistaSecurity Class

First we need to create a VistaSecurity class. Inside it we need the SendMessage API.

[DllImport("user32")]
public static extern UInt32 SendMessage
    (IntPtr hWnd, UInt32 msg, UInt32 wParam, UInt32 lParam);

internal const int BCM_FIRST = 0x1600; //Normal button
internal const int BCM_SETSHIELD = (BCM_FIRST + 0x000C); //Elevated button

First, before we add a shield, we need to know if the process is elevated or not which is easily done. To add a shield to a button, make sure that the button uses FlatStyle.System and then sends the appropriate message using the API. The important API function parameters are the handle of the button and the BCM_SETSHIELD message.

static internal bool IsAdmin()
{
    WindowsIdentity id = WindowsIdentity.GetCurrent();
    WindowsPrincipal p = new WindowsPrincipal(id);
    return p.IsInRole(WindowsBuiltInRole.Administrator);
}

static internal void AddShieldToButton(Button b)
{
    b.FlatStyle = FlatStyle.System;
    SendMessage(b.Handle, BCM_SETSHIELD, 0, 0xFFFFFFFF);
}

We then need a way to elevate the process if required. We just restart the process with the "runas" Verb. The current unelevated process is then exited unless the System.ComponentModel.Win32Exception is thrown, as this indicates that the user has clicked Cancel on the UAC prompt.

internal static void RestartElevated()
{
    ProcessStartInfo startInfo = new ProcessStartInfo();
    startInfo.UseShellExecute = true;
    startInfo.WorkingDirectory = Environment.CurrentDirectory;
    startInfo.FileName = Application.ExecutablePath;
    startInfo.Verb = "runas";
    try
    {
        Process p = Process.Start(startInfo);
    }
    catch(System.ComponentModel.Win32Exception ex)
    {
        return;
    }

    Application.Exit();
}

That's it for the VistaSecurity class.

Using the code

On the form, you need the following code in your constructor after InitializeComponent which will add the shield.

if (!VistaSecurity.IsAdmin())
{
    this.Text += " (Standard)"; //Unnecessary
    VistaSecurity.AddShieldToButton(buttonGetElevation); //Important
}
else
    this.Text += " (Elevated)";

When the button is clicked we need to check if you have permission to perform the required action or elevate.

if (VistaSecurity.IsAdmin())
{
    DoAdminTask();
}
else
{
    VistaSecurity.RestartElevated();
}

Well, that's how to display a shield and get elevation!

Points of Interest

For the admin task in this demo, I added a file VISTA.TXT to All Users' start menu at \ProgramData\Microsoft\Windows\Start Menu\. To get rid of the file, run the demo again (admin credentials may be required!). The Try Admin Task button just tries to create the file without asking for elevation.

History

  • 22 Apr 07 - Article written
  • 24 Apr 07 - Checking of Administrator account works for non-English versions of Windows by using WindowsBuiltInRole.Administrator instead of a non-localized string. In DoAdminTask(), MessageBoxes show what has been done, more error catching added, and an Environment.SpecialFolder is used to get the path.

License

This article has no explicit license attached to it but may contain usage terms in the article text or the download files themselves. If in doubt please contact the author via the discussion board below.

A list of licenses authors might use can be found here

About the Author

hackman3vilGuy
Software Developer (Senior)
United Kingdom United Kingdom
No Biography provided

Comments and Discussions

 
QuestionCant get the UAC icon to a button [modified] Pinmemberhanonymouss1-Mar-12 4:11 
QuestionLicense for this Component Pinmemberstreetsjingo30-Nov-11 10:45 
QuestionAwesome! + How wrong can you be? PinmemberVercas3-Oct-10 4:10 
GeneralCode PinmemberBrandon Holland12-Jul-09 3:21 
GeneralRestart application Pinmemberjammmie99924-May-09 22:55 
GeneralRe: Restart application PinmemberMcoreD30-Aug-09 15:34 
AnswerRe: Restart application PinmemberIndivara28-Jan-10 14:14 
Questionhow to remove it ? Pinmembervishal108221-May-09 16:46 
AnswerRe: how to remove it ? Pinmemberjammmie99924-May-09 22:54 
AnswerRe: how to remove it ? PinmemberSky Sanders27-Dec-09 7:23 
GeneralIsAdmin Function - an easier way PinmemberDmex877-Jan-09 7:51 
GeneralRe: IsAdmin Function - an easier way Pinmembervtpdawg7-Jan-10 8:11 
GeneralVb.net PinmemberJorge Rocha29-Dec-08 15:05 
GeneralRe: Vb.net PinmemberJumpyCODE13-Mar-09 20:55 
GeneralRe: Vb.net Pinmemberzerosoft_uk26-Jul-11 12:14 
GeneralExcellant... and i rolled it all into a handy control :D PinmemberFocusedWolf25-May-08 19:32 
GeneralRe: Excellant... and i rolled it all into a handy control :D PinmemberFocusedWolf25-May-08 19:35 
GeneralRe: Excellant... and i rolled it all into a handy control :D PinmemberFocusedWolf26-May-08 10:11 
GeneralRe: Excellant... and i rolled it all into a handy control :D PinmemberFocusedWolf30-May-08 9:40 
GeneralRe: Excellant... and i rolled it all into a handy control :D Pinmemberf r i s c h19-Aug-08 1:54 
RantRe: Excellant... and i rolled it all into a handy control :D Pinmember98z2825-Feb-10 8:48 
GeneralWont detect properly except for first user in Vista Pinmemberleifre11-Apr-08 8:08 
Question.NET 3? PinmemberJimShabadoo2-Nov-07 8:34 
AnswerRe: .NET 3? PinmemberJimShabadoo2-Nov-07 8:48 
GeneralRe: .NET 3? PinmemberKyferEz15-Oct-09 15:28 

General General    News News    Suggestion Suggestion    Question Question    Bug Bug    Answer Answer    Joke Joke    Rant Rant    Admin Admin   

Use Ctrl+Left/Right to switch messages, Ctrl+Up/Down to switch threads, Ctrl+Shift+Left/Right to switch pages.

| Advertise | Privacy | Mobile
Web04 | 2.8.140421.2 | Last Updated 22 Apr 2007
Article Copyright 2007 by hackman3vilGuy
Everything else Copyright © CodeProject, 1999-2014
Terms of Use
Layout: fixed | fluid