sandbox/win/sandbox_poc/pocdll/utils.h

   1 // Copyright (c) 2010 The Chromium Authors. All rights reserved.
   2 // Use of this source code is governed by a BSD-style license that can be
   3 // found in the LICENSE file.
   4
   5 #ifndef SANDBOX_SANDBOX_POC_POCDLL_UTILS_H__
   6 #define SANDBOX_SANDBOX_POC_POCDLL_UTILS_H__
   7
   8 #include <stdio.h>
   9 #include <io.h>
  10 #include "base/basictypes.h"
  11
  12 // Class to convert a HANDLE to a FILE *. The FILE * is closed when the
  13 // object goes out of scope
  14 class HandleToFile {
  15  public:
  16   HandleToFile() {
  17     file_ = NULL;
  18   };
  19
  20   // Note: c_file_handle_ does not need to be closed because fclose does it.
  21   ~HandleToFile() {
  22     if (file_) {
  23       fflush(file_);
  24       fclose(file_);
  25     }
  26   };
  27
  28   // Translates a HANDLE (handle) to a FILE * opened with the mode "mode".
  29   // The return value is the FILE * or NULL if there is an error.
  30   FILE* Translate(HANDLE handle, const char *mode) {
  31     if (file_) {
  32       return  NULL;
  33     }
  34
  35     HANDLE new_handle;
  36     BOOL result = ::DuplicateHandle(::GetCurrentProcess(),
  37                                     handle,
  38                                     ::GetCurrentProcess(),
  39                                     &new_handle,
  40                                     0,  // Don't ask for a specific
  41                                         // desired access.
  42                                     FALSE,  // Not inheritable.
  43                                     DUPLICATE_SAME_ACCESS);
  44
  45     if (!result) {
  46       return NULL;
  47     }
  48
  49     int c_file_handle = _open_osfhandle(reinterpret_cast<LONG_PTR>(new_handle),
  50                                         0);  // No flags
  51     if (-1 == c_file_handle) {
  52       return NULL;
  53     }
  54
  55     file_ = _fdopen(c_file_handle, mode);
  56     return file_;
  57   };
  58  private:
  59   // the FILE* returned. We need to closed it at the end.
  60   FILE* file_;
  61
  62   DISALLOW_COPY_AND_ASSIGN(HandleToFile);
  63 };
  64
  65 #endif  // SANDBOX_SANDBOX_POC_POCDLL_UTILS_H__