使用WMI来操作Windows共享机制

using System;
using System.Collections.Generic;
using System.Text;
using System.Management;
using System.Diagnostics;
using System.Threading;

namespace TJVictor.WMI
{
    public class Win32_Share:WMIBaseClass
    {
        #region Property
        private int timeout = 30;
        public int TimeOut
        {
            get { return timeout; }
            set { timeout = value; }
        }

        string wqlSelect = string.Empty;
        #endregion

        #region Construction
        public Win32_Share()
            : base()
        {
            wqlSelect = "select * FROM Win32_Share where Name='{0}'";
            base.Connection();
        }
        public Win32_Share(string domain, string Ip, string user, string psd)
            : base(domain, Ip, user, psd)
        {
            wqlSelect = "select * FROM Win32_Share where Name='{0}'";
            base.Connection();
        }
        #endregion

        #region public function
        public bool IsShareDirectoryExist(string shareName)
        {
            if (!GetSelectQueryCollection(wqlSelect, shareName).Count.Equals(0))
                return true;
            else
                return false;
        }

        public void ConnectionCredit()
        {
            string creditUser = base.User;
            //判断是否为域用户
            if (!base.Domain.Equals(string.Empty))
                creditUser = base.Domain + "\\" + base.User;

            Process connectionCreditProcess = new Process();
            connectionCreditProcess.StartInfo.CreateNoWindow = true;
            connectionCreditProcess.StartInfo.UseShellExecute = false;
            connectionCreditProcess.StartInfo.FileName = string.Format(@"C:\WINDOWS\system32\cmd.exe");
            connectionCreditProcess.StartInfo.Arguments = string.Format("/c net use \\\\{0} \"{1}\" /user:\"{2}\"", base.Ip
                , base.Password, creditUser);
            connectionCreditProcess.Start();

            int tempTimeout = this.timeout;
            while (!connectionCreditProcess.HasExited)
            {
                Thread.Sleep(500);
                connectionCreditProcess.Refresh();
                if(tempTimeout.Equals(0))
                    throw new TJVictor.WMI.WmiException.ShareException(
                        string.Format("与{0}建立信任关系失败，建立超时",base.Ip));
                tempTimeout--;
            }
        }

        public void DisconnectionCredit()
        {
            Process disconnectionCreditProcess = new Process();
            disconnectionCreditProcess.StartInfo.CreateNoWindow = true;
            disconnectionCreditProcess.StartInfo.UseShellExecute = false;
            disconnectionCreditProcess.StartInfo.FileName = string.Format(@"C:\WINDOWS\system32\cmd.exe");
            disconnectionCreditProcess.StartInfo.Arguments = string.Format(@"/c net use \\{0}\ipc$ /delete", base.Ip);
            disconnectionCreditProcess.Start();

            int tempTimeout = this.timeout;
            while (!disconnectionCreditProcess.HasExited)
            {
                Thread.Sleep(500);
                disconnectionCreditProcess.Refresh();
                if(tempTimeout.Equals(0))
                    throw new TJVictor.WMI.WmiException.ShareException(
                        string.Format("与{0}断开信任关系失败，断开超时", base.Ip));
                tempTimeout--;
            }
        }

        public void CreateShareDir(string name, string shareDir)
        {
            Win32_Directory directory = new Win32_Directory(base.Domain, base.Ip, base.User, base.Password);
            if(!directory.IsDirExist(name))
                throw new TJVictor.WMI.WmiException.ShareException(
                    string.Format("无法在{0}上建立{1}目录共享，目录不存在", base.Ip,shareDir));

            ManagementClass processClass = new ManagementClass("Win32_Share");
            processClass.Scope = base.Scope;

            string method = "Create";
            ManagementBaseObject inputArgs = processClass.GetMethodParameters(method);
            inputArgs["Name"] = name;
            inputArgs["Path"] = shareDir;
            inputArgs["Description"] = "wmi Create shared";
            inputArgs["Type"] = 0; // Disk share type
            ManagementBaseObject result = processClass.InvokeMethod(method, inputArgs, null);
            CheckExceptionClass.CheckShareException(int.Parse(result["ReturnValue"].ToString()));

            //检测是否已经建立共享
            int tempTimeout = this.timeout;
            while (!IsShareDirectoryExist(name))
            {
                processClass.InvokeMethod(method, inputArgs, null);//再建立一次
                if(tempTimeout.Equals(0))
                    throw new TJVictor.WMI.WmiException.ShareException(
                        string.Format("无法在{0}上建立{1}目录共享，建立超时", base.Ip, name));
                tempTimeout--;
            }
        }

        public void DeleteShareDir(string name)
        {
            object result = 0;
            foreach (ManagementObject mo in GetSelectQueryCollection(wqlSelect,name))
            {
                result = mo.InvokeMethod("Delete", null);
                break;
            }
            CheckExceptionClass.CheckShareException(int.Parse(result.ToString()));

            //检测是否已经删除共享
            int tempTimeout = this.timeout;
            while (IsShareDirectoryExist(name))
            {
                foreach (ManagementObject mo in GetSelectQueryCollection(wqlSelect, name))//再删除一次
                {
                    result = mo.InvokeMethod("Delete", null);
                    break;
                }
                if(tempTimeout.Equals(0))
                    throw new TJVictor.WMI.WmiException.ShareException(
                         string.Format("无法在{0}上删除{1}共享，建立超时", base.Ip, name));
                tempTimeout--;
            }
        }
        #endregion
    }
}