| /* |
| * Copyright (c) 2005, 2015, Oracle and/or its affiliates. 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. |
| * |
| * 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 Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA |
| * or visit www.oracle.com if you need additional information or have any |
| * questions. |
| */ |
| |
| /* |
| * @test |
| * @bug 6222826 |
| * @summary Test that each thread in the thread pool runs |
| * in the context of the monitor.start() caller. |
| * @author Luis-Miguel Alventosa |
| * |
| * @run clean ThreadPoolAccTest |
| * @run build ThreadPoolAccTest |
| * @run main ThreadPoolAccTest |
| */ |
| |
| import java.security.AccessController; |
| import java.security.PrivilegedAction; |
| import java.util.Date; |
| import java.util.Set; |
| import javax.management.MBeanServer; |
| import javax.management.MBeanServerFactory; |
| import javax.management.ObjectName; |
| import javax.management.monitor.CounterMonitor; |
| import javax.management.monitor.GaugeMonitor; |
| import javax.management.monitor.Monitor; |
| import javax.management.monitor.StringMonitor; |
| import javax.management.remote.JMXPrincipal; |
| import javax.security.auth.Subject; |
| |
| public class ThreadPoolAccTest { |
| |
| // MBean class |
| public static class ObservedObject implements ObservedObjectMBean { |
| public volatile String principal; |
| public Integer getInteger() { |
| setPrincipal(); |
| return 0; |
| } |
| public Double getDouble() { |
| setPrincipal(); |
| return 0.0; |
| } |
| public String getString() { |
| setPrincipal(); |
| return ""; |
| } |
| private void setPrincipal() { |
| Subject subject = Subject.getSubject(AccessController.getContext()); |
| Set<JMXPrincipal> principals = subject.getPrincipals(JMXPrincipal.class); |
| principal = principals.iterator().next().getName(); |
| } |
| } |
| |
| // MBean interface |
| public interface ObservedObjectMBean { |
| public Integer getInteger(); |
| public Double getDouble(); |
| public String getString(); |
| } |
| |
| public static void main (String args[]) throws Exception { |
| |
| ObjectName[] mbeanNames = new ObjectName[6]; |
| ObservedObject[] monitored = new ObservedObject[6]; |
| ObjectName[] monitorNames = new ObjectName[6]; |
| Monitor[] monitor = new Monitor[6]; |
| String[] principals = { "role1", "role2" }; |
| String[] attributes = { "Integer", "Double", "String" }; |
| |
| try { |
| echo(">>> CREATE MBeanServer"); |
| MBeanServer server = MBeanServerFactory.newMBeanServer(); |
| |
| for (int i = 0; i < 6; i++) { |
| mbeanNames[i] = |
| new ObjectName(":type=ObservedObject,instance=" + i); |
| monitored[i] = new ObservedObject(); |
| echo(">>> CREATE ObservedObject = " + mbeanNames[i].toString()); |
| server.registerMBean(monitored[i], mbeanNames[i]); |
| |
| switch (i) { |
| case 0: |
| case 3: |
| monitorNames[i] = |
| new ObjectName(":type=CounterMonitor,instance=" + i); |
| monitor[i] = new CounterMonitor(); |
| break; |
| case 1: |
| case 4: |
| monitorNames[i] = |
| new ObjectName(":type=GaugeMonitor,instance=" + i); |
| monitor[i] = new GaugeMonitor(); |
| break; |
| case 2: |
| case 5: |
| monitorNames[i] = |
| new ObjectName(":type=StringMonitor,instance=" + i); |
| monitor[i] = new StringMonitor(); |
| break; |
| } |
| |
| echo(">>> CREATE Monitor = " + monitorNames[i].toString()); |
| server.registerMBean(monitor[i], monitorNames[i]); |
| monitor[i].addObservedObject(mbeanNames[i]); |
| monitor[i].setObservedAttribute(attributes[i % 3]); |
| monitor[i].setGranularityPeriod(500); |
| final Monitor m = monitor[i]; |
| Subject subject = new Subject(); |
| echo(">>> RUN Principal = " + principals[i / 3]); |
| subject.getPrincipals().add(new JMXPrincipal(principals[i / 3])); |
| PrivilegedAction<Void> action = new PrivilegedAction<Void>() { |
| public Void run() { |
| m.start(); |
| return null; |
| } |
| }; |
| Subject.doAs(subject, action); |
| } |
| |
| while(!testPrincipals(monitored, monitorNames, monitor, principals)); |
| |
| } finally { |
| for (int i = 0; i < 6; i++) |
| if (monitor[i] != null) |
| monitor[i].stop(); |
| } |
| } |
| |
| private static boolean testPrincipals(ObservedObject[] monitored, ObjectName[] monitorNames, |
| Monitor[] monitor, String[] principals) throws Exception { |
| for (int i = 0; i < 6; i++) { |
| String principal = monitored[i].principal; |
| String expected = principals[i / 3]; |
| if (principal == null) { |
| echo("Task not submitted " + new Date() + ". RETRY"); |
| return false; |
| } |
| echo(">>> Monitor = " + monitorNames[i]); |
| echo(">>> ObservedObject = " + monitor[i].getObservedObject()); |
| echo(">>> ObservedAttribute = " + monitor[i].getObservedAttribute()); |
| echo(">>> Principal = " + principal); |
| |
| if (expected.equals(principal)) { |
| echo("\tOK: Got Expected principal"); |
| } else { |
| throw new Exception("Unexpected principal. Got: " + principal + " Expected: " + expected); |
| } |
| } |
| return true; |
| } |
| |
| private static void echo(String message) { |
| System.out.println(message); |
| } |
| } |