技术改变世界!学习改变自己!

每天进步一点点,遥不可及的事情,在你一点一滴的努力下,也会变成现实。

  博客园 :: 首页 :: 博问 :: 闪存 :: 新随笔 :: 联系 :: 订阅 订阅 :: 管理 ::
  55 随笔 :: 0 文章 :: 33 评论 :: 0 引用

公告

通过WMI实现了远程创建共享目录,远程拷贝文件等操作:

 代码下载

using System;
using System.Collections.Generic;
using System.Text;
using System.Management;
using System.IO;
using System.Diagnostics;
using System.Threading;
using Log;
using System.Windows.Forms;

namespace BankSDS
{
    
/// <summary>
    
/// 注册表主键枚举
    
/// </summary>
    public enum RootKey : uint
    {
        HKEY_CLASSES_ROOT 
= 0x80000000,
        HKEY_CURRENT_USER 
= 0x80000001,
        HKEY_LOCAL_MACHINE 
= 0x80000002,
        HKEY_USERS 
= 0x80000003,
        HKEY_CURRENT_CONFIG 
= 0x80000005
    }

    
/// <summary>
    
/// 端机信息结构
    
/// </summary>
    public struct ServerInfo
    {
        
public string ServerID;
        
public string ServerSec;
        
public string ServerIP;
        
public string ServerUser;
        
public string ServerPw;
        
/// <summary>
        
/// 远程推送的目标目录
        
/// </summary>
        public string FolderPath;
        
/// <summary>
        
/// 远程共享名?
        
/// </summary>
        public string ShareName;
        
/// <summary>
        
/// 远程推送的说明
        
/// </summary>
        public string Description;
        
/// <summary>
        
/// 远程推送的文件集合
        
/// </summary>
        public string [] strSource;
        
/// <summary>
        
/// 推送成功后,要设置的注册表自动启动项名称
        
/// </summary>
        public string strTarget;
        
/// <summary>
        
/// 传送类型
        
/// </summary>
        public int transType;
        
public bool Result;
    }

    
/// <summary>
    
/// 用于WMI连接的用户信息结构
    
/// </summary>
    public struct UserInfo
    {
        
public string UserName;
        
public string UserPW;
        
public bool IsEnable;
    }

    
/// <summary>
    
/// WMI基本封装
    
/// </summary>
    public class WmiShareFunction
    {
        
public static CLog Wmilog = new CLog(Application.StartupPath + @"\\AppLogs\\Wmi.log");

        
/// <summary>
        
/// 创建远程目录
        
/// </summary>
        
/// <param name="scope">ManagementScope Instance</param>
        
/// <param name="DirectoryName">要建立的目录名</param>
        
/// <param name="CurrentDirectory">根目录所在位置,缺省为C:\</param>
        
/// <returns>0表示建立成功,-1表示建立失败</returns>
        public static int CreateRemoteDirectory(ManagementScope scope, string DirectoryName, string CurrentDirectory)
        {
            
try
            {
                
string currentDirectory = CurrentDirectory;
                
if (String.IsNullOrEmpty(CurrentDirectory))
                {
                    currentDirectory 
= @"C:\";
                }
                ObjectGetOptions objectGetOptions 
= new ObjectGetOptions(null, System.TimeSpan.MaxValue, true);
                ManagementPath managementPath 
= new ManagementPath("Win32_Process");

                ManagementClass processBatch 
= new ManagementClass(scope, managementPath, objectGetOptions);
                ManagementBaseObject inParams 
= processBatch.GetMethodParameters("Create");

                inParams[
"CommandLine"= @"cmd /CMd " + DirectoryName;
                inParams[
"CurrentDirectory"= currentDirectory;

                ManagementBaseObject outParams 
= null;
                outParams 
= processBatch.InvokeMethod("Create", inParams, null);

                Wmilog.WriteToLog(
"Excute CreateRemoteDirectory:" + outParams.ToString());
            }
            
catch (Exception ex)
            {
                Wmilog.WriteToLog(
"in CreateRemoteDirectory Exception:" + ex.Message);
                
return -1;
            }
            Wmilog.WriteToLog(
"建远程目录成功!");
            
return 0;
        }

        
/// <summary>
        
/// 在远程目标机器上创建一个注册表主键
        
/// </summary>
        
/// <param name="connectionScope">ManagementScope</param>
        
/// <param name="machineName">目标机器IP</param>
        
/// <param name="BaseKey">注册表分支名</param>
        
/// <param name="key">主键名称</param>
        
/// <returns>创建成功则返回0</returns>
        public static int CreateRemoteKey(ManagementScope connectionScope,
                                  
string machineName,
                                  RootKey BaseKey,
                                  
string key)
        {
            
try
            {
                ObjectGetOptions objectGetOptions 
= new ObjectGetOptions(null, System.TimeSpan.MaxValue, true);
                connectionScope.Path 
= new ManagementPath(@"\\" + machineName + @"\root\DEFAULT:StdRegProv");
                connectionScope.Connect();
                ManagementClass registryTask 
= new ManagementClass(connectionScope,
                               
new ManagementPath(@"DEFAULT:StdRegProv"), objectGetOptions);
                ManagementBaseObject methodParams 
= registryTask.GetMethodParameters("CreateKey");

                methodParams[
"hDefKey"= BaseKey;
                methodParams[
"sSubKeyName"= key;

                ManagementBaseObject exitCode 
= registryTask.InvokeMethod("CreateKey",
                                                                      methodParams, 
null);

                Wmilog.WriteToLog(
"in CreateRemoteKey:" + exitCode.ToString());
            }
            
catch (ManagementException e)
            {
                Wmilog.WriteToLog(
"in CreateRemoteKey(ManagementException):" + e.Message);
                
return -1;
            }
            Wmilog.WriteToLog(
"注册表主键创建成功!");
            
return 0;
        }

        
/// <summary>
        
/// 在远程目标机器上创建一个注册表键值
        
/// </summary>
        
/// <param name="connectionScope">ManagementScope</param>
        
/// <param name="machineName">目标机器IP</param>
        
/// <param name="BaseKey">注册表分支名</param>
        
/// <param name="key">主键名称</param>
        
/// <param name="valueName">键值名称</param>
        
/// <param name="value">键值</param>
        
/// <returns>创建成功则返回0</returns>
        public static int CreateRemoteValue(ManagementScope connectionScope,
                                    
string machineName,
                                    RootKey BaseKey,
                                    
string key,
                                    
string valueName,
                                    
string value)
        {
            
try
            {
                ObjectGetOptions objectGetOptions 
= new ObjectGetOptions(null, System.TimeSpan.MaxValue, true);
                connectionScope.Path 
= new ManagementPath(@"\\" + machineName + @"\root\DEFAULT:StdRegProv");
                connectionScope.Connect();
                ManagementClass registryTask 
= new ManagementClass(connectionScope,
                               
new ManagementPath(@"DEFAULT:StdRegProv"), objectGetOptions);
                ManagementBaseObject methodParams 
= registryTask.GetMethodParameters("SetStringValue");

                methodParams[
"hDefKey"= BaseKey;
                methodParams[
"sSubKeyName"= key;
                methodParams[
"sValue"= value;
                methodParams[
"sValueName"= valueName;

                ManagementBaseObject exitCode 
= registryTask.InvokeMethod("SetStringValue",
                                                                         methodParams, 
null);

                Wmilog.WriteToLog(
"in CreateRemoteValue:" + exitCode.ToString());
            }
            
catch (ManagementException e)
            {
                Wmilog.WriteToLog(
"in CreateRemoteValue(ManagementException):" + e.Message);
                
return -1;
            }
            Wmilog.WriteToLog(
"注册表键值写入成功!");
            
return 0;
        }

        
/// <summary>
        
/// 执行远程文件/目录推送
        
/// </summary>
        
/// <param name="Server">目标地址</param>
        
/// <param name="Username">目标设备登录用户</param>
        
/// <param name="Password">目标设备登录PWD</param>
        
/// <param name="FolderPath">远程目标目录</param>
        
/// <param name="ShareName">网络共享名</param>
        
/// <param name="Description">网络共享描述</param>
        
/// <param name="reginfo">推送成功后,可以设置的注册表项名称</param>
        
/// <param name="strSources">要推送的源文件信息</param>
        
/// <param name="iType">推送类型;0表示推送文件,1表示推送目录</param>
        
/// <returns>推送成功则返回true</returns>
        public static bool WmiFileCopyToRemote(string Server,
                      
string Username,
                      
string Password,
                      
string FolderPath,
                      
string ShareName,
                      
string Description,
                      
string reginfo,
                      
string[] strSources,
                      
int iType
                     )
        {
            
#region 远程共享目录组合

            
//远程共享名
            StringBuilder TargetDir = new StringBuilder();
            TargetDir.Append(
@"\\");
            TargetDir.Append(Server);
            TargetDir.Append(
"\\");
            TargetDir.Append(ShareName);
            TargetDir.Append(
"\\");

            
#endregion

            
#region 注册表值组合

            StringBuilder KeyValue 
= new StringBuilder();
            KeyValue.Append(FolderPath);
            KeyValue.Append(
"\\");

            
#endregion

            
#region WMI连接设置

            ConnectionOptions options 
= new ConnectionOptions();
            options.Username 
= Username;
            options.Password 
= Password;

            
//options.Authority = "ntdlmdomain:DOMAIN";
            options.Impersonation = ImpersonationLevel.Impersonate;
            options.EnablePrivileges 
= true;
            
//options.Authentication = AuthenticationLevel.Connect;
            
//options.Authority
            ManagementScope scope = new ManagementScope(
                    
"\\\\" + Server + "\\root\\cimv2",
                    options);
            
#endregion

            
try
            {
                Wmilog.WriteToLog(
"开始WMI连接...... ");
                scope.Connect();
                
if (scope.IsConnected)
                {
                    Wmilog.WriteToLog(
"WMI连接成功......");

                    Wmilog.WriteToLog(
"建远程目录开始......");
                    CreateRemoteDirectory(scope, FolderPath, 
"");
                    Wmilog.WriteToLog(
"建远程目录结束......");

                    Wmilog.WriteToLog(
"建远程共享目录开始......");
                    CreateShareNetFolder(scope, FolderPath, ShareName, Description);
                    Wmilog.WriteToLog(
"建远程共享目录结束......");

                    Wmilog.WriteToLog(
"连接远程共享开始......");
                    CreateShareNetConnect(Server, ShareName, Username, Password);
                    Wmilog.WriteToLog(
"连接远程共享结束......");

                    
//通过远程共享名,向远程共享目录中拷贝文件
                    if (iType == 0)
                    {
                        
//文件拷贝
                        Wmilog.WriteToLog("in CopyFiles:" + "Start".ToString());
                        System.Diagnostics.Debug.WriteLine(
"in CopyFiles:" + "Start".ToString());
                        
foreach (string filename in strSources)
                        {
                            
if (!String.IsNullOrEmpty(filename))
                            {
                                FileInfo fileinfo 
= new FileInfo(filename);
                                File.Copy(filename, TargetDir 
+ fileinfo.Name, true);
                                System.Diagnostics.Debug.WriteLine(filename 
+ " " + TargetDir + fileinfo.Name);

                            }
                        }
                        KeyValue.Append(reginfo);
                        Wmilog.WriteToLog(
"in CopyFiles:" + "End".ToString());
                        System.Diagnostics.Debug.WriteLine(
"in CopyFiles:" + "End".ToString());
                    }
                    
else if (iType == 1)
                    {
                        
//目录拷贝
                        DirectoryInfo DirInfo = new DirectoryInfo(strSources[0]);
                        KeyValue.Append(DirInfo.Name);
                        KeyValue.Append(
"\\");
                        KeyValue.Append(reginfo);
                        Wmilog.WriteToLog(
"in CopyDirectory:" + "Start".ToString());
                        System.Diagnostics.Debug.WriteLine(
"in CopyDirectory:" + "Start".ToString());
                        CopyDirectory(strSources[
0], TargetDir + DirInfo.Name);
                        Wmilog.WriteToLog(
"in CopyDirectory:" + "End".ToString());
                        System.Diagnostics.Debug.WriteLine(
"in CopyDirectory:" + "End".ToString());
                    }

                    Wmilog.WriteToLog(
"断开远程共享开始");
                    RemoveShareNetConnect(Server, ShareName, Username, Password);
                    Wmilog.WriteToLog(
"断开远程共享结束");

                    Wmilog.WriteToLog(
"删除远程共享目录开始");
                    RemoveShareNetFolder(scope, ShareName);
                    Wmilog.WriteToLog(
"删除远程共享目录结束");

                    
if (!String.IsNullOrEmpty(reginfo))
                    {
                        Wmilog.WriteToLog(
"查找或建立注册表主键开始");
                        CreateRemoteKey(scope, Server, RootKey.HKEY_LOCAL_MACHINE, 
@"Software\Microsoft\Windows\CurrentVersion\Run");
                        Wmilog.WriteToLog(
"查找或建立注册表主键结束");

                        Wmilog.WriteToLog(
"设置注册表值开始");
                        CreateRemoteValue(scope, Server, RootKey.HKEY_LOCAL_MACHINE, 
@"Software\Microsoft\Windows\CurrentVersion\Run", reginfo, KeyValue.ToString());
                        Wmilog.WriteToLog(
"设置注册表值结");
                    }
                }
            }
            
catch (Exception e)
            {
                Wmilog.WriteToLog(
"WMI连接失败(Exception): " + e.Message);
                System.Diagnostics.Debug.WriteLine(
"in WmiFileCopyToRemote Exception:" + e.Message);
                
return false;
            }
            Wmilog.WriteToLog(
"WMI操作成功结束 ");

            
return true;
        }

        
/// <summary>
        
/// 创建远程共享目录
        
/// </summary>
        
/// <param name="scope">ManagementScope</param>
        
/// <param name="FolderPath">要共享的目录路径</param>
        
/// <param name="ShareName">共享名</param>
        
/// <param name="Description">网络共享文件夹的描述</param>
        
/// <returns>0表示创建成功,-1表示创建失败</returns>
        public static int CreateShareNetFolder(ManagementScope scope, string FolderPath, string ShareName, string Description)
        {
            
try
            {
                
//创建一个ManagementClass对像
                ManagementClass managementClass = new ManagementClass(scope, new ManagementPath("Win32_Share"), null);

                
//创建ManagementBaseObject的输入和输出参数
                ManagementBaseObject inParams = managementClass.GetMethodParameters("Create");
                ManagementBaseObject outParams;

                
//设置输入参数
                inParams["Description"= Description;
                inParams[
"Name"= ShareName;
                inParams[
"Path"= FolderPath;
                inParams[
"Type"= 0x0;//DISK_DRIVE

                
//其它类型
                
//DISK_DRIVE=0x0
                
//PRINT_QUEUE=0x1
                
//DEVICE = 0x2
                
//IPC = 0x3
                
//DISK_DRIVE_ADMIN = 0x80000000
                
//PRINT_QUEUE_ADMIN=0x80000001
                
//DEVICE_ADMIN=0x80000002
                
//IPC_ADMIN=0x80000003
                
//inParams["MaximunAllowed"] = intMaxConnectionsNum;

                
//调用方法ManagementClass对像
                outParams = managementClass.InvokeMethod("Create", inParams, null);

                Wmilog.WriteToLog(
"Excute ShareNetFolder Result:" + outParams.ToString());
                
//检测方法是否调用成功
                if ((uint)(outParams.Properties["RetrunValue"].Value) != 0)
                {
                    
throw new Exception("Unable to create net share directiory!");
                }
            }
            
catch (Exception ex)
            {
                Wmilog.WriteToLog(
"创建远程共享目录失败:" + ex.Message);
                
return -1;
            }
            Wmilog.WriteToLog(
"创建远程共享目录成功!");
            
return 0;
        }

        
/// <summary>
        
/// 移除远程共享目录
        
/// </summary>
        
/// <param name="scope">ManagementScope</param>
        
/// <param name="ShareName">共享名</param>
        
/// <returns>移除成功则返回0</returns>
        public static int RemoveShareNetFolder(ManagementScope scope, string ShareName)
        {
            
try
            {
                
//SelectQuery selectQuery = new SelectQuery("Select * from Win32_Share Where Name = '" + ShareName + "'");
                ObjectQuery selectQuery = new ObjectQuery("Select * from Win32_Share Where Name = '" + ShareName + "'");
                ManagementObjectSearcher searcher 
= new ManagementObjectSearcher(scope, selectQuery);
                
foreach (ManagementObject mo in searcher.Get())
                {
                    mo.InvokeMethod(
"Delete"nullnull);
                }
                System.Diagnostics.Debug.WriteLine(
"in CancelShareNetFolder:" + ShareName);
            }
            
catch (ManagementException me)
            {
                Wmilog.WriteToLog(
"移除远程共享目录失败(ManagementException):" + me.Message);
                
return -1;
            }
            
catch (Exception e)
            {
                Wmilog.WriteToLog(
"移除远程共享目录失败(Exception):" + e.Message);
                
return -1;
            }

            Wmilog.WriteToLog(
"移除远程共享目录成功!");
            
return 0;
        }

        
/// <summary>
        
/// 使用net use命令远程连接目标机器,拷贝文件
        
/// </summary>
        
/// <param name="strSource">源文件</param>
        
/// <param name="strTarget">目标文件</param>
        
/// <param name="Server">目标ip</param>
        
/// <param name="ShareName">远程共享名</param>
        
/// <param name="Username">远程登录用户</param>
        
/// <param name="Password">远程登录密码</param>
        
/// <returns>拷贝成功则返回0</returns>
        public static int FileCopy(string strSource, string strTarget, string Server, string ShareName, string Username, string Password)
        {
            Process process 
= new Process();
            
try
            {

                process.StartInfo.FileName 
= "net.exe";
                process.StartInfo.Arguments 
= @"use \\" + Server + @"\" + ShareName + " \"" + Password + "\" /user:\"" + Username + "\" ";
                process.StartInfo.CreateNoWindow 
= true;
                process.StartInfo.UseShellExecute 
= false;
                process.Start();
                System.Diagnostics.Debug.WriteLine(process.StartInfo.FileName 
+ " " + process.StartInfo.Arguments);
                process.WaitForExit();
                System.Diagnostics.Debug.WriteLine(
"Start Copy File.....");
                File.Copy(strSource, strTarget, 
true);
                System.Diagnostics.Debug.WriteLine(
"End Copy File.....");
                process.StartInfo.Arguments 
= @"use \\" + Server + @"\" + ShareName + " /delete";
                process.Start();
                process.Close();
            }
            
catch (IOException ex)
            {
                System.Diagnostics.Debug.WriteLine(
"in FileCopy IOException:" + ex.Message);
                
return -1;
            }
            
catch (Exception ex)
            {
                System.Diagnostics.Debug.WriteLine(
"in FileCopy Exception:" + ex.Message);
                
return -1;
            }
            
finally
            {
                process.Dispose();
            }
            
return 0;
        }
        
        
/// <summary>
        
/// 通过远程共享名,递归拷贝指定目录下的所有信息到目标目录
        
/// </summary>
        
/// <param name="sPath">源目录</param>
        
/// <param name="dPath">目的目录</param>
        public static void CopyDirectory(string sPath, string dPath)
        {
            
string[] directories = System.IO.Directory.GetDirectories(sPath);
            
if (!System.IO.Directory.Exists(dPath))
                System.IO.Directory.CreateDirectory(dPath);
            System.IO.DirectoryInfo dir 
= new System.IO.DirectoryInfo(sPath);
            System.IO.DirectoryInfo[] dirs 
= dir.GetDirectories();
            CopyFile(dir, dPath);

            
if (dirs.Length > 0)
            {
                
foreach (System.IO.DirectoryInfo temDirectoryInfo in dirs)
                {
                    
string sourceDirectoryFullName = temDirectoryInfo.FullName;
                    
string destDirectoryFullName = sourceDirectoryFullName.Replace(sPath, dPath);
                    
if (!System.IO.Directory.Exists(destDirectoryFullName))
                    {
                        System.IO.Directory.CreateDirectory(destDirectoryFullName);
                    }
                    CopyFile(temDirectoryInfo, destDirectoryFullName);
                    CopyDirectory(sourceDirectoryFullName, destDirectoryFullName);
                }
            }
        }

        
/// <summary>
        
/// 通过远程共享名,拷贝指定目录下的所有文件到目标目录
        
/// </summary>
        
/// <param name="path">源路径</param>
        
/// <param name="desPath">目的路径(\\targetServer\\shareName\\DirectoryPath)</param>
        public static void CopyFile(System.IO.DirectoryInfo path, string desPath)
        {
            
string sourcePath = path.FullName;
            System.IO.FileInfo[] files 
= path.GetFiles();
            
foreach (System.IO.FileInfo file in files)
            {
                
string sourceFileFullName = file.FullName;
                
string destFileFullName = sourceFileFullName.Replace(sourcePath, desPath);
                file.CopyTo(destFileFullName, 
true);
            }
        }

        
/// <summary>
        
/// 用net use命令连接到远程共享目录上,创建网络共享连接
        
/// </summary>
        
/// <param name="Server">目标ip</param>
        
/// <param name="ShareName">远程共享名</param>
        
/// <param name="Username">远程登录用户</param>
        
/// <param name="Password">远程登录密码</param>
        public static void CreateShareNetConnect(string Server, string ShareName, string Username, string Password)
        {
            Process process 
= new Process();
            process.StartInfo.FileName 
= "net.exe";
            process.StartInfo.Arguments 
= @"use \\" + Server + @"\" + ShareName + " \"" + Password + "\" /user:\"" + Username + "\" ";
            process.StartInfo.CreateNoWindow 
= true;
            process.StartInfo.UseShellExecute 
= false;
            process.Start();
            process.WaitForExit();
            process.Close();
            process.Dispose();
        }

        
/// <summary>
        
/// 用net use delete命令移除网络共享连接
        
/// </summary>
        
/// <param name="Server">目标ip</param>
        
/// <param name="ShareName">远程共享名</param>
        
/// <param name="Username">远程登录用户</param>
        
/// <param name="Password">远程登录密码</param>
        public static void RemoveShareNetConnect(string Server, string ShareName, string Username, string Password)
        {
            
//System.Diagnostics.Process.Start("net.exe", @"use \\" + Server + @"\" + ShareName + " \"" + Password + "\" /user:\"" + Username + "\" ");
            Process process = new Process();
            process.StartInfo.FileName 
= "net.exe";
            process.StartInfo.Arguments 
= @"use \\" + Server + @"\" + ShareName + " /delete";
            process.StartInfo.CreateNoWindow 
= true;
            process.StartInfo.UseShellExecute 
= false;
            process.Start();
            process.WaitForExit();
            process.Close();
            process.Dispose();

        }
    }
}

 

 

 

posted on 2010-05-11 21:47 阿捷 阅读(...) 评论(...) 编辑 收藏