зеркало из https://github.com/mozilla/gecko-dev.git
128 строки
4.2 KiB
C++
128 строки
4.2 KiB
C++
/* -*- Mode: C++; tab-width: 2; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
|
|
/* This Source Code Form is subject to the terms of the Mozilla Public
|
|
* License, v. 2.0. If a copy of the MPL was not distributed with this
|
|
* file, You can obtain one at http://mozilla.org/MPL/2.0/. */
|
|
|
|
#include "nsPrintSettingsServiceWin.h"
|
|
|
|
#include "nsCOMPtr.h"
|
|
#include "nsPrintSettingsWin.h"
|
|
#include "nsPrintDialogUtil.h"
|
|
|
|
#include "nsGfxCIID.h"
|
|
#include "nsIServiceManager.h"
|
|
#include "nsWindowsHelpers.h"
|
|
#include "ipc/IPCMessageUtils.h"
|
|
|
|
const char kPrinterListContractID[] = "@mozilla.org/gfx/printerlist;1";
|
|
|
|
using namespace mozilla::embedding;
|
|
|
|
NS_IMETHODIMP
|
|
nsPrintSettingsServiceWin::SerializeToPrintData(nsIPrintSettings* aSettings,
|
|
PrintData* data) {
|
|
nsresult rv = nsPrintSettingsService::SerializeToPrintData(aSettings, data);
|
|
NS_ENSURE_SUCCESS(rv, rv);
|
|
|
|
nsCOMPtr<nsIPrintSettingsWin> psWin = do_QueryInterface(aSettings);
|
|
if (!psWin) {
|
|
return NS_ERROR_FAILURE;
|
|
}
|
|
|
|
nsAutoString deviceName;
|
|
nsAutoString driverName;
|
|
|
|
psWin->GetDeviceName(deviceName);
|
|
psWin->GetDriverName(driverName);
|
|
|
|
data->deviceName().Assign(deviceName);
|
|
data->driverName().Assign(driverName);
|
|
|
|
// When creating the print dialog on Windows, we only need to send certain
|
|
// print settings information from the parent to the child not vice versa.
|
|
if (XRE_IsParentProcess()) {
|
|
// A DEVMODE can actually be of arbitrary size. If it turns out that it'll
|
|
// make our IPC message larger than the limit, then we'll error out.
|
|
LPDEVMODEW devModeRaw;
|
|
psWin->GetDevMode(&devModeRaw); // This actually allocates a copy of the
|
|
// the nsIPrintSettingsWin DEVMODE, so
|
|
// we're now responsible for deallocating
|
|
// it. We'll use an nsAutoDevMode helper
|
|
// to do this.
|
|
if (devModeRaw) {
|
|
nsAutoDevMode devMode(devModeRaw);
|
|
devModeRaw = nullptr;
|
|
|
|
size_t devModeTotalSize = devMode->dmSize + devMode->dmDriverExtra;
|
|
size_t msgTotalSize = sizeof(PrintData) + devModeTotalSize;
|
|
|
|
if (msgTotalSize > IPC::Channel::kMaximumMessageSize / 2) {
|
|
return NS_ERROR_FAILURE;
|
|
}
|
|
|
|
// Instead of reaching in and manually reading each member, we'll just
|
|
// copy the bits over.
|
|
const char* devModeData = reinterpret_cast<const char*>(devMode.get());
|
|
nsTArray<uint8_t> arrayBuf;
|
|
arrayBuf.AppendElements(devModeData, devModeTotalSize);
|
|
data->devModeData() = std::move(arrayBuf);
|
|
}
|
|
}
|
|
|
|
return NS_OK;
|
|
}
|
|
|
|
NS_IMETHODIMP
|
|
nsPrintSettingsServiceWin::DeserializeToPrintSettings(
|
|
const PrintData& data, nsIPrintSettings* settings) {
|
|
nsresult rv =
|
|
nsPrintSettingsService::DeserializeToPrintSettings(data, settings);
|
|
NS_ENSURE_SUCCESS(rv, rv);
|
|
|
|
nsCOMPtr<nsIPrintSettingsWin> psWin = do_QueryInterface(settings);
|
|
if (!settings) {
|
|
return NS_ERROR_FAILURE;
|
|
}
|
|
|
|
if (XRE_IsContentProcess()) {
|
|
psWin->SetDeviceName(data.deviceName());
|
|
psWin->SetDriverName(data.driverName());
|
|
|
|
if (data.devModeData().IsEmpty()) {
|
|
psWin->SetDevMode(nullptr);
|
|
} else {
|
|
// Check minimum length of DEVMODE data.
|
|
auto devModeDataLength = data.devModeData().Length();
|
|
if (devModeDataLength < sizeof(DEVMODEW)) {
|
|
NS_WARNING("DEVMODE data is too short.");
|
|
return NS_ERROR_FAILURE;
|
|
}
|
|
|
|
DEVMODEW* devMode = reinterpret_cast<DEVMODEW*>(
|
|
const_cast<uint8_t*>(data.devModeData().Elements()));
|
|
|
|
// Check actual length of DEVMODE data.
|
|
if ((devMode->dmSize + devMode->dmDriverExtra) != devModeDataLength) {
|
|
NS_WARNING("DEVMODE length is incorrect.");
|
|
return NS_ERROR_FAILURE;
|
|
}
|
|
|
|
psWin->SetDevMode(devMode); // Copies
|
|
}
|
|
}
|
|
|
|
return NS_OK;
|
|
}
|
|
|
|
nsresult nsPrintSettingsServiceWin::_CreatePrintSettings(
|
|
nsIPrintSettings** _retval) {
|
|
*_retval = nullptr;
|
|
nsPrintSettingsWin* printSettings =
|
|
new nsPrintSettingsWin(); // does not initially ref count
|
|
NS_ENSURE_TRUE(printSettings, NS_ERROR_OUT_OF_MEMORY);
|
|
|
|
NS_ADDREF(*_retval = printSettings); // ref count
|
|
|
|
return NS_OK;
|
|
}
|