IndexedDB: Protect against use-after-free in ChainedBlobWriter.
[chromium-blink-merge.git] / printing / backend / printing_info_win.h
blob11341943ce2bdd6d508f93263972ab1599743688
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.
5 #ifndef PRINTING_BACKEND_PRINTING_INFO_WIN_H_
6 #define PRINTING_BACKEND_PRINTING_INFO_WIN_H_
8 #include <objidl.h>
9 #include <winspool.h>
11 #include "base/memory/scoped_ptr.h"
12 #include "printing/printing_export.h"
14 namespace printing {
16 namespace internal {
18 PRINTING_EXPORT uint8* GetDriverInfo(HANDLE printer, int level);
19 PRINTING_EXPORT uint8* GetPrinterInfo(HANDLE printer, int level);
21 // This class is designed to work with PRINTER_INFO_X structures
22 // and calls GetPrinter internally with correctly allocated buffer.
23 template <typename PrinterInfoType, int level>
24 class PrinterInfo {
25 public:
26 bool Init(HANDLE printer) {
27 buffer_.reset(GetPrinterInfo(printer, level));
28 return buffer_;
31 const PrinterInfoType* get() const {
32 return reinterpret_cast<const PrinterInfoType*>(buffer_.get());
35 private:
36 scoped_ptr<uint8[]> buffer_;
39 // This class is designed to work with DRIVER_INFO_X structures
40 // and calls GetDriverInfo internally with correctly allocated buffer.
41 template <typename DriverInfoType, int level>
42 class DriverInfo {
43 public:
44 bool Init(HANDLE printer) {
45 buffer_.reset(GetDriverInfo(printer, level));
46 return buffer_;
49 const DriverInfoType* get() const {
50 return reinterpret_cast<const DriverInfoType*>(buffer_.get());
53 private:
54 scoped_ptr<uint8[]> buffer_;
57 } // namespace internal
59 typedef internal::PrinterInfo<PRINTER_INFO_2, 2> PrinterInfo2;
60 typedef internal::PrinterInfo<PRINTER_INFO_5, 5> PrinterInfo5;
62 typedef internal::DriverInfo<DRIVER_INFO_6, 6> DriverInfo6;
64 } // namespace printing
66 #endif // PRINTING_BACKEND_PRINTING_INFO_WIN_H_