Aaron的测试生活小说

半两五钱,笃志向前
  首页  :: 新随笔  :: 联系 :: 管理

Run As(运行方式) 的几种代码实现方式(Python和C#)

Posted on 2009-05-26 00:13  Aaron Wu  阅读(5701)  评论(9编辑  收藏  举报

1,引入

     最近因为项目的原因,需要在自动化测试代码中实现用户角色的切换,自然第一印象就想到了Run As(中文版windows中也叫“运行方式”)。比如我们可以在打开IE浏览器的时候右键单击“运行方式”:

Run AS

然后输入一个新的用户帐户,如下图:

 Run As

上图中我本机使用的是域帐户本地帐户登陆的操作系统,但是我现在需要使用域帐户来运行IE浏览器,以方便我在打开相应的绑定域帐户权限的页面,如公司内部站点。这样就不用在打开相应的页面时候输入域帐户名了。

  有些时候我们也可以应用相同的方式来使用管理员帐户启动相应的应用程序:

Run As

在较近操作系统中,右键菜单中没有了“Run as”,而是变成了“Run As Administrator”,目的是提高运行相应程序的用户权限:

Run AS Admin

  好了,言归正传,今天笔者提到的则是使用代码来实现相同的功能,即以更改运行程序的用户帐户。

2,代码实现命令行Run As

  命令行

     用命令行实现上述的Run as功能大致可以这样写:

      runas /user:user@domain.com "C:\Program Files\Internet Explorer\iexplore.exe"

      Python实现

  RunAsWithinPython.py文件内容如下:

Runas Commandline in Python
import os
os.system(
"runas /user:user@domain.com \"C:\Program Files\Internet Explorer\iexplore.exe\"")

     额,就这两行,看起来比命令行内容多不了几个字符,不得不惊诧Python的强悍。另外,python使用缩进来表示代码层次而不是我们在C#等语言中用的大括号“{”,代码看起来很爽,当然,这是闲话了。

  C#实现 

      Program.cs文件内容如下:

Runas Commandline in C#
using System.Diagnostics;

namespace MainTest
{
    
class Program
    {        
        
public static void Main(string[] args)
        {
            System.Diagnostics.Process cmd 
= System.Diagnostics.Process.Start("runas"
                
"/user:user@domain.com  \"C:\\Program Files\\Internet Explorer\\iexplore.exe\"");
        }
    }
}

   恩,C#表现也不错,看来确实是一个比较简单的命令而已,如果你没有运行一下这两个程序,你可能以为这样就搞定了,相当简单。但是如果你是一个细心的人,或者你是用过runas命令行的话,你应该知道,这里有问题:

      很可惜的是,这两种方式存在一个致命的缺陷:只能在第一个命令行中输入用户名,然后才会由系统提示输入密码,这样就不能在一条命令行中将用户名和密码完全包含。这样我们只能使用代码实现一半,另外一半则仍然需要手动输入(密码)——显然这就造成了一个半拉子工程。我们也可以查看一下runas帮助说明:

Runas Helper

  这样设计的目的在于防止像我们这种弱弱的想要使用命令行来提高运行程序的权限的人——至少我们是不能使用批处理文件来干这件事了,这样更安全了,不是吗?好吧,接下来,我们将使用另外一种方法,这种方法则不会出现上述半拉子工程的尴尬情况。

3, 使用Process的ProcessStartInfo实现

ProcessUserLogonHelper
using System;
using System.Collections.Generic;
using System.Linq;
using System.Text;
using System.Diagnostics;

namespace MainTest
{
    
class ProcessUserLogonHelper
    {
        
/// <summary>
        
/// 在Main函数中调用该方法的的示例代码
        
/// </summary>
        private void InvokeExample()
        {
            ProcessUserLogonHelper phelper 
= new ProcessUserLogonHelper();
            Process pro 
= phelper.CreateProcessWithUserToken(@"C:\Program Files\Internet Explorer\iexplore.exe""bestreme.com""chdwu""************");
            pro.Start();
        }

        /// <summary>
        /// 使用指定帐户绑定进程
        /// </summary>
        /// <param name="appPath">进程的宿主程序</param>
        /// <param name="domain">指定帐户的域名</param>
        /// <param name="userName">指定帐户的用户名</param>
        /// <param name="password">指定帐户的密码</param>
        /// <returns>绑定了特定帐户的进程</returns>
        public Process CreateProcessWithUserToken(string appPath,string domain, string userName, string password)
        {
            Process pro = new Process();
            ProcessStartInfo processInfo = new ProcessStartInfo(appPath);
            processInfo.UseShellExecute = false;
            processInfo.UserName = userName;
            processInfo.Domain = domain;
            System.Security.SecureString psw = new System.Security.SecureString();
            foreach (char c in password.ToCharArray())
            {
                psw.AppendChar(c);
            }
            processInfo.Password = psw;
            processInfo.WorkingDirectory = System.IO.Path.GetDirectoryName(appPath);
            pro.StartInfo = processInfo;
            return pro;
        }

        /// <summary>
        /// 使用指定帐户绑定进程
        /// </summary>
        /// <param name="appPath">进程的宿主程序</param>
        /// <param name="userName">指定本地帐户的用户名</param>
        /// <param name="password">指定帐户的密码</param>
        /// <returns>绑定了特定帐户的进程</returns>
        public Process CreateProcessWithUserToken(string appPath, string userName, string password)
        {
            return CreateProcessWithUserToken(appPath, "", userName, password);
        }
    }
}

    上面的代码有几个地方需要指出的是:

    1, 应该待运行的程序指定相应的启动路径,否则的话在上述代码运行的过程中可能提示“目录不存在”;

    2,Domain可以不设置,而直接使用bestreme\chdwu的方式给UserName赋值即可;

     3,UseShellExecute 应该设置为False

     4,ProcessStartInfo中的password是System.Security.SecureString类型的,所以如果我们只是传入了一个string作为password,我们应该将其转换为正确的类型。

结束语

  上面为大家介绍了几种Run as的实现方式,其中的前面两种分别用Python和C#实现了命令行Run as的功能,第三种方法则是在ProcessStartInfo中加入了用户信息以实现用指定用户启动运行应用程序。

     当然实现Run as的代码实现方法很多,在下一篇中将介绍一下有关使用Win32 API的实现。需要指出的是,使用Win32 API几种实现方式存在着一些潜在的问题,详细内容将尽快在下一篇文章中给大家展示。