1 /*
   2  * Copyright (c) 2015, Oracle and/or its affiliates. All rights reserved.
   3  * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
   4  *
   5  * This code is free software; you can redistribute it and/or modify it
   6  * under the terms of the GNU General Public License version 2 only, as
   7  * published by the Free Software Foundation.
   8  *
   9  * This code is distributed in the hope that it will be useful, but WITHOUT
  10  * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
  11  * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License
  12  * version 2 for more details (a copy is included in the LICENSE file that
  13  * accompanied this code).
  14  *
  15  * You should have received a copy of the GNU General Public License version
  16  * 2 along with this work; if not, write to the Free Software Foundation,
  17  * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
  18  *
  19  * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA
  20  * or visit www.oracle.com if you need additional information or have any
  21  * questions.
  22  */
  23 
  24 /*
  25   @test
  26   @bug 8071668
  27   @summary Check whether clipboard see changes from external process after taking ownership
  28   @author Anton Nashatyrev: area=datatransfer
  29   @library /lib/testlibrary
  30   @build jdk.testlibrary.Utils
  31   @run main ClipboardInterVMTest
  32 */
  33 
  34 import jdk.testlibrary.Utils;
  35 
  36 import java.awt.*;
  37 import java.awt.datatransfer.*;
  38 import java.io.BufferedReader;
  39 import java.io.File;
  40 import java.io.IOException;
  41 import java.io.Reader;
  42 import java.util.ArrayList;
  43 import java.util.Arrays;
  44 import java.util.List;
  45 import java.util.concurrent.CountDownLatch;
  46 import java.util.concurrent.TimeUnit;
  47 
  48 public class ClipboardInterVMTest {
  49 
  50     static CountDownLatch lostOwnershipMonitor = new CountDownLatch(1);
  51     static CountDownLatch flavorChangedMonitor = new CountDownLatch(1);
  52     static Process process;
  53 
  54     public static void main(String[] args) throws Throwable {
  55         Clipboard clip = Toolkit.getDefaultToolkit().getSystemClipboard();
  56 
  57         if (args.length > 0) {
  58             System.out.println("Changing clip...");
  59             clip.setContents(new StringSelection("pong"), null);
  60             System.out.println("done");
  61             // keeping this process running for a while since on Mac the clipboard
  62             // will be invalidated via NSApplicationDidBecomeActiveNotification
  63             // callback in the main process after this child process finishes
  64             Thread.sleep(60 * 1000);
  65             return;
  66         };
  67 
  68 
  69         clip.setContents(new CustomSelection(), new ClipboardOwner() {
  70             @Override
  71             public void lostOwnership(Clipboard clipboard, Transferable contents) {
  72                 System.out.println("ClipboardInterVMTest.lostOwnership");
  73                 lostOwnershipMonitor.countDown();
  74             }
  75         });
  76 
  77         clip.addFlavorListener(new FlavorListener() {
  78             @Override
  79             public void flavorsChanged(FlavorEvent e) {
  80                 System.out.println("ClipboardInterVMTest.flavorsChanged");
  81                 flavorChangedMonitor.countDown();
  82             }
  83         });
  84 
  85         System.out.println("Starting external clipborad modifier...");
  86         new Thread(() -> runTest(ClipboardInterVMTest.class.getCanonicalName(), "pong")).start();
  87 
  88         String content = "";
  89         long startTime = System.currentTimeMillis();
  90         while (System.currentTimeMillis() - startTime < 30 * 1000) {
  91             Transferable c = clip.getContents(null);
  92             if (c.isDataFlavorSupported(DataFlavor.plainTextFlavor)) {
  93                 Reader reader = DataFlavor.plainTextFlavor.getReaderForText(c);
  94                 content = new BufferedReader(reader).readLine();
  95                 System.out.println(content);
  96                 if (content.equals("pong")) {
  97                     break;
  98                 }
  99             }
 100             Thread.sleep(200);
 101         }
 102 
 103         if (!lostOwnershipMonitor.await(10, TimeUnit.SECONDS)) {
 104             throw new RuntimeException("No LostOwnership event received.");
 105         };
 106 
 107         if (!flavorChangedMonitor.await(10, TimeUnit.SECONDS)) {
 108             throw new RuntimeException("No LostOwnership event received.");
 109         };
 110 
 111         if (!content.equals("pong")) {
 112             throw new RuntimeException("Content was not passed.");
 113         }
 114 
 115         process.destroy();
 116 
 117         System.out.println("Passed.");
 118     }
 119 
 120     private static void runTest(String main, String... args)  {
 121 
 122         try {
 123             List<String> opts = new ArrayList<>();
 124             opts.add(getJavaExe());
 125             opts.addAll(Arrays.asList(Utils.getTestJavaOpts()));
 126             opts.add("-cp");
 127             opts.add(System.getProperty("test.class.path", System.getProperty("java.class.path")));
 128 
 129             opts.add(main);
 130             opts.addAll(Arrays.asList(args));
 131 
 132             ProcessBuilder pb = new ProcessBuilder(opts.toArray(new String[0]));
 133             process = pb.start();
 134         } catch (Throwable throwable) {
 135             throw new RuntimeException(throwable);
 136         }
 137     }
 138 
 139     private static String getJavaExe() throws IOException {
 140         File p  = new File(System.getProperty("java.home"), "bin");
 141         File j = new File(p, "java");
 142         if (!j.canRead()) {
 143             j = new File(p, "java.exe");
 144         }
 145         if (!j.canRead()) {
 146             throw new RuntimeException("Can't find java executable in " + p);
 147         }
 148         return j.getCanonicalPath();
 149     }
 150 
 151     static class CustomSelection implements Transferable {
 152         private static final DataFlavor[] flavors = { DataFlavor.allHtmlFlavor };
 153 
 154         public DataFlavor[] getTransferDataFlavors() {
 155             return flavors;
 156         }
 157 
 158         public boolean isDataFlavorSupported(DataFlavor flavor) {
 159             return flavors[0].equals(flavor);
 160         }
 161 
 162         public Object getTransferData(DataFlavor flavor)
 163                 throws UnsupportedFlavorException, java.io.IOException {
 164             if (isDataFlavorSupported(flavor)) {
 165                 return "ping";
 166             } else {
 167                 throw new UnsupportedFlavorException(flavor);
 168             }
 169         }
 170     }
 171 }