812144971b320bd0b2f57485a41cc00cf97fca39
[occt.git] / src / OSD / OSD_Disk.cxx
1 // Copyright (c) 1998-1999 Matra Datavision
2 // Copyright (c) 1999-2014 OPEN CASCADE SAS
3 //
4 // This file is part of Open CASCADE Technology software library.
5 //
6 // This library is free software; you can redistribute it and/or modify it under
7 // the terms of the GNU Lesser General Public License version 2.1 as published
8 // by the Free Software Foundation, with special exception defined in the file
9 // OCCT_LGPL_EXCEPTION.txt. Consult the file LICENSE_LGPL_21.txt included in OCCT
10 // distribution for complete text of the license and disclaimer of any warranty.
11 //
12 // Alternatively, this file may be used under the terms of Open CASCADE
13 // commercial license or contractual agreement.
14
15 #ifndef _WIN32
16
17
18 #include <OSD_Disk.hxx>
19 #include <OSD_OSDError.hxx>
20 #include <OSD_Path.hxx>
21 #include <OSD_WhoAmI.hxx>
22
23 const OSD_WhoAmI Iam = OSD_WDisk;
24
25 #ifdef __cplusplus
26 extern "C" {
27 #endif
28
29 #if defined(__ANDROID__)
30   #include <sys/vfs.h>
31   #define statvfs  statfs
32   #define fstatvfs fstatfs
33 #else
34   #include <sys/statvfs.h>
35 #endif
36
37 #ifdef __cplusplus
38 }
39 #endif
40
41 #include <errno.h>
42
43 OSD_Disk::OSD_Disk() : myQuotaSize(0) {}
44
45
46 OSD_Disk::OSD_Disk(const OSD_Path& name){
47  DiskName = name.Disk();
48  myQuotaSize = 0;
49 }
50
51 OSD_Disk::OSD_Disk(const Standard_CString name)
52 {
53  DiskName = name;
54  myQuotaSize = 0;
55 }
56
57 void OSD_Disk::SetName(const OSD_Path& name){
58  DiskName = name.Disk();
59 }
60
61
62 OSD_Path OSD_Disk::Name()const{
63  OSD_Path result;
64  result.SetDisk(DiskName);
65  return(result);
66 }
67
68 Standard_Integer OSD_Disk::DiskSize(){
69
70 struct statvfs buffer;
71
72   if ( statvfs(DiskName.ToCString(),&buffer) == 0 ){
73     int BSize512 = buffer.f_frsize / 512 ;
74     return buffer.f_blocks * BSize512 ;
75   }
76   else {
77     myError.SetValue(errno, Iam, "OSD_Disk: statvfs failed.");
78     return 0;
79   }
80 }
81
82 Standard_Integer OSD_Disk::DiskFree(){
83
84 struct statvfs buffer;
85   if ( statvfs (DiskName.ToCString(),&buffer) == 0 ){
86     int BSize512 = buffer.f_frsize / 512 ;
87     return buffer.f_bavail * BSize512 ;
88   }
89   else {
90     myError.SetValue(errno, Iam, "OSD_Disk: statvfs failed.");
91     return 0;
92   }
93 }
94
95 Standard_Integer OSD_Disk::DiskQuota(){
96 //@@@ A faire
97 return 0;
98 }
99
100
101 void OSD_Disk::SetDiskQuota(const Standard_Integer ){
102 // int status;
103 // struct dqblk quota_info;
104 #ifdef ULTRIX
105 // status = quota(Q_SETDLIM,<< User Id>>,&quota_info);
106 #else
107 // status = quotactl(Q_SETQLIM,"",<< User Id >>,&quota_info);
108 #endif
109 //@@@ A terminer
110 }
111
112
113 void OSD_Disk::SetQuotaOff(){
114 //int status;
115 #ifdef ULTRIX
116 // status = setquota("","");
117 #else
118 // status = quotactl(Q_QUOTAOFF,"",0,NULL);
119 #endif
120 //@@@ A faire
121 }
122
123 void OSD_Disk::SetQuotaOn(){
124 //TCollection_AsciiString quota_file="????";
125 //int status;
126 #ifdef ULTRIX
127 // status = setquota("",quota_file);
128 #else
129 // status = quotactl(Q_QUOTAON,"",0,quota_file);
130 #endif
131 //@@@ A faire
132 }
133
134
135 void OSD_Disk::Reset(){
136  myError.Reset();
137 }
138
139 Standard_Boolean OSD_Disk::Failed()const{
140  return( myError.Failed());
141 }
142
143 void OSD_Disk::Perror() {
144  myError.Perror();
145 }
146
147
148 Standard_Integer OSD_Disk::Error()const{
149  return( myError.Error());
150 }
151
152 #else
153
154 //-------------------------------------------------------------------------------
155 //---------------------------- Windows NT System --------------------------------
156 //-------------------------------------------------------------------------------
157
158 #include <OSD_Disk.hxx>
159 #include <OSD_OSDError.hxx>
160 #include <OSD_Path.hxx>
161 #include <Standard_ProgramError.hxx>
162 #include <NCollection_String.hxx>
163 #include <TCollection_ExtendedString.hxx>
164
165 #include <windows.h>
166
167 void _osd_wnt_set_error ( OSD_Error&, OSD_WhoAmI, ... );
168
169 static void __fastcall _osd_wnt_set_disk_name ( TCollection_AsciiString&, const OSD_Path& );
170
171 OSD_Disk :: OSD_Disk () {
172  DWORD aBuffLen = GetCurrentDirectoryW(0, NULL);
173  wchar_t* aBuff = new wchar_t[size_t(aBuffLen) + 1];
174  GetCurrentDirectoryW(aBuffLen, aBuff);
175  aBuff[aBuffLen - 1] = (aBuff[aBuffLen - 2] == L'\\') ? L'\0' : L'\\';
176  aBuff[aBuffLen] = L'\0';
177  if (aBuffLen > 3 && aBuff[0] != L'\\')
178  {
179    aBuff[3] = L'\0';
180    DiskName = TCollection_AsciiString (aBuff);
181    delete[] aBuff;
182  }
183  else
184  {
185    DiskName = "";
186  }
187 }  // end constructor ( 1 )
188
189 OSD_Disk :: OSD_Disk ( const OSD_Path& Name ) {
190
191  _osd_wnt_set_disk_name ( DiskName, Name );
192
193 }  // end constructor ( 2 )
194
195 OSD_Disk :: OSD_Disk ( const Standard_CString PathName ) {
196
197  OSD_Path path ( PathName );
198
199  _osd_wnt_set_disk_name ( DiskName, path );
200
201 }  // end constructor ( 3 )
202
203 OSD_Path OSD_Disk :: Name () const {
204
205  return DiskName;
206
207 }  // end OSD_Disk :: Name
208
209 void OSD_Disk :: SetName ( const OSD_Path& Name ) {
210
211  DiskName = Name.Disk ();
212
213 }  // end OSD_Disk :: SetName
214
215 Standard_Integer OSD_Disk :: DiskSize () {
216
217  Standard_Integer retVal = 0;
218
219
220 //  DWORD            dwSpC;
221 //  DWORD            dwBpS;
222 //  DWORD            dwFC;
223 //  DWORD            dwC;
224
225 //  if (   !GetDiskFreeSpace (  DiskName.ToCString (), &dwSpC, &dwBpS, &dwFC, &dwC  )   )
226
227   ULARGE_INTEGER lpFreeBytesAvailableToCaller; // receives the number of bytes on
228                                                 // disk available to the caller
229   ULARGE_INTEGER lpTotalNumberOfBytes;    // receives the number of bytes on disk
230   ULARGE_INTEGER lpTotalNumberOfFreeBytes;// receives the free bytes on disk
231
232   TCollection_ExtendedString DiskNameW(DiskName);
233   if (!GetDiskFreeSpaceExW (DiskNameW.ToWideString(),
234                            &lpFreeBytesAvailableToCaller,
235                            &lpTotalNumberOfBytes,
236                            &lpTotalNumberOfFreeBytes))
237     
238     _osd_wnt_set_error ( myError, OSD_WDisk );
239   
240   else {
241     
242     ULONGLONG  aSize = lpTotalNumberOfBytes.QuadPart /512;
243    
244     retVal = ( Standard_Integer ) aSize; // may be an overflow
245     
246     // retVal = ( Standard_Integer )( dwSpC * dwBpS * dwFC );
247   }
248
249
250  return retVal;
251
252 }  // end OSD_Disk :: DiskSize
253
254 Standard_Integer OSD_Disk :: DiskFree () {
255
256  Standard_Integer retVal = -1;
257
258
259 //  DWORD            dwSpC;
260 //  DWORD            dwBpS;
261 //  DWORD            dwFC;
262 //  DWORD            dwC;
263
264   ULARGE_INTEGER lpFreeBytesAvailableToCaller; // receives the number of bytes on
265                                                 // disk available to the caller
266   ULARGE_INTEGER lpTotalNumberOfBytes;    // receives the number of bytes on disk
267   ULARGE_INTEGER lpTotalNumberOfFreeBytes;// receives the free bytes on disk
268
269   // if (   !GetDiskFreeSpace (  DiskName.ToCString (), &dwSpC, &dwBpS, &dwFC, &dwC  )   )
270   TCollection_ExtendedString DiskNameW(DiskName);
271   if (!GetDiskFreeSpaceExW (DiskNameW.ToWideString(),
272                            &lpFreeBytesAvailableToCaller,
273                            &lpTotalNumberOfBytes,
274                            &lpTotalNumberOfFreeBytes))
275     
276     _osd_wnt_set_error ( myError, OSD_WDisk );
277   
278   else {
279     
280     ULONGLONG  aSize = lpFreeBytesAvailableToCaller.QuadPart /512;
281     
282     retVal = ( Standard_Integer ) aSize; // may be an overflow
283
284     //  retVal = ( Standard_Integer )( dwSpC * dwBpS * dwFC );
285   }
286
287  return retVal;
288
289 }  // end OSD_Disk :: DiskFree
290
291 Standard_Integer OSD_Disk :: DiskQuota () {
292
293  return DiskSize ();
294
295 }  // end OSD_Disk :: DiskQuota
296
297 void OSD_Disk :: SetDiskQuota ( const Standard_Integer /*QuotaSize*/ ) {
298
299  SetLastError (  ( DWORD )STG_E_UNIMPLEMENTEDFUNCTION  );
300
301  _osd_wnt_set_error ( myError, OSD_WDisk );
302
303 }  // end OSD_Disk :: SetDiskQuota
304
305 void OSD_Disk :: SetQuotaOn () {
306
307  SetLastError (  ( DWORD )STG_E_UNIMPLEMENTEDFUNCTION  );
308
309  _osd_wnt_set_error ( myError, OSD_WDisk );
310
311 }  // end OSD_Disk :: SetQuotaOn
312
313 void OSD_Disk :: SetQuotaOff () {
314
315  SetLastError (  ( DWORD )STG_E_UNIMPLEMENTEDFUNCTION  );
316
317  _osd_wnt_set_error ( myError, OSD_WDisk );
318
319 }  // end OSD_Disk :: SetQuotaOff
320
321 Standard_Boolean OSD_Disk :: Failed () const {
322
323  return myError.Failed ();
324
325 }  // end OSD_Disk :: Failed
326
327 void OSD_Disk :: Reset () {
328
329  myError.Reset ();
330
331 }  // end OSD_Disk :: Reset
332
333 void OSD_Disk :: Perror () {
334
335  myError.Perror ();
336
337 }  // end OSD_Disk :: Perror
338
339 Standard_Integer OSD_Disk :: Error () const {
340
341  return myError.Error ();
342
343 }  // end OSD_Disk :: Error
344
345 static void __fastcall _osd_wnt_set_disk_name ( TCollection_AsciiString& result, const OSD_Path& path ) {
346
347  TCollection_AsciiString dir;
348
349  result = path.Disk ();
350
351  if (  result.UsefullLength () == 0  ) {
352  
353   int i, j, k;
354
355   dir = path.Trek ();
356   
357   if (   (  j = dir.UsefullLength ()  ) > 2 &&
358          dir.Value ( 1 ) == '|'     &&
359          dir.Value ( 2 ) == '|'
360   ) {
361   
362    dir.SetValue (  1, '\\');
363    dir.SetValue (  2, '\\');
364    
365    for ( i = 3, k = 0; i <= j; ++i )
366
367     if (  dir.Value ( i ) == '|') {
368     
369      if ( k == 0 ) {
370
371       dir.SetValue (  i, '\\');
372       ++k;
373       continue; 
374      
375      }  // end if
376
377      dir.SetValue (  i, '\\');
378      break;
379      
380     }  /* end if */
381
382     if ( k == 0 )
383     {
384      if (  path.Name ().UsefullLength () == 0 && path.Extension ().UsefullLength () == 0  )
385      
386       goto badPath;
387
388      else {
389
390       dir += '\\';
391       dir += path.Name ();
392       dir += path.Extension ();
393       
394      }  // end else    
395     }
396
397     if (   dir.Value (  dir.UsefullLength ()  ) != '\\') dir += '\\';
398
399     result = dir;
400   
401   } else {
402 badPath:  
403    throw Standard_ProgramError ( "OSD_Disk: bad disk name" );
404
405   }  // end else
406  
407  } else result += '/';
408
409 }  // end _osd_set_disk_name
410
411 #endif