[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 }