daemon/sync.c

   1 /* libguestfs - the guestfsd daemon
   2  * Copyright (C) 2009 Red Hat Inc.
   3  *
   4  * This program is free software; you can redistribute it and/or modify
   5  * it under the terms of the GNU General Public License as published by
   6  * the Free Software Foundation; either version 2 of the License, or
   7  * (at your option) any later version.
   8  *
   9  * This program is distributed in the hope that it will be useful,
  10  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  11  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  12  * GNU General Public License for more details.
  13  *
  14  * You should have received a copy of the GNU General Public License
  15  * along with this program; if not, write to the Free Software
  16  * Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
  17  */
  18
  19 #include <config.h>
  20
  21 #ifdef HAVE_WINDOWS_H
  22 #include <windows.h>
  23 #endif
  24
  25 #include <stdio.h>
  26 #include <unistd.h>
  27
  28 #include "daemon.h"
  29 #include "actions.h"
  30
  31 #ifdef WIN32
  32 static int sync_win32 (void);
  33 #endif
  34
  35 int
  36 do_sync (void)
  37 {
  38   if (sync_disks () == -1) {
  39     reply_with_perror ("sync");
  40     return -1;
  41   }
  42
  43   return 0;
  44 }
  45
  46 /* This is a replacement for sync(2) which is called from
  47  * this file and from other places in the daemon.  It works
  48  * on Windows too.
  49  */
  50 int
  51 sync_disks (void)
  52 {
  53 #if defined(HAVE_SYNC)
  54   return sync ();
  55 #elif defined(WIN32)
  56   return sync_win32 ();
  57 #else
  58 #error "no known sync() API"
  59 #endif
  60 }
  61
  62 #ifdef WIN32
  63 static int
  64 sync_win32 (void)
  65 {
  66   DWORD n1, n2;
  67
  68   n1 = GetLogicalDriveStrings (0, NULL);
  69   if (n1 == 0)
  70     return -1;
  71
  72   TCHAR buffer[n1+2]; /* sic */
  73   n2 = GetLogicalDriveStrings (n1, buffer);
  74   if (n2 == 0)
  75     return -1;
  76
  77   TCHAR *p = buffer;
  78
  79   /* The MSDN example code itself assumes that there is always one
  80    * drive in the system.  However we will be better than that and not
  81    * make the assumption ...
  82    */
  83   while (*p) {
  84     HANDLE drive;
  85     DWORD drive_type;
  86
  87     if (verbose)
  88       fprintf (stderr, "sync_win32: examining drive %s\n", p);
  89
  90     /* Ignore removable drives. */
  91     drive_type = GetDriveType (p);
  92     if (drive_type == DRIVE_FIXED) {
  93       /* To open the volume you have to specify the volume name, not
  94        * the mount point.  MSDN documents use of the constant 50
  95        * below.
  96        */
  97       TCHAR volname[50];
  98       if (!GetVolumeNameForVolumeMountPoint (p, volname, 50))
  99         return -1;
 100
 101       drive = CreateFile (volname, GENERIC_READ|GENERIC_WRITE,
 102                           FILE_SHARE_READ|FILE_SHARE_WRITE,
 103                           NULL, OPEN_EXISTING, 0, 0);
 104       if (drive == INVALID_HANDLE_VALUE)
 105         return -1;
 106       if (verbose)
 107         fprintf (stderr, "sync_win32: flushing %s\n", volname);
 108
 109       BOOL r;
 110       /* This always fails in Wine:
 111        * http://bugs.winehq.org/show_bug.cgi?id=14915
 112        */
 113       r = FlushFileBuffers (drive);
 114       CloseHandle (drive);
 115       if (!r)
 116         return -1;
 117     }
 118
 119     /* Skip to next \0 character. */
 120     while (*p++);
 121   }
 122
 123   return 0;
 124 }
 125 #endif /* WIN32 */