jdk/src/windows/classes/sun/print/Win32PrintServiceLookup.java - platform/libcore - Gitiles

 /*
  * Copyright 2000-2007 Sun Microsystems, Inc.  All Rights Reserved.
  * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
  *
  * This code is free software; you can redistribute it and/or modify it
  * under the terms of the GNU General Public License version 2 only, as
  * published by the Free Software Foundation.  Sun designates this
  * particular file as subject to the "Classpath" exception as provided
  * by Sun in the LICENSE file that accompanied this code.
  *
  * This code is distributed in the hope that it will be useful, but WITHOUT
  * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
  * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License
  * version 2 for more details (a copy is included in the LICENSE file that
  * accompanied this code).
  *
  * You should have received a copy of the GNU General Public License version
  * 2 along with this work; if not, write to the Free Software Foundation,
  * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
  *
  * Please contact Sun Microsystems, Inc., 4150 Network Circle, Santa Clara,
  * CA 95054 USA or visit www.sun.com if you need additional information or
  * have any questions.
  */

 package sun.print;

 import java.io.BufferedReader;
 import java.io.InputStream;
 import java.io.InputStreamReader;
 import java.io.IOException;
 import java.util.ArrayList;
 import java.security.AccessController;
 import java.security.PrivilegedActionException;
 import java.security.PrivilegedExceptionAction;
 import javax.print.DocFlavor;
 import javax.print.MultiDocPrintService;
 import javax.print.PrintService;
 import javax.print.PrintServiceLookup;
 import javax.print.attribute.Attribute;
 import javax.print.attribute.AttributeSet;
 import javax.print.attribute.HashPrintRequestAttributeSet;
 import javax.print.attribute.HashPrintServiceAttributeSet;
 import javax.print.attribute.PrintRequestAttribute;
 import javax.print.attribute.PrintRequestAttributeSet;
 import javax.print.attribute.PrintServiceAttribute;
 import javax.print.attribute.PrintServiceAttributeSet;
 import javax.print.attribute.standard.PrinterName;

 public class Win32PrintServiceLookup extends PrintServiceLookup {

     private String defaultPrinter;
     private PrintService defaultPrintService;
     private String[] printers; /* excludes the default printer */
     private PrintService[] printServices; /* includes the default printer */

     static {
         java.security.AccessController.doPrivileged(
                     new sun.security.action.LoadLibraryAction("awt"));
     }

     /* The singleton win32 print lookup service.
      * Code that is aware of this field and wants to use it must first
      * see if its null, and if so instantiate it by calling a method such as
      * javax.print.PrintServiceLookup.defaultPrintService() so that the
      * same instance is stored there.
      */
     private static Win32PrintServiceLookup win32PrintLUS;

     /* Think carefully before calling this. Preferably don't call it. */
     public static Win32PrintServiceLookup getWin32PrintLUS() {
         if (win32PrintLUS == null) {
             /* This call is internally synchronized.
              * When it returns an instance of this class will have
              * been instantiated - else there's a JDK internal error.
              */
             PrintServiceLookup.lookupDefaultPrintService();
         }
         return win32PrintLUS;
     }

     public Win32PrintServiceLookup() {

         if (win32PrintLUS == null) {
             win32PrintLUS = this;

             String osName = AccessController.doPrivileged(
                 new sun.security.action.GetPropertyAction("os.name"));
             // There's no capability for Win98 to refresh printers.
             // See "OpenPrinter" for more info.
             if (osName != null && osName.startsWith("Windows 98")) {
                 return;
             }
             // start the printer listener thread
             PrinterChangeListener thr = new PrinterChangeListener();
             thr.setDaemon(true);
             thr.start();
         } /* else condition ought to never happen! */
     }

     /* Want the PrintService which is default print service to have
      * equality of reference with the equivalent in list of print services
      * This isn't required by the API and there's a risk doing this will
      * lead people to assume its guaranteed.
      */
     public synchronized PrintService[] getPrintServices() {
         SecurityManager security = System.getSecurityManager();
         if (security != null) {
             security.checkPrintJobAccess();
         }
         if (printServices == null) {
             refreshServices();
         }
         return printServices;
     }

     private synchronized void refreshServices() {
         printers = getAllPrinterNames();
         if (printers == null) {
             // In Windows it is safe to assume no default if printers == null so we
             // don't get the default.
             printServices = new PrintService[0];
             return;
         }

         PrintService[] newServices = new PrintService[printers.length];
         PrintService defService = getDefaultPrintService();
         for (int p = 0; p < printers.length; p++) {
             if (defService != null &&
                 printers[p].equals(defService.getName())) {
                 newServices[p] = defService;
             } else {
                 if (printServices == null) {
                     newServices[p] = new Win32PrintService(printers[p]);
                 } else {
                     int j;
                     for (j = 0; j < printServices.length; j++) {
                         if ((printServices[j]!= null) &&
                             (printers[p].equals(printServices[j].getName()))) {
                             newServices[p] = printServices[j];
                             printServices[j] = null;
                             break;
                         }
                     }
                     if (j == printServices.length) {
                         newServices[p] = new Win32PrintService(printers[p]);
                     }
                 }
             }
         }

         // Look for deleted services and invalidate these
         if (printServices != null) {
             for (int j=0; j < printServices.length; j++) {
                 if ((printServices[j] instanceof Win32PrintService) &&
                     (!printServices[j].equals(defaultPrintService))) {
                     ((Win32PrintService)printServices[j]).invalidateService();
                 }
             }
         }
         printServices = newServices;
     }


     public synchronized PrintService getPrintServiceByName(String name) {

         if (name == null || name.equals("")) {
             return null;
         } else {
             /* getPrintServices() is now very fast. */
             PrintService[] printServices = getPrintServices();
             for (int i=0; i<printServices.length; i++) {
                 if (printServices[i].getName().equals(name)) {
                     return printServices[i];
                 }
             }
             return null;
         }
     }

     boolean matchingService(PrintService service,
                             PrintServiceAttributeSet serviceSet) {
         if (serviceSet != null) {
             Attribute [] attrs =  serviceSet.toArray();
             Attribute serviceAttr;
             for (int i=0; i<attrs.length; i++) {
                 serviceAttr
                     = service.getAttribute((Class<PrintServiceAttribute>)attrs[i].getCategory());
                 if (serviceAttr == null || !serviceAttr.equals(attrs[i])) {
                     return false;
                 }
             }
         }
         return true;
     }

     public PrintService[] getPrintServices(DocFlavor flavor,
                                            AttributeSet attributes) {

         SecurityManager security = System.getSecurityManager();
         if (security != null) {
           security.checkPrintJobAccess();
         }
         PrintRequestAttributeSet requestSet = null;
         PrintServiceAttributeSet serviceSet = null;

         if (attributes != null && !attributes.isEmpty()) {

             requestSet = new HashPrintRequestAttributeSet();
             serviceSet = new HashPrintServiceAttributeSet();

             Attribute[] attrs = attributes.toArray();
             for (int i=0; i<attrs.length; i++) {
                 if (attrs[i] instanceof PrintRequestAttribute) {
                     requestSet.add(attrs[i]);
                 } else if (attrs[i] instanceof PrintServiceAttribute) {
                     serviceSet.add(attrs[i]);
                 }
             }
         }

         /*
          * Special case: If client is asking for a particular printer
          * (by name) then we can save time by getting just that service
          * to check against the rest of the specified attributes.
          */
         PrintService[] services = null;
         if (serviceSet != null && serviceSet.get(PrinterName.class) != null) {
             PrinterName name = (PrinterName)serviceSet.get(PrinterName.class);
             PrintService service = getPrintServiceByName(name.getValue());
             if (service == null || !matchingService(service, serviceSet)) {
                 services = new PrintService[0];
             } else {
                 services = new PrintService[1];
                 services[0] = service;
             }
         } else {
             services = getPrintServices();
         }

         if (services.length == 0) {
             return services;
         } else {
             ArrayList matchingServices = new ArrayList();
             for (int i=0; i<services.length; i++) {
                 try {
                     if (services[i].
                         getUnsupportedAttributes(flavor, requestSet) == null) {
                         matchingServices.add(services[i]);
                     }
                 } catch (IllegalArgumentException e) {
                 }
             }
             services = new PrintService[matchingServices.size()];
             return (PrintService[])matchingServices.toArray(services);
         }
     }

     /*
      * return empty array as don't support multi docs
      */
     public MultiDocPrintService[]
         getMultiDocPrintServices(DocFlavor[] flavors,
                                  AttributeSet attributes) {
         SecurityManager security = System.getSecurityManager();
         if (security != null) {
           security.checkPrintJobAccess();
         }
         return new MultiDocPrintService[0];
     }


     public synchronized PrintService getDefaultPrintService() {
         SecurityManager security = System.getSecurityManager();
         if (security != null) {
           security.checkPrintJobAccess();
         }


         // Windows does not have notification for a change in default
         // so we always get the latest.
         defaultPrinter = getDefaultPrinterName();
         if (defaultPrinter == null) {
             return null;
         }

         if ((defaultPrintService != null) &&
             defaultPrintService.getName().equals(defaultPrinter)) {

             return defaultPrintService;
         }

          // Not the same as default so proceed to get new PrintService.

         // clear defaultPrintService
         defaultPrintService = null;

         if (printServices != null) {
             for (int j=0; j<printServices.length; j++) {
                 if (defaultPrinter.equals(printServices[j].getName())) {
                     defaultPrintService = printServices[j];
                     break;
                 }
             }
         }

         if (defaultPrintService == null) {
             defaultPrintService = new Win32PrintService(defaultPrinter);
         }
         return defaultPrintService;
     }

     class PrinterChangeListener extends Thread {
         long chgObj;
         PrinterChangeListener() {
             chgObj = notifyFirstPrinterChange(null);
         }

         public void run() {
             if (chgObj != -1) {
                 while (true) {
                     // wait for configuration to change
                     if (notifyPrinterChange(chgObj) != 0) {
                         try {
                             refreshServices();
                         } catch (SecurityException se) {
                             break;
                         }
                     } else {
                         notifyClosePrinterChange(chgObj);
                         break;
                     }
                 }
             }
         }
     }

     private native String getDefaultPrinterName();
     private native String[] getAllPrinterNames();
     private native long notifyFirstPrinterChange(String printer);
     private native void notifyClosePrinterChange(long chgObj);
     private native int notifyPrinterChange(long chgObj);
 }
	/*
	* Copyright 2000-2007 Sun Microsystems, Inc. All Rights Reserved.
	* DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
	*
	* This code is free software; you can redistribute it and/or modify it
	* under the terms of the GNU General Public License version 2 only, as
	* published by the Free Software Foundation. Sun designates this
	* particular file as subject to the "Classpath" exception as provided
	* by Sun in the LICENSE file that accompanied this code.
	*
	* This code is distributed in the hope that it will be useful, but WITHOUT
	* ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
	* FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License
	* version 2 for more details (a copy is included in the LICENSE file that
	* accompanied this code).
	*
	* You should have received a copy of the GNU General Public License version
	* 2 along with this work; if not, write to the Free Software Foundation,
	* Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
	*
	* Please contact Sun Microsystems, Inc., 4150 Network Circle, Santa Clara,
	* CA 95054 USA or visit www.sun.com if you need additional information or
	* have any questions.
	*/

	package sun.print;

	import java.io.BufferedReader;
	import java.io.InputStream;
	import java.io.InputStreamReader;
	import java.io.IOException;
	import java.util.ArrayList;
	import java.security.AccessController;
	import java.security.PrivilegedActionException;
	import java.security.PrivilegedExceptionAction;
	import javax.print.DocFlavor;
	import javax.print.MultiDocPrintService;
	import javax.print.PrintService;
	import javax.print.PrintServiceLookup;
	import javax.print.attribute.Attribute;
	import javax.print.attribute.AttributeSet;
	import javax.print.attribute.HashPrintRequestAttributeSet;
	import javax.print.attribute.HashPrintServiceAttributeSet;
	import javax.print.attribute.PrintRequestAttribute;
	import javax.print.attribute.PrintRequestAttributeSet;
	import javax.print.attribute.PrintServiceAttribute;
	import javax.print.attribute.PrintServiceAttributeSet;
	import javax.print.attribute.standard.PrinterName;

	public class Win32PrintServiceLookup extends PrintServiceLookup {

	private String defaultPrinter;
	private PrintService defaultPrintService;
	private String[] printers; /* excludes the default printer */
	private PrintService[] printServices; /* includes the default printer */

	static {
	java.security.AccessController.doPrivileged(
	new sun.security.action.LoadLibraryAction("awt"));
	}

	/* The singleton win32 print lookup service.
	* Code that is aware of this field and wants to use it must first
	* see if its null, and if so instantiate it by calling a method such as
	* javax.print.PrintServiceLookup.defaultPrintService() so that the
	* same instance is stored there.
	*/
	private static Win32PrintServiceLookup win32PrintLUS;

	/* Think carefully before calling this. Preferably don't call it. */
	public static Win32PrintServiceLookup getWin32PrintLUS() {
	if (win32PrintLUS == null) {
	/* This call is internally synchronized.
	* When it returns an instance of this class will have
	* been instantiated - else there's a JDK internal error.
	*/
	PrintServiceLookup.lookupDefaultPrintService();
	}
	return win32PrintLUS;
	}

	public Win32PrintServiceLookup() {

	if (win32PrintLUS == null) {
	win32PrintLUS = this;

	String osName = AccessController.doPrivileged(
	new sun.security.action.GetPropertyAction("os.name"));
	// There's no capability for Win98 to refresh printers.
	// See "OpenPrinter" for more info.
	if (osName != null && osName.startsWith("Windows 98")) {
	return;
	}
	// start the printer listener thread
	PrinterChangeListener thr = new PrinterChangeListener();
	thr.setDaemon(true);
	thr.start();
	} /* else condition ought to never happen! */
	}

	/* Want the PrintService which is default print service to have
	* equality of reference with the equivalent in list of print services
	* This isn't required by the API and there's a risk doing this will
	* lead people to assume its guaranteed.
	*/
	public synchronized PrintService[] getPrintServices() {
	SecurityManager security = System.getSecurityManager();
	if (security != null) {
	security.checkPrintJobAccess();
	}
	if (printServices == null) {
	refreshServices();
	}
	return printServices;
	}

	private synchronized void refreshServices() {
	printers = getAllPrinterNames();
	if (printers == null) {
	// In Windows it is safe to assume no default if printers == null so we
	// don't get the default.
	printServices = new PrintService[0];
	return;
	}

	PrintService[] newServices = new PrintService[printers.length];
	PrintService defService = getDefaultPrintService();
	for (int p = 0; p < printers.length; p++) {
	if (defService != null &&
	printers[p].equals(defService.getName())) {
	newServices[p] = defService;
	} else {
	if (printServices == null) {
	newServices[p] = new Win32PrintService(printers[p]);
	} else {
	int j;
	for (j = 0; j < printServices.length; j++) {
	if ((printServices[j]!= null) &&
	(printers[p].equals(printServices[j].getName()))) {
	newServices[p] = printServices[j];
	printServices[j] = null;
	break;
	}
	}
	if (j == printServices.length) {
	newServices[p] = new Win32PrintService(printers[p]);
	}
	}
	}
	}

	// Look for deleted services and invalidate these
	if (printServices != null) {
	for (int j=0; j < printServices.length; j++) {
	if ((printServices[j] instanceof Win32PrintService) &&
	(!printServices[j].equals(defaultPrintService))) {
	((Win32PrintService)printServices[j]).invalidateService();
	}
	}
	}
	printServices = newServices;
	}


	public synchronized PrintService getPrintServiceByName(String name) {

	if (name == null \|\| name.equals("")) {
	return null;
	} else {
	/* getPrintServices() is now very fast. */
	PrintService[] printServices = getPrintServices();
	for (int i=0; i<printServices.length; i++) {
	if (printServices[i].getName().equals(name)) {
	return printServices[i];
	}
	}
	return null;
	}
	}

	boolean matchingService(PrintService service,
	PrintServiceAttributeSet serviceSet) {
	if (serviceSet != null) {
	Attribute [] attrs = serviceSet.toArray();
	Attribute serviceAttr;
	for (int i=0; i<attrs.length; i++) {
	serviceAttr
	= service.getAttribute((Class<PrintServiceAttribute>)attrs[i].getCategory());
	if (serviceAttr == null \|\| !serviceAttr.equals(attrs[i])) {
	return false;
	}
	}
	}
	return true;
	}

	public PrintService[] getPrintServices(DocFlavor flavor,
	AttributeSet attributes) {

	SecurityManager security = System.getSecurityManager();
	if (security != null) {
	security.checkPrintJobAccess();
	}
	PrintRequestAttributeSet requestSet = null;
	PrintServiceAttributeSet serviceSet = null;

	if (attributes != null && !attributes.isEmpty()) {

	requestSet = new HashPrintRequestAttributeSet();
	serviceSet = new HashPrintServiceAttributeSet();

	Attribute[] attrs = attributes.toArray();
	for (int i=0; i<attrs.length; i++) {
	if (attrs[i] instanceof PrintRequestAttribute) {
	requestSet.add(attrs[i]);
	} else if (attrs[i] instanceof PrintServiceAttribute) {
	serviceSet.add(attrs[i]);
	}
	}
	}

	/*
	* Special case: If client is asking for a particular printer
	* (by name) then we can save time by getting just that service
	* to check against the rest of the specified attributes.
	*/
	PrintService[] services = null;
	if (serviceSet != null && serviceSet.get(PrinterName.class) != null) {
	PrinterName name = (PrinterName)serviceSet.get(PrinterName.class);
	PrintService service = getPrintServiceByName(name.getValue());
	if (service == null \|\| !matchingService(service, serviceSet)) {
	services = new PrintService[0];
	} else {
	services = new PrintService[1];
	services[0] = service;
	}
	} else {
	services = getPrintServices();
	}

	if (services.length == 0) {
	return services;
	} else {
	ArrayList matchingServices = new ArrayList();
	for (int i=0; i<services.length; i++) {
	try {
	if (services[i].
	getUnsupportedAttributes(flavor, requestSet) == null) {
	matchingServices.add(services[i]);
	}
	} catch (IllegalArgumentException e) {
	}
	}
	services = new PrintService[matchingServices.size()];
	return (PrintService[])matchingServices.toArray(services);
	}
	}

	/*
	* return empty array as don't support multi docs
	*/
	public MultiDocPrintService[]
	getMultiDocPrintServices(DocFlavor[] flavors,
	AttributeSet attributes) {
	SecurityManager security = System.getSecurityManager();
	if (security != null) {
	security.checkPrintJobAccess();
	}
	return new MultiDocPrintService[0];
	}


	public synchronized PrintService getDefaultPrintService() {
	SecurityManager security = System.getSecurityManager();
	if (security != null) {
	security.checkPrintJobAccess();
	}


	// Windows does not have notification for a change in default
	// so we always get the latest.
	defaultPrinter = getDefaultPrinterName();
	if (defaultPrinter == null) {
	return null;
	}

	if ((defaultPrintService != null) &&
	defaultPrintService.getName().equals(defaultPrinter)) {

	return defaultPrintService;
	}

	// Not the same as default so proceed to get new PrintService.

	// clear defaultPrintService
	defaultPrintService = null;

	if (printServices != null) {
	for (int j=0; j<printServices.length; j++) {
	if (defaultPrinter.equals(printServices[j].getName())) {
	defaultPrintService = printServices[j];
	break;
	}
	}
	}

	if (defaultPrintService == null) {
	defaultPrintService = new Win32PrintService(defaultPrinter);
	}
	return defaultPrintService;
	}

	class PrinterChangeListener extends Thread {
	long chgObj;
	PrinterChangeListener() {
	chgObj = notifyFirstPrinterChange(null);
	}

	public void run() {
	if (chgObj != -1) {
	while (true) {
	// wait for configuration to change
	if (notifyPrinterChange(chgObj) != 0) {
	try {
	refreshServices();
	} catch (SecurityException se) {
	break;
	}
	} else {
	notifyClosePrinterChange(chgObj);
	break;
	}
	}
	}
	}
	}

	private native String getDefaultPrinterName();
	private native String[] getAllPrinterNames();
	private native long notifyFirstPrinterChange(String printer);
	private native void notifyClosePrinterChange(long chgObj);
	private native int notifyPrinterChange(long chgObj);
	}