1 // Copyright (c) 2012 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 #include "content/child/npapi/plugin_stream.h"
6
7 #include "base/logging.h"
8 #include "content/child/npapi/plugin_instance.h"
9
10 namespace content {
11
ResetTempFileHandle()12 void PluginStream::ResetTempFileHandle() {
13 temp_file_handle_ = INVALID_HANDLE_VALUE;
14 }
15
ResetTempFileName()16 void PluginStream::ResetTempFileName() {
17 temp_file_name_[0] = '\0';
18 }
19
WriteAsFile()20 void PluginStream::WriteAsFile() {
21 if (RequestedPluginModeIsAsFile())
22 instance_->NPP_StreamAsFile(&stream_, temp_file_name_);
23 }
24
WriteBytes(const char * buf,size_t length)25 size_t PluginStream::WriteBytes(const char *buf, size_t length) {
26 DWORD bytes;
27
28 if (!WriteFile(temp_file_handle_, buf, length, &bytes, 0))
29 return 0U;
30
31 return static_cast<size_t>(bytes);
32 }
33
OpenTempFile()34 bool PluginStream::OpenTempFile() {
35 DCHECK_EQ(INVALID_HANDLE_VALUE, temp_file_handle_);
36
37 // The reason for using all the Ascii versions of these filesystem
38 // calls is that the filename which we pass back to the plugin
39 // via NPAPI is an ascii filename. Otherwise, we'd use wide-chars.
40 //
41 // TODO:
42 // This is a bug in NPAPI itself, and it needs to be fixed.
43 // The case which will fail is if a user has a multibyte name,
44 // but has the system locale set to english. GetTempPathA will
45 // return junk in this case, causing us to be unable to open the
46 // file.
47
48 char temp_directory[MAX_PATH];
49 if (GetTempPathA(MAX_PATH, temp_directory) == 0)
50 return false;
51 if (GetTempFileNameA(temp_directory, "npstream", 0, temp_file_name_) == 0)
52 return false;
53 temp_file_handle_ = CreateFileA(temp_file_name_,
54 FILE_ALL_ACCESS,
55 FILE_SHARE_READ,
56 0,
57 CREATE_ALWAYS,
58 FILE_ATTRIBUTE_NORMAL,
59 0);
60 if (temp_file_handle_ == INVALID_HANDLE_VALUE) {
61 ResetTempFileName();
62 return false;
63 }
64 return true;
65 }
66
CloseTempFile()67 void PluginStream::CloseTempFile() {
68 if (!TempFileIsValid())
69 return;
70
71 CloseHandle(temp_file_handle_);
72 ResetTempFileHandle();
73 }
74
TempFileIsValid() const75 bool PluginStream::TempFileIsValid() const {
76 return temp_file_handle_ != INVALID_HANDLE_VALUE;
77 }
78
79 } // namespace content
80