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
10namespace content {
11
12void PluginStream::ResetTempFileHandle() {
13  temp_file_handle_ = INVALID_HANDLE_VALUE;
14}
15
16void PluginStream::ResetTempFileName() {
17  temp_file_name_[0] = '\0';
18}
19
20void PluginStream::WriteAsFile() {
21  if (RequestedPluginModeIsAsFile())
22    instance_->NPP_StreamAsFile(&stream_, temp_file_name_);
23}
24
25size_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
34bool 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
67void PluginStream::CloseTempFile() {
68  if (!TempFileIsValid())
69    return;
70
71  CloseHandle(temp_file_handle_);
72  ResetTempFileHandle();
73}
74
75bool PluginStream::TempFileIsValid() const {
76  return temp_file_handle_ != INVALID_HANDLE_VALUE;
77}
78
79}  // namespace content
80