[WinAPI] API 6 [操作驱动器挂载点]

  1 #define _WIN32_WINNT 0x0501
  2 #include <windows.h>
  3 #include <stdio.h>
  4 #include <tchar.h>
  5 #define BUFSIZE            MAX_PATH
  6 #define FILESYSNAMEBUFSIZE MAX_PATH
  7 
  8 /* ************************************
  9 * ProcessVolumeMountPoint
 10 * 功能 列举挂载点
 11 **************************************/
 12 BOOL ProcessVolumeMountPoint (HANDLE hPt,
 13                               TCHAR *PtBuf, DWORD dwPtBufSize,
 14                               TCHAR *Buf)
 15 {
 16     BOOL bFlag;      // 结果
 17     TCHAR Path[BUFSIZE]; // 全路径
 18     TCHAR Target[BUFSIZE];   // 挂载点设备
 19 
 20     printf ("\tVolume mount point found is \"%s\"\n", PtBuf);
 21 
 22     lstrcpy (Path, Buf);
 23     lstrcat (Path, PtBuf);
 24 
 25     bFlag = GetVolumeNameForVolumeMountPoint(Path,Target,BUFSIZE);
 26 
 27     if (!bFlag)
 28         printf ("\tAttempt to get volume name for %s failed.\n", Path);
 29     else 
 30         printf ("\tTarget of the volume mount point is %s.\n", Target);
 31 
 32     bFlag = FindNextVolumeMountPoint(hPt,PtBuf,dwPtBufSize);
 33     return (bFlag);
 34 }
 35 /* ************************************
 36 * ProcessVolume
 37 * 功能 判断卷类型，列举挂载点
 38 **************************************/
 39 BOOL ProcessVolume (HANDLE hVol, TCHAR *Buf, DWORD iBufSize)
 40 {
 41     BOOL bFlag;           // 返回标志
 42     HANDLE hPt;           // 卷句柄
 43     TCHAR PtBuf[BUFSIZE]; // 挂载点路径
 44     DWORD dwSysFlags;     // 文件系统标记
 45     TCHAR FileSysNameBuf[FILESYSNAMEBUFSIZE];
 46 
 47     printf ("Volume found is \"%s\".\n", Buf);
 48 
 49     // 是否NTFS
 50     GetVolumeInformation( Buf, NULL, 0, NULL, NULL,
 51         &dwSysFlags, FileSysNameBuf, 
 52         FILESYSNAMEBUFSIZE);
 53 
 54     if (! (dwSysFlags & FILE_SUPPORTS_REPARSE_POINTS)) 
 55     {
 56         printf ("\tThis file system does not support volume mount points.\n");
 57     } 
 58     else 
 59     {
 60         // 本卷中的挂载点
 61         hPt = FindFirstVolumeMountPoint(
 62             Buf, // 卷的跟跟踪
 63             PtBuf, // 挂载点路径
 64             BUFSIZE 
 65             );
 66 
 67         if (hPt == INVALID_HANDLE_VALUE)
 68         {
 69             printf ("\tNo volume mount points found!\n");
 70         } 
 71         else 
 72         {
 73             // 处理挂载点
 74             bFlag = ProcessVolumeMountPoint (hPt, 
 75                 PtBuf, 
 76                 BUFSIZE, 
 77                 Buf);
 78             // 循环
 79             while (bFlag) 
 80                 bFlag = 
 81                 ProcessVolumeMountPoint (hPt, PtBuf, BUFSIZE, Buf);
 82             // 结束
 83             FindVolumeMountPointClose(hPt);
 84         }
 85     }
 86 
 87     // 下一个
 88     bFlag = FindNextVolume(
 89         hVol, Buf, iBufSize);
 90 
 91     return (bFlag); 
 92 }
 93 /* ************************************
 94 * int GetMountPoint(void)
 95 * 功能 获取挂载点
 96 **************************************/
 97 int GetMountPoint(void)
 98 {
 99     TCHAR buf[BUFSIZE];   // 卷标识符
100     HANDLE hVol;      // 卷句柄
101     BOOL bFlag;      // 结果标志
102 
103     printf("Volume mount points info of this computer:\n\n");
104     // 打开卷
105     hVol = FindFirstVolume (buf, BUFSIZE );
106     if (hVol == INVALID_HANDLE_VALUE)
107     {
108         printf ("No volumes found!\n");
109         return (-1);
110     }
111 
112     bFlag = ProcessVolume (hVol, buf, BUFSIZE);
113     while (bFlag) 
114     {
115         bFlag = ProcessVolume (hVol, buf, BUFSIZE);
116     }
117 
118     bFlag = FindVolumeClose( hVol );
119     return (bFlag);
120 }
121 
122 /* ************************************
123 * void Usage (PCHAR argv)
124 * 功能 使用方法
125 **************************************/
126 void Usage (PCHAR argv)
127 {
128     printf( "\n\n\t%s, mount a volume at a mount point.\n", argv );
129     printf( "\tFor example, \"mount D:\\mnt\\drives\\ E:\\\"\n" );
130 }
131 
132 /* ************************************
133 * main
134 * 功能 入口函数
135 **************************************/
136 int main( int argc, PCHAR argv[] )
137 {
138     BOOL bFlag;
139     CHAR Buf[BUFSIZE];    
140 
141     if( argc != 3 ) 
142     {
143         GetMountPoint();
144         Usage( argv[0] );
145         return( -1 );
146     }
147 
148     bFlag = GetVolumeNameForVolumeMountPointA(
149         argv[2],   // 输入挂载点或目录
150         Buf,     // 输出卷名
151         BUFSIZE
152         );
153 
154     if (bFlag != TRUE) 
155     {
156         printf( "Retrieving volume name for %s failed.\n", argv[2] );
157         return (-2);
158     }
159 
160     printf( "Volume name of %s is %s\n", argv[2], Buf );
161     bFlag = SetVolumeMountPointA(
162         argv[1], // 挂载点
163         Buf    // 需要挂载的卷
164         );
165 
166     if (!bFlag)
167     {
168         printf ("Attempt to mount %s at %s failed. error code is\n", 
169             argv[2], argv[1], GetLastError());
170     }
171 
172     return (bFlag);
173 }