Source code

Revision control

Other Tools

/* -*- 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;
}