ikvm-fork/awt/printing.cs

/*
  Copyright (C) 2009 Volker Berlin (i-net software)

  This software is provided 'as-is', without any express or implied
  warranty.  In no event will the authors be held liable for any damages
  arising from the use of this software.

  Permission is granted to anyone to use this software for any purpose,
  including commercial applications, and to alter it and redistribute it
  freely, subject to the following restrictions:

  1. The origin of this software must not be misrepresented; you must not
     claim that you wrote the original software. If you use this software
     in a product, an acknowledgment in the product documentation would be
     appreciated but is not required.
  2. Altered source versions must be plainly marked as such, and must not be
     misrepresented as being the original software.
  3. This notice may not be removed or altered from any source distribution.

  Jeroen Frijters
  jeroen@frijters.net 

*/

using System;
using System.Runtime.InteropServices;
using Microsoft.Win32.SafeHandles;

namespace ikvm.awt.printing
{
    /// <summary>
    /// Base Implementation of the PrintPeer
    /// </summary>
    abstract class BasePrintPeer : sun.print.PrintPeer
    {
        public virtual Object getPrinterStatus(String PrinterName, java.lang.Class category)
        {
            return null;
        }
    }

    /// <summary>
    /// Implementation of the PrintPeer for Linux
    /// </summary>
    class LinuxPrintPeer : BasePrintPeer
    {
    }

    /// <summary>
    /// Implementation of the PrintPeer for Windows
    /// </summary>
    class Win32PrintPeer : BasePrintPeer
    {
        [StructLayout(LayoutKind.Sequential, CharSet = CharSet.Auto)]
        private struct PRINTER_INFO_2
        {
            [MarshalAs(UnmanagedType.LPTStr)]
            public string pServerName;
            [MarshalAs(UnmanagedType.LPTStr)]
            public string pPrinterName;
            [MarshalAs(UnmanagedType.LPTStr)]
            public string pShareName;
            [MarshalAs(UnmanagedType.LPTStr)]
            public string pPortName;
            [MarshalAs(UnmanagedType.LPTStr)]
            public string pDriverName;
            [MarshalAs(UnmanagedType.LPTStr)]
            public string pComment;
            [MarshalAs(UnmanagedType.LPTStr)]
            public string pLocation;
            public IntPtr pDevMode;
            [MarshalAs(UnmanagedType.LPTStr)]
            public string pSepFile;
            [MarshalAs(UnmanagedType.LPTStr)]
            public string pPrintProcessor;
            [MarshalAs(UnmanagedType.LPTStr)]
            public string pDatatype;
            [MarshalAs(UnmanagedType.LPTStr)]
            public string pParameters;
            public IntPtr pSecurityDescriptor;
            public uint Attributes;
            public uint Priority;
            public uint DefaultPriority;
            public uint StartTime;
            public uint UntilTime;
            public int Status;
            public int cJobs;
            public uint AveragePPM;
        }

        [DllImport("winspool.drv", CharSet = CharSet.Auto, SetLastError = true)]
        private static extern bool GetPrinter(IntPtr hPrinter, int dwLevel,
            IntPtr pPrinter, int cbBuf, out int pcbNeeded);

        private const int ERROR_INSUFFICIENT_BUFFER = 122;

        private const int PRINTER_STATUS_DOOR_OPEN = 0x400000;
        private const int PRINTER_STATUS_ERROR = 0x2;
        private const int PRINTER_STATUS_NO_TONER = 0x40000;
        private const int PRINTER_STATUS_OFFLINE = 0x80;
        private const int PRINTER_STATUS_OUT_OF_MEMORY = 0x200000;
        private const int PRINTER_STATUS_USER_INTERVENTION = 0x100000;

        /// <summary>
        /// Get a printer status
        /// </summary>
        /// <param name="printerName">a valid printer name</param>
        /// <param name="category">a category that should request</param>
        /// <returns>a printer attribute or null</returns>
        public override Object getPrinterStatus(String printerName, java.lang.Class category)
        {
            int cJobs;
            int status;
            if (GetPrinterInfo2(printerName, out cJobs, out status))
            {
                if (category == (java.lang.Class)typeof(javax.print.attribute.standard.QueuedJobCount))
                {
                    return new javax.print.attribute.standard.QueuedJobCount(cJobs);
                }
                if (category == (java.lang.Class)typeof(javax.print.attribute.standard.PrinterIsAcceptingJobs))
                {
                    if ((status &
                            (PRINTER_STATUS_ERROR |
                             PRINTER_STATUS_NO_TONER |
                             PRINTER_STATUS_OUT_OF_MEMORY |
                             PRINTER_STATUS_OFFLINE |
                             PRINTER_STATUS_USER_INTERVENTION |
                             PRINTER_STATUS_DOOR_OPEN)) > 0)
                    {
                        return javax.print.attribute.standard.PrinterIsAcceptingJobs.NOT_ACCEPTING_JOBS;
                    }
                    else
                    {
                        return javax.print.attribute.standard.PrinterIsAcceptingJobs.ACCEPTING_JOBS;
                    }
                }
            }
            return null;
        }

        /// <summary>
        /// Get infos from the PRINTER_INFO_2 struc
        /// </summary>
        /// <param name="printerName">The name of a valid printer</param>
        /// <param name="cJobs">returns the current count of print jobs</param>
        /// <param name="status">returns the current status of the printer</param>
        /// <returns>true if the return is valid</returns>
        private static bool GetPrinterInfo2(string printerName, out int cJobs, out int status)
        {
            using (SafePrinterHandle printer = new SafePrinterHandle(printerName))
            {
                int needed = 0;
                if (!GetPrinter(printer.DangerousGetHandle(), 2, IntPtr.Zero, 0, out needed))
                {
                    int lastWin32Error = Marshal.GetLastWin32Error();
                    if (lastWin32Error == ERROR_INSUFFICIENT_BUFFER)
                    {
                        IntPtr pPrinter = Marshal.AllocHGlobal((int)needed);
                        try
                        {

                            if (GetPrinter(printer.DangerousGetHandle(), 2, pPrinter, needed, out needed))
                            {
                                PRINTER_INFO_2 printerInfo2 = (PRINTER_INFO_2)Marshal.PtrToStructure(pPrinter, typeof(PRINTER_INFO_2));
                                cJobs = printerInfo2.cJobs;
                                status = printerInfo2.Status;
                                return true;
                            }

                        }
                        finally
                        {
                            Marshal.FreeHGlobal(pPrinter);
                        }
                    }
                }
            }
            cJobs = 0;
            status = 0;
            return false;
        }

    }

    /// <summary>
    /// SafeHandle to an open printer
    /// </summary>
    class SafePrinterHandle : SafeHandleZeroOrMinusOneIsInvalid
    {
        [DllImport("winspool.drv", CharSet = CharSet.Auto, SetLastError = true)]
        private static extern bool OpenPrinter(string pPrinterName, out IntPtr phPrinter, IntPtr pDefault);

        [DllImport("winspool.drv", CharSet = CharSet.Auto, SetLastError = true)]
        private static extern bool ClosePrinter(IntPtr hPrinter);

        /// <summary>
        /// Create a safe printer handle for given name
        /// </summary>
        /// <param name="printername"></param>
        internal SafePrinterHandle(string printername)
            : base(true)
        {
            OpenPrinter(printername, out this.handle, IntPtr.Zero);
        }


        [return: MarshalAs(UnmanagedType.U1)]
        protected override bool ReleaseHandle()
        {
            // Only close printer if handle is not invalid
            if (IsInvalid)
                return true;

            if (!ClosePrinter(this.handle))
                return false;

            // Make the handle invalid
            this.SetHandle(IntPtr.Zero);
            return true;
        }
    }
    
}