| OLD | NEW |
| (Empty) |
| 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 | |
| 11 #include "base/macros.h" | |
| 12 | |
| 13 // Class to convert a HANDLE to a FILE *. The FILE * is closed when the | |
| 14 // object goes out of scope | |
| 15 class HandleToFile { | |
| 16 public: | |
| 17 HandleToFile() { | |
| 18 file_ = NULL; | |
| 19 }; | |
| 20 | |
| 21 // Note: c_file_handle_ does not need to be closed because fclose does it. | |
| 22 ~HandleToFile() { | |
| 23 if (file_) { | |
| 24 fflush(file_); | |
| 25 fclose(file_); | |
| 26 } | |
| 27 }; | |
| 28 | |
| 29 // Translates a HANDLE (handle) to a FILE * opened with the mode "mode". | |
| 30 // The return value is the FILE * or NULL if there is an error. | |
| 31 FILE* Translate(HANDLE handle, const char *mode) { | |
| 32 if (file_) { | |
| 33 return NULL; | |
| 34 } | |
| 35 | |
| 36 HANDLE new_handle; | |
| 37 BOOL result = ::DuplicateHandle(::GetCurrentProcess(), | |
| 38 handle, | |
| 39 ::GetCurrentProcess(), | |
| 40 &new_handle, | |
| 41 0, // Don't ask for a specific | |
| 42 // desired access. | |
| 43 FALSE, // Not inheritable. | |
| 44 DUPLICATE_SAME_ACCESS); | |
| 45 | |
| 46 if (!result) { | |
| 47 return NULL; | |
| 48 } | |
| 49 | |
| 50 int c_file_handle = _open_osfhandle(reinterpret_cast<LONG_PTR>(new_handle), | |
| 51 0); // No flags | |
| 52 if (-1 == c_file_handle) { | |
| 53 return NULL; | |
| 54 } | |
| 55 | |
| 56 file_ = _fdopen(c_file_handle, mode); | |
| 57 return file_; | |
| 58 }; | |
| 59 private: | |
| 60 // the FILE* returned. We need to closed it at the end. | |
| 61 FILE* file_; | |
| 62 | |
| 63 DISALLOW_COPY_AND_ASSIGN(HandleToFile); | |
| 64 }; | |
| 65 | |
| 66 #endif // SANDBOX_SANDBOX_POC_POCDLL_UTILS_H__ | |
| OLD | NEW |