diff options
Diffstat (limited to 'MediaBrowser.Controller/IO')
| -rw-r--r-- | MediaBrowser.Controller/IO/FileData.cs | 4 | ||||
| -rw-r--r-- | MediaBrowser.Controller/IO/FileSystem.cs | 3 | ||||
| -rw-r--r-- | MediaBrowser.Controller/IO/NativeMethods.cs | 631 | ||||
| -rw-r--r-- | MediaBrowser.Controller/IO/NetworkShares.cs | 644 |
4 files changed, 633 insertions, 649 deletions
diff --git a/MediaBrowser.Controller/IO/FileData.cs b/MediaBrowser.Controller/IO/FileData.cs index 130f08625..7cefcd71a 100644 --- a/MediaBrowser.Controller/IO/FileData.cs +++ b/MediaBrowser.Controller/IO/FileData.cs @@ -1,6 +1,4 @@ -using MediaBrowser.Common.IO; -using MediaBrowser.Common.Win32; -using MediaBrowser.Controller.Library; +using MediaBrowser.Controller.Library; using MediaBrowser.Model.Logging; using System; using System.Collections.Generic; diff --git a/MediaBrowser.Controller/IO/FileSystem.cs b/MediaBrowser.Controller/IO/FileSystem.cs index c7b7c660b..28deca100 100644 --- a/MediaBrowser.Controller/IO/FileSystem.cs +++ b/MediaBrowser.Controller/IO/FileSystem.cs @@ -1,5 +1,4 @@ -using MediaBrowser.Common.Win32; -using System; +using System; using System.Collections.Generic; using System.Collections.Specialized; using System.IO; diff --git a/MediaBrowser.Controller/IO/NativeMethods.cs b/MediaBrowser.Controller/IO/NativeMethods.cs new file mode 100644 index 000000000..632ef0216 --- /dev/null +++ b/MediaBrowser.Controller/IO/NativeMethods.cs @@ -0,0 +1,631 @@ +using System; +using System.IO; +using System.Runtime.InteropServices; +using System.Security; +using System.Text; + +namespace MediaBrowser.Controller.IO +{ + /// <summary> + /// Class NativeMethods + /// </summary> + [SuppressUnmanagedCodeSecurity] + public static class NativeMethods + { + //declare the Netapi32 : NetServerEnum method import + /// <summary> + /// Nets the server enum. + /// </summary> + /// <param name="ServerName">Name of the server.</param> + /// <param name="dwLevel">The dw level.</param> + /// <param name="pBuf">The p buf.</param> + /// <param name="dwPrefMaxLen">The dw pref max len.</param> + /// <param name="dwEntriesRead">The dw entries read.</param> + /// <param name="dwTotalEntries">The dw total entries.</param> + /// <param name="dwServerType">Type of the dw server.</param> + /// <param name="domain">The domain.</param> + /// <param name="dwResumeHandle">The dw resume handle.</param> + /// <returns>System.Int32.</returns> + [DllImport("Netapi32", CharSet = CharSet.Auto, SetLastError = true), + SuppressUnmanagedCodeSecurityAttribute] + + public static extern int NetServerEnum( + string ServerName, // must be null + int dwLevel, + ref IntPtr pBuf, + int dwPrefMaxLen, + out int dwEntriesRead, + out int dwTotalEntries, + int dwServerType, + string domain, // null for login domain + out int dwResumeHandle + ); + + //declare the Netapi32 : NetApiBufferFree method import + /// <summary> + /// Nets the API buffer free. + /// </summary> + /// <param name="pBuf">The p buf.</param> + /// <returns>System.Int32.</returns> + [DllImport("Netapi32", SetLastError = true), + SuppressUnmanagedCodeSecurityAttribute] + + public static extern int NetApiBufferFree( + IntPtr pBuf); + + /// <summary> + /// The MA x_ PATH + /// </summary> + public const int MAX_PATH = 260; + /// <summary> + /// The MA x_ ALTERNATE + /// </summary> + public const int MAX_ALTERNATE = 14; + /// <summary> + /// The INVALI d_ HANDL e_ VALUE + /// </summary> + public static IntPtr INVALID_HANDLE_VALUE = new IntPtr(-1); + /// <summary> + /// The STG m_ READ + /// </summary> + public const uint STGM_READ = 0; + + /// <summary> + /// Finds the first file ex. + /// </summary> + /// <param name="lpFileName">Name of the lp file.</param> + /// <param name="fInfoLevelId">The f info level id.</param> + /// <param name="lpFindFileData">The lp find file data.</param> + /// <param name="fSearchOp">The f search op.</param> + /// <param name="lpSearchFilter">The lp search filter.</param> + /// <param name="dwAdditionalFlags">The dw additional flags.</param> + /// <returns>IntPtr.</returns> + [DllImport("kernel32.dll", CharSet = CharSet.Auto, SetLastError = true)] + public static extern IntPtr FindFirstFileEx(string lpFileName, FINDEX_INFO_LEVELS fInfoLevelId, out WIN32_FIND_DATA lpFindFileData, FINDEX_SEARCH_OPS fSearchOp, IntPtr lpSearchFilter, int dwAdditionalFlags); + + /// <summary> + /// Finds the first file. + /// </summary> + /// <param name="fileName">Name of the file.</param> + /// <param name="lpFindFileData">The lp find file data.</param> + /// <returns>IntPtr.</returns> + [DllImport("kernel32.dll", CharSet = CharSet.Auto, SetLastError = true)] + public static extern IntPtr FindFirstFile(string fileName, out WIN32_FIND_DATA lpFindFileData); + + /// <summary> + /// Finds the next file. + /// </summary> + /// <param name="hFindFile">The h find file.</param> + /// <param name="lpFindFileData">The lp find file data.</param> + /// <returns>IntPtr.</returns> + [DllImport("kernel32.dll", CharSet = CharSet.Auto, SetLastError = true)] + public static extern IntPtr FindNextFile(IntPtr hFindFile, out WIN32_FIND_DATA lpFindFileData); + + /// <summary> + /// Finds the close. + /// </summary> + /// <param name="hFindFile">The h find file.</param> + /// <returns><c>true</c> if XXXX, <c>false</c> otherwise</returns> + [DllImport("kernel32")] + public static extern bool FindClose(IntPtr hFindFile); + } + + //create a _SERVER_INFO_100 STRUCTURE + /// <summary> + /// Struct _SERVER_INFO_100 + /// </summary> + [StructLayout(LayoutKind.Sequential)] + public struct _SERVER_INFO_100 + { + /// <summary> + /// The sv100_platform_id + /// </summary> + internal int sv100_platform_id; + /// <summary> + /// The sv100_name + /// </summary> + [MarshalAs(UnmanagedType.LPWStr)] + internal string sv100_name; + } + + /// <summary> + /// Class FindFirstFileExFlags + /// </summary> + public class FindFirstFileExFlags + { + /// <summary> + /// The NONE + /// </summary> + public const int NONE = 0; + + /// <summary> + /// Searches are case-sensitive.Searches are case-sensitive. + /// </summary> + public const int FIND_FIRST_EX_CASE_SENSITIVE = 1; + + /// <summary> + /// Uses a larger buffer for directory queries, which can increase performance of the find operation. + /// </summary> + public const int FIND_FIRST_EX_LARGE_FETCH = 2; + } + + /// <summary> + /// Enum FINDEX_INFO_LEVELS + /// </summary> + public enum FINDEX_INFO_LEVELS + { + /// <summary> + /// The FindFirstFileEx function retrieves a standard set of attribute information. The data is returned in a WIN32_FIND_DATA structure. + /// </summary> + FindExInfoStandard = 0, + + /// <summary> + /// The FindFirstFileEx function does not query the short file name, improving overall enumeration speed. The data is returned in a WIN32_FIND_DATA structure, and the cAlternateFileName member is always a NULL string. + /// </summary> + FindExInfoBasic = 1 + } + + /// <summary> + /// Enum FINDEX_SEARCH_OPS + /// </summary> + public enum FINDEX_SEARCH_OPS + { + /// <summary> + /// The search for a file that matches a specified file name. + /// The lpSearchFilter parameter of FindFirstFileEx must be NULL when this search operation is used. + /// </summary> + FindExSearchNameMatch = 0, + + /// <summary> + /// The find ex search limit to directories + /// </summary> + FindExSearchLimitToDirectories = 1, + + /// <summary> + /// This filtering type is not available. + /// </summary> + FindExSearchLimitToDevices = 2 + } + + /// <summary> + /// Struct FILETIME + /// </summary> + [StructLayout(LayoutKind.Sequential)] + public struct FILETIME + { + /// <summary> + /// The dw low date time + /// </summary> + public uint dwLowDateTime; + /// <summary> + /// The dw high date time + /// </summary> + public uint dwHighDateTime; + } + + + /// <summary> + /// Struct WIN32_FIND_DATA + /// </summary> + [StructLayout(LayoutKind.Sequential, CharSet = CharSet.Unicode)] + public struct WIN32_FIND_DATA + { + /// <summary> + /// The dw file attributes + /// </summary> + public FileAttributes dwFileAttributes; + /// <summary> + /// The ft creation time + /// </summary> + public FILETIME ftCreationTime; + /// <summary> + /// The ft last access time + /// </summary> + public FILETIME ftLastAccessTime; + /// <summary> + /// The ft last write time + /// </summary> + public FILETIME ftLastWriteTime; + /// <summary> + /// The n file size high + /// </summary> + public int nFileSizeHigh; + /// <summary> + /// The n file size low + /// </summary> + public int nFileSizeLow; + /// <summary> + /// The dw reserved0 + /// </summary> + public int dwReserved0; + /// <summary> + /// The dw reserved1 + /// </summary> + public int dwReserved1; + + /// <summary> + /// The c file name + /// </summary> + [MarshalAs(UnmanagedType.ByValTStr, SizeConst = NativeMethods.MAX_PATH)] + public string cFileName; + + /// <summary> + /// This will always be null when FINDEX_INFO_LEVELS = basic + /// </summary> + [MarshalAs(UnmanagedType.ByValTStr, SizeConst = NativeMethods.MAX_ALTERNATE)] + public string cAlternate; + + /// <summary> + /// Gets a value indicating whether this instance is hidden. + /// </summary> + /// <value><c>true</c> if this instance is hidden; otherwise, <c>false</c>.</value> + public bool IsHidden + { + get + { + return dwFileAttributes.HasFlag(FileAttributes.Hidden); + } + } + + /// <summary> + /// Gets a value indicating whether this instance is system file. + /// </summary> + /// <value><c>true</c> if this instance is system file; otherwise, <c>false</c>.</value> + public bool IsSystemFile + { + get + { + return dwFileAttributes.HasFlag(FileAttributes.System); + } + } + + /// <summary> + /// Gets a value indicating whether this instance is directory. + /// </summary> + /// <value><c>true</c> if this instance is directory; otherwise, <c>false</c>.</value> + public bool IsDirectory + { + get + { + return dwFileAttributes.HasFlag(FileAttributes.Directory); + } + } + + /// <summary> + /// Gets the creation time UTC. + /// </summary> + /// <value>The creation time UTC.</value> + public DateTime CreationTimeUtc + { + get + { + return ParseFileTime(ftCreationTime); + } + } + + /// <summary> + /// Gets the last access time UTC. + /// </summary> + /// <value>The last access time UTC.</value> + public DateTime LastAccessTimeUtc + { + get + { + return ParseFileTime(ftLastAccessTime); + } + } + + /// <summary> + /// Gets the last write time UTC. + /// </summary> + /// <value>The last write time UTC.</value> + public DateTime LastWriteTimeUtc + { + get + { + return ParseFileTime(ftLastWriteTime); + } + } + + /// <summary> + /// Parses the file time. + /// </summary> + /// <param name="filetime">The filetime.</param> + /// <returns>DateTime.</returns> + private DateTime ParseFileTime(FILETIME filetime) + { + long highBits = filetime.dwHighDateTime; + highBits = highBits << 32; + + var val = highBits + (long) filetime.dwLowDateTime; + + if (val < 0L) + { + return DateTime.MinValue; + } + + if (val > 2650467743999999999L) + { + return DateTime.MaxValue; + } + + return DateTime.FromFileTimeUtc(val); + } + + /// <summary> + /// Gets or sets the path. + /// </summary> + /// <value>The path.</value> + public string Path { get; set; } + + /// <summary> + /// Returns a <see cref="System.String" /> that represents this instance. + /// </summary> + /// <returns>A <see cref="System.String" /> that represents this instance.</returns> + public override string ToString() + { + return Path ?? string.Empty; + } + } + + /// <summary> + /// Enum SLGP_FLAGS + /// </summary> + [Flags] + public enum SLGP_FLAGS + { + /// <summary> + /// Retrieves the standard short (8.3 format) file name + /// </summary> + SLGP_SHORTPATH = 0x1, + /// <summary> + /// Retrieves the Universal Naming Convention (UNC) path name of the file + /// </summary> + SLGP_UNCPRIORITY = 0x2, + /// <summary> + /// Retrieves the raw path name. A raw path is something that might not exist and may include environment variables that need to be expanded + /// </summary> + SLGP_RAWPATH = 0x4 + } + /// <summary> + /// Enum SLR_FLAGS + /// </summary> + [Flags] + public enum SLR_FLAGS + { + /// <summary> + /// Do not display a dialog box if the link cannot be resolved. When SLR_NO_UI is set, + /// the high-order word of fFlags can be set to a time-out value that specifies the + /// maximum amount of time to be spent resolving the link. The function returns if the + /// link cannot be resolved within the time-out duration. If the high-order word is set + /// to zero, the time-out duration will be set to the default value of 3,000 milliseconds + /// (3 seconds). To specify a value, set the high word of fFlags to the desired time-out + /// duration, in milliseconds. + /// </summary> + SLR_NO_UI = 0x1, + /// <summary> + /// Obsolete and no longer used + /// </summary> + SLR_ANY_MATCH = 0x2, + /// <summary> + /// If the link object has changed, update its path and list of identifiers. + /// If SLR_UPDATE is set, you do not need to call IPersistFile::IsDirty to determine + /// whether or not the link object has changed. + /// </summary> + SLR_UPDATE = 0x4, + /// <summary> + /// Do not update the link information + /// </summary> + SLR_NOUPDATE = 0x8, + /// <summary> + /// Do not execute the search heuristics + /// </summary> + SLR_NOSEARCH = 0x10, + /// <summary> + /// Do not use distributed link tracking + /// </summary> + SLR_NOTRACK = 0x20, + /// <summary> + /// Disable distributed link tracking. By default, distributed link tracking tracks + /// removable media across multiple devices based on the volume name. It also uses the + /// Universal Naming Convention (UNC) path to track remote file systems whose drive letter + /// has changed. Setting SLR_NOLINKINFO disables both types of tracking. + /// </summary> + SLR_NOLINKINFO = 0x40, + /// <summary> + /// Call the Microsoft Windows Installer + /// </summary> + SLR_INVOKE_MSI = 0x80 + } + + + /// <summary> + /// The IShellLink interface allows Shell links to be created, modified, and resolved + /// </summary> + [ComImport, InterfaceType(ComInterfaceType.InterfaceIsIUnknown), Guid("000214F9-0000-0000-C000-000000000046")] + public interface IShellLinkW + { + /// <summary> + /// Retrieves the path and file name of a Shell link object + /// </summary> + /// <param name="pszFile">The PSZ file.</param> + /// <param name="cchMaxPath">The CCH max path.</param> + /// <param name="pfd">The PFD.</param> + /// <param name="fFlags">The f flags.</param> + void GetPath([Out, MarshalAs(UnmanagedType.LPWStr)] StringBuilder pszFile, int cchMaxPath, out WIN32_FIND_DATA pfd, SLGP_FLAGS fFlags); + /// <summary> + /// Retrieves the list of item identifiers for a Shell link object + /// </summary> + /// <param name="ppidl">The ppidl.</param> + void GetIDList(out IntPtr ppidl); + /// <summary> + /// Sets the pointer to an item identifier list (PIDL) for a Shell link object. + /// </summary> + /// <param name="pidl">The pidl.</param> + void SetIDList(IntPtr pidl); + /// <summary> + /// Retrieves the description string for a Shell link object + /// </summary> + /// <param name="pszName">Name of the PSZ.</param> + /// <param name="cchMaxName">Name of the CCH max.</param> + void GetDescription([Out, MarshalAs(UnmanagedType.LPWStr)] StringBuilder pszName, int cchMaxName); + /// <summary> + /// Sets the description for a Shell link object. The description can be any application-defined string + /// </summary> + /// <param name="pszName">Name of the PSZ.</param> + void SetDescription([MarshalAs(UnmanagedType.LPWStr)] string pszName); + /// <summary> + /// Retrieves the name of the working directory for a Shell link object + /// </summary> + /// <param name="pszDir">The PSZ dir.</param> + /// <param name="cchMaxPath">The CCH max path.</param> + void GetWorkingDirectory([Out, MarshalAs(UnmanagedType.LPWStr)] StringBuilder pszDir, int cchMaxPath); + /// <summary> + /// Sets the name of the working directory for a Shell link object + /// </summary> + /// <param name="pszDir">The PSZ dir.</param> + void SetWorkingDirectory([MarshalAs(UnmanagedType.LPWStr)] string pszDir); + /// <summary> + /// Retrieves the command-line arguments associated with a Shell link object + /// </summary> + /// <param name="pszArgs">The PSZ args.</param> + /// <param name="cchMaxPath">The CCH max path.</param> + void GetArguments([Out, MarshalAs(UnmanagedType.LPWStr)] StringBuilder pszArgs, int cchMaxPath); + /// <summary> + /// Sets the command-line arguments for a Shell link object + /// </summary> + /// <param name="pszArgs">The PSZ args.</param> + void SetArguments([MarshalAs(UnmanagedType.LPWStr)] string pszArgs); + /// <summary> + /// Retrieves the hot key for a Shell link object + /// </summary> + /// <param name="pwHotkey">The pw hotkey.</param> + void GetHotkey(out short pwHotkey); + /// <summary> + /// Sets a hot key for a Shell link object + /// </summary> + /// <param name="wHotkey">The w hotkey.</param> + void SetHotkey(short wHotkey); + /// <summary> + /// Retrieves the show command for a Shell link object + /// </summary> + /// <param name="piShowCmd">The pi show CMD.</param> + void GetShowCmd(out int piShowCmd); + /// <summary> + /// Sets the show command for a Shell link object. The show command sets the initial show state of the window. + /// </summary> + /// <param name="iShowCmd">The i show CMD.</param> + void SetShowCmd(int iShowCmd); + /// <summary> + /// Retrieves the location (path and index) of the icon for a Shell link object + /// </summary> + /// <param name="pszIconPath">The PSZ icon path.</param> + /// <param name="cchIconPath">The CCH icon path.</param> + /// <param name="piIcon">The pi icon.</param> + void GetIconLocation([Out, MarshalAs(UnmanagedType.LPWStr)] StringBuilder pszIconPath, + int cchIconPath, out int piIcon); + /// <summary> + /// Sets the location (path and index) of the icon for a Shell link object + /// </summary> + /// <param name="pszIconPath">The PSZ icon path.</param> + /// <param name="iIcon">The i icon.</param> + void SetIconLocation([MarshalAs(UnmanagedType.LPWStr)] string pszIconPath, int iIcon); + /// <summary> + /// Sets the relative path to the Shell link object + /// </summary> + /// <param name="pszPathRel">The PSZ path rel.</param> + /// <param name="dwReserved">The dw reserved.</param> + void SetRelativePath([MarshalAs(UnmanagedType.LPWStr)] string pszPathRel, int dwReserved); + /// <summary> + /// Attempts to find the target of a Shell link, even if it has been moved or renamed + /// </summary> + /// <param name="hwnd">The HWND.</param> + /// <param name="fFlags">The f flags.</param> + void Resolve(IntPtr hwnd, SLR_FLAGS fFlags); + /// <summary> + /// Sets the path and file name of a Shell link object + /// </summary> + /// <param name="pszFile">The PSZ file.</param> + void SetPath([MarshalAs(UnmanagedType.LPWStr)] string pszFile); + + } + + /// <summary> + /// Interface IPersist + /// </summary> + [ComImport, Guid("0000010c-0000-0000-c000-000000000046"), + InterfaceType(ComInterfaceType.InterfaceIsIUnknown)] + public interface IPersist + { + /// <summary> + /// Gets the class ID. + /// </summary> + /// <param name="pClassID">The p class ID.</param> + [PreserveSig] + void GetClassID(out Guid pClassID); + } + + + /// <summary> + /// Interface IPersistFile + /// </summary> + [ComImport, Guid("0000010b-0000-0000-C000-000000000046"), + InterfaceType(ComInterfaceType.InterfaceIsIUnknown)] + public interface IPersistFile : IPersist + { + /// <summary> + /// Gets the class ID. + /// </summary> + /// <param name="pClassID">The p class ID.</param> + new void GetClassID(out Guid pClassID); + /// <summary> + /// Determines whether this instance is dirty. + /// </summary> + [PreserveSig] + int IsDirty(); + + /// <summary> + /// Loads the specified PSZ file name. + /// </summary> + /// <param name="pszFileName">Name of the PSZ file.</param> + /// <param name="dwMode">The dw mode.</param> + [PreserveSig] + void Load([In, MarshalAs(UnmanagedType.LPWStr)] + string pszFileName, uint dwMode); + + /// <summary> + /// Saves the specified PSZ file name. + /// </summary> + /// <param name="pszFileName">Name of the PSZ file.</param> + /// <param name="remember">if set to <c>true</c> [remember].</param> + [PreserveSig] + void Save([In, MarshalAs(UnmanagedType.LPWStr)] string pszFileName, + [In, MarshalAs(UnmanagedType.Bool)] bool remember); + + /// <summary> + /// Saves the completed. + /// </summary> + /// <param name="pszFileName">Name of the PSZ file.</param> + [PreserveSig] + void SaveCompleted([In, MarshalAs(UnmanagedType.LPWStr)] string pszFileName); + + /// <summary> + /// Gets the cur file. + /// </summary> + /// <param name="ppszFileName">Name of the PPSZ file.</param> + [PreserveSig] + void GetCurFile([In, MarshalAs(UnmanagedType.LPWStr)] string ppszFileName); + } + + // CLSID_ShellLink from ShlGuid.h + /// <summary> + /// Class ShellLink + /// </summary> + [ + ComImport, + Guid("00021401-0000-0000-C000-000000000046") + ] + public class ShellLink + { + } +} diff --git a/MediaBrowser.Controller/IO/NetworkShares.cs b/MediaBrowser.Controller/IO/NetworkShares.cs deleted file mode 100644 index 93edc6447..000000000 --- a/MediaBrowser.Controller/IO/NetworkShares.cs +++ /dev/null @@ -1,644 +0,0 @@ -using System; -using System.IO; -using System.Collections; -using System.Runtime.InteropServices; - -namespace MediaBrowser.Controller.IO -{ - /// <summary> - /// Type of share - /// </summary> - [Flags] - public enum ShareType - { - /// <summary>Disk share</summary> - Disk = 0, - /// <summary>Printer share</summary> - Printer = 1, - /// <summary>Device share</summary> - Device = 2, - /// <summary>IPC share</summary> - IPC = 3, - /// <summary>Special share</summary> - Special = -2147483648, // 0x80000000, - } - - #region Share - - /// <summary> - /// Information about a local share - /// </summary> - public class Share - { - #region Private data - - private string _server; - private string _netName; - private string _path; - private ShareType _shareType; - private string _remark; - - #endregion - - #region Constructor - - /// <summary> - /// Constructor - /// </summary> - /// <param name="Server"></param> - /// <param name="shi"></param> - public Share(string server, string netName, string path, ShareType shareType, string remark) - { - if (ShareType.Special == shareType && "IPC$" == netName) - { - shareType |= ShareType.IPC; - } - - _server = server; - _netName = netName; - _path = path; - _shareType = shareType; - _remark = remark; - } - - #endregion - - #region Properties - - /// <summary> - /// The name of the computer that this share belongs to - /// </summary> - public string Server - { - get { return _server; } - } - - /// <summary> - /// Share name - /// </summary> - public string NetName - { - get { return _netName; } - } - - /// <summary> - /// Local path - /// </summary> - public string Path - { - get { return _path; } - } - - /// <summary> - /// Share type - /// </summary> - public ShareType ShareType - { - get { return _shareType; } - } - - /// <summary> - /// Comment - /// </summary> - public string Remark - { - get { return _remark; } - } - - /// <summary> - /// Returns true if this is a file system share - /// </summary> - public bool IsFileSystem - { - get - { - // Shared device - if (0 != (_shareType & ShareType.Device)) return false; - // IPC share - if (0 != (_shareType & ShareType.IPC)) return false; - // Shared printer - if (0 != (_shareType & ShareType.Printer)) return false; - - // Standard disk share - if (0 == (_shareType & ShareType.Special)) return true; - - // Special disk share (e.g. C$) - if (ShareType.Special == _shareType && null != _netName && 0 != _netName.Length) - return true; - else - return false; - } - } - - /// <summary> - /// Get the root of a disk-based share - /// </summary> - public DirectoryInfo Root - { - get - { - if (IsFileSystem) - { - if (null == _server || 0 == _server.Length) - if (null == _path || 0 == _path.Length) - return new DirectoryInfo(ToString()); - else - return new DirectoryInfo(_path); - else - return new DirectoryInfo(ToString()); - } - else - return null; - } - } - - #endregion - - /// <summary> - /// Returns the path to this share - /// </summary> - /// <returns></returns> - public override string ToString() - { - if (null == _server || 0 == _server.Length) - { - return string.Format(@"\\{0}\{1}", Environment.MachineName, _netName); - } - else - return string.Format(@"\\{0}\{1}", _server, _netName); - } - - /// <summary> - /// Returns true if this share matches the local path - /// </summary> - /// <param name="path"></param> - /// <returns></returns> - public bool MatchesPath(string path) - { - if (!IsFileSystem) return false; - if (null == path || 0 == path.Length) return true; - - return path.ToLower().StartsWith(_path.ToLower()); - } - } - - #endregion - - /// <summary> - /// A collection of shares - /// </summary> - public class ShareCollection : ReadOnlyCollectionBase - { - #region Platform - - /// <summary> - /// Is this an NT platform? - /// </summary> - protected static bool IsNT - { - get { return (PlatformID.Win32NT == Environment.OSVersion.Platform); } - } - - /// <summary> - /// Returns true if this is Windows 2000 or higher - /// </summary> - protected static bool IsW2KUp - { - get - { - OperatingSystem os = Environment.OSVersion; - if (PlatformID.Win32NT == os.Platform && os.Version.Major >= 5) - return true; - else - return false; - } - } - - #endregion - - #region Interop - - #region Constants - - /// <summary>Maximum path length</summary> - protected const int MAX_PATH = 260; - /// <summary>No error</summary> - protected const int NO_ERROR = 0; - /// <summary>Access denied</summary> - protected const int ERROR_ACCESS_DENIED = 5; - /// <summary>Access denied</summary> - protected const int ERROR_WRONG_LEVEL = 124; - /// <summary>More data available</summary> - protected const int ERROR_MORE_DATA = 234; - /// <summary>Not connected</summary> - protected const int ERROR_NOT_CONNECTED = 2250; - /// <summary>Level 1</summary> - protected const int UNIVERSAL_NAME_INFO_LEVEL = 1; - /// <summary>Max extries (9x)</summary> - protected const int MAX_SI50_ENTRIES = 20; - - #endregion - - #region Structures - - /// <summary>Unc name</summary> - [StructLayout(LayoutKind.Sequential, CharSet = CharSet.Auto)] - protected struct UNIVERSAL_NAME_INFO - { - [MarshalAs(UnmanagedType.LPTStr)] - public string lpUniversalName; - } - - /// <summary>Share information, NT, level 2</summary> - /// <remarks> - /// Requires admin rights to work. - /// </remarks> - [StructLayout(LayoutKind.Sequential, CharSet = CharSet.Unicode)] - protected struct SHARE_INFO_2 - { - [MarshalAs(UnmanagedType.LPWStr)] - public string NetName; - public ShareType ShareType; - [MarshalAs(UnmanagedType.LPWStr)] - public string Remark; - public int Permissions; - public int MaxUsers; - public int CurrentUsers; - [MarshalAs(UnmanagedType.LPWStr)] - public string Path; - [MarshalAs(UnmanagedType.LPWStr)] - public string Password; - } - - /// <summary>Share information, NT, level 1</summary> - /// <remarks> - /// Fallback when no admin rights. - /// </remarks> - [StructLayout(LayoutKind.Sequential, CharSet = CharSet.Unicode)] - protected struct SHARE_INFO_1 - { - [MarshalAs(UnmanagedType.LPWStr)] - public string NetName; - public ShareType ShareType; - [MarshalAs(UnmanagedType.LPWStr)] - public string Remark; - } - - /// <summary>Share information, Win9x</summary> - [StructLayout(LayoutKind.Sequential, CharSet = CharSet.Ansi, Pack = 1)] - protected struct SHARE_INFO_50 - { - [MarshalAs(UnmanagedType.ByValTStr, SizeConst = 13)] - public string NetName; - - public byte bShareType; - public ushort Flags; - - [MarshalAs(UnmanagedType.LPTStr)] - public string Remark; - [MarshalAs(UnmanagedType.LPTStr)] - public string Path; - - [MarshalAs(UnmanagedType.ByValTStr, SizeConst = 9)] - public string PasswordRW; - [MarshalAs(UnmanagedType.ByValTStr, SizeConst = 9)] - public string PasswordRO; - - public ShareType ShareType - { - get { return (ShareType)((int)bShareType & 0x7F); } - } - } - - /// <summary>Share information level 1, Win9x</summary> - [StructLayout(LayoutKind.Sequential, CharSet = CharSet.Ansi, Pack = 1)] - protected struct SHARE_INFO_1_9x - { - [MarshalAs(UnmanagedType.ByValTStr, SizeConst = 13)] - public string NetName; - public byte Padding; - - public ushort bShareType; - - [MarshalAs(UnmanagedType.LPTStr)] - public string Remark; - - public ShareType ShareType - { - get { return (ShareType)((int)bShareType & 0x7FFF); } - } - } - - #endregion - - #region Functions - - /// <summary>Get a UNC name</summary> - [DllImport("mpr", CharSet = CharSet.Auto)] - protected static extern int WNetGetUniversalName(string lpLocalPath, - int dwInfoLevel, ref UNIVERSAL_NAME_INFO lpBuffer, ref int lpBufferSize); - - /// <summary>Get a UNC name</summary> - [DllImport("mpr", CharSet = CharSet.Auto)] - protected static extern int WNetGetUniversalName(string lpLocalPath, - int dwInfoLevel, IntPtr lpBuffer, ref int lpBufferSize); - - /// <summary>Enumerate shares (NT)</summary> - [DllImport("netapi32", CharSet = CharSet.Unicode)] - protected static extern int NetShareEnum(string lpServerName, int dwLevel, - out IntPtr lpBuffer, int dwPrefMaxLen, out int entriesRead, - out int totalEntries, ref int hResume); - - /// <summary>Enumerate shares (9x)</summary> - [DllImport("svrapi", CharSet = CharSet.Ansi)] - protected static extern int NetShareEnum( - [MarshalAs(UnmanagedType.LPTStr)] string lpServerName, int dwLevel, - IntPtr lpBuffer, ushort cbBuffer, out ushort entriesRead, - out ushort totalEntries); - - /// <summary>Free the buffer (NT)</summary> - [DllImport("netapi32")] - protected static extern int NetApiBufferFree(IntPtr lpBuffer); - - #endregion - - #region Enumerate shares - - /// <summary> - /// Enumerates the shares on Windows NT - /// </summary> - /// <param name="server">The server name</param> - /// <param name="shares">The ShareCollection</param> - protected static void EnumerateSharesNT(string server, ShareCollection shares) - { - int level = 2; - int entriesRead, totalEntries, nRet, hResume = 0; - IntPtr pBuffer = IntPtr.Zero; - - try - { - nRet = NetShareEnum(server, level, out pBuffer, -1, - out entriesRead, out totalEntries, ref hResume); - - if (ERROR_ACCESS_DENIED == nRet) - { - //Need admin for level 2, drop to level 1 - level = 1; - nRet = NetShareEnum(server, level, out pBuffer, -1, - out entriesRead, out totalEntries, ref hResume); - } - - if (NO_ERROR == nRet && entriesRead > 0) - { - Type t = (2 == level) ? typeof(SHARE_INFO_2) : typeof(SHARE_INFO_1); - int offset = Marshal.SizeOf(t); - - for (int i = 0, lpItem = pBuffer.ToInt32(); i < entriesRead; i++, lpItem += offset) - { - IntPtr pItem = new IntPtr(lpItem); - if (1 == level) - { - SHARE_INFO_1 si = (SHARE_INFO_1)Marshal.PtrToStructure(pItem, t); - shares.Add(si.NetName, string.Empty, si.ShareType, si.Remark); - } - else - { - SHARE_INFO_2 si = (SHARE_INFO_2)Marshal.PtrToStructure(pItem, t); - shares.Add(si.NetName, si.Path, si.ShareType, si.Remark); - } - } - } - - } - finally - { - // Clean up buffer allocated by system - if (IntPtr.Zero != pBuffer) - NetApiBufferFree(pBuffer); - } - } - - /// <summary> - /// Enumerates the shares on Windows 9x - /// </summary> - /// <param name="server">The server name</param> - /// <param name="shares">The ShareCollection</param> - protected static void EnumerateShares9x(string server, ShareCollection shares) - { - int level = 50; - int nRet = 0; - ushort entriesRead, totalEntries; - - Type t = typeof(SHARE_INFO_50); - int size = Marshal.SizeOf(t); - ushort cbBuffer = (ushort)(MAX_SI50_ENTRIES * size); - //On Win9x, must allocate buffer before calling API - IntPtr pBuffer = Marshal.AllocHGlobal(cbBuffer); - - try - { - nRet = NetShareEnum(server, level, pBuffer, cbBuffer, - out entriesRead, out totalEntries); - - if (ERROR_WRONG_LEVEL == nRet) - { - level = 1; - t = typeof(SHARE_INFO_1_9x); - size = Marshal.SizeOf(t); - - nRet = NetShareEnum(server, level, pBuffer, cbBuffer, - out entriesRead, out totalEntries); - } - - if (NO_ERROR == nRet || ERROR_MORE_DATA == nRet) - { - for (int i = 0, lpItem = pBuffer.ToInt32(); i < entriesRead; i++, lpItem += size) - { - IntPtr pItem = new IntPtr(lpItem); - - if (1 == level) - { - SHARE_INFO_1_9x si = (SHARE_INFO_1_9x)Marshal.PtrToStructure(pItem, t); - shares.Add(si.NetName, string.Empty, si.ShareType, si.Remark); - } - else - { - SHARE_INFO_50 si = (SHARE_INFO_50)Marshal.PtrToStructure(pItem, t); - shares.Add(si.NetName, si.Path, si.ShareType, si.Remark); - } - } - } - else - Console.WriteLine(nRet); - - } - finally - { - //Clean up buffer - Marshal.FreeHGlobal(pBuffer); - } - } - - /// <summary> - /// Enumerates the shares - /// </summary> - /// <param name="server">The server name</param> - /// <param name="shares">The ShareCollection</param> - protected static void EnumerateShares(string server, ShareCollection shares) - { - if (null != server && 0 != server.Length && !IsW2KUp) - { - server = server.ToUpper(); - - // On NT4, 9x and Me, server has to start with "\\" - if (!('\\' == server[0] && '\\' == server[1])) - server = @"\\" + server; - } - - if (IsNT) - EnumerateSharesNT(server, shares); - else - EnumerateShares9x(server, shares); - } - - #endregion - - #endregion - - #region Static methods - - /// <summary> - /// Returns true if fileName is a valid local file-name of the form: - /// X:\, where X is a drive letter from A-Z - /// </summary> - /// <param name="fileName">The filename to check</param> - /// <returns></returns> - public static bool IsValidFilePath(string fileName) - { - if (null == fileName || 0 == fileName.Length) return false; - - char drive = char.ToUpper(fileName[0]); - if ('A' > drive || drive > 'Z') - return false; - - else if (Path.VolumeSeparatorChar != fileName[1]) - return false; - else if (Path.DirectorySeparatorChar != fileName[2]) - return false; - else - return true; - } - - #endregion - - /// <summary>The name of the server this collection represents</summary> - private string _server; - - #region Constructor - - /// <summary> - /// Default constructor - local machine - /// </summary> - public ShareCollection() - { - _server = string.Empty; - EnumerateShares(_server, this); - } - - /// <summary> - /// Constructor - /// </summary> - /// <param name="Server"></param> - public ShareCollection(string server) - { - _server = server; - EnumerateShares(_server, this); - } - - #endregion - - #region Add - - protected void Add(Share share) - { - InnerList.Add(share); - } - - protected void Add(string netName, string path, ShareType shareType, string remark) - { - InnerList.Add(new Share(_server, netName, path, shareType, remark)); - } - - #endregion - - #region Properties - - /// <summary> - /// Returns the name of the server this collection represents - /// </summary> - public string Server - { - get { return _server; } - } - - /// <summary> - /// Returns the <see cref="Share"/> at the specified index. - /// </summary> - public Share this[int index] - { - get { return (Share)InnerList[index]; } - } - - /// <summary> - /// Returns the <see cref="Share"/> which matches a given local path - /// </summary> - /// <param name="path">The path to match</param> - public Share this[string path] - { - get - { - if (null == path || 0 == path.Length) return null; - - path = Path.GetFullPath(path); - if (!IsValidFilePath(path)) return null; - - Share match = null; - - for (int i = 0; i < InnerList.Count; i++) - { - Share s = (Share)InnerList[i]; - - if (s.IsFileSystem && s.MatchesPath(path)) - { - //Store first match - if (null == match) - match = s; - - // If this has a longer path, - // and this is a disk share or match is a special share, - // then this is a better match - else if (match.Path.Length < s.Path.Length) - { - if (ShareType.Disk == s.ShareType || ShareType.Disk != match.ShareType) - match = s; - } - } - } - - return match; - } - } - - #endregion - - /// <summary> - /// Copy this collection to an array - /// </summary> - /// <param name="array"></param> - /// <param name="index"></param> - public void CopyTo(Share[] array, int index) - { - InnerList.CopyTo(array, index); - } - } -}
\ No newline at end of file |
