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 #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__ | |
OLD | NEW |