1 /*

     2  * Copyright (c) 2014, 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 package util;

    24 

    25 import java.io.FilePermission;

    26 import java.security.AllPermission;

    27 import java.security.CodeSource;

    28 import java.security.Permission;

    29 import java.security.PermissionCollection;

    30 import java.security.Permissions;

    31 import java.security.Policy;

    32 import java.security.ProtectionDomain;

    33 import java.security.SecurityPermission;

    34 import java.sql.SQLPermission;

    35 import java.util.Enumeration;

    36 import java.util.PropertyPermission;

    37 import java.util.StringJoiner;

    38 

    39 /*

    40  * Simple Policy class that supports the required Permissions to validate the

    41  * JDBC concrete classes

    42  */

    43 public class TestPolicy extends Policy {

    44 

    45     final PermissionCollection permissions = new Permissions();

    46 

    47     /**

    48      * Constructor which sets the minimum permissions allowing testNG to work

    49      * with a SecurityManager

    50      */

    51     public TestPolicy() {

    52         setMinimalPermissions();

    53     }

    54 

    55     /*

    56      * Constructor which determines which permissions are defined for this

    57      * Policy used by the JDBC tests Possible values are: all (ALLPermissions),

    58      * setLog (SQLPemission("setLog"), deregisterDriver

    59      * (SQLPermission("deregisterDriver") (SQLPermission("deregisterDriver"),

    60      * and setSyncFactory(SQLPermission(setSyncFactory),

    61      *

    62      * @param policy Permissions to set

    63      */

    64     public TestPolicy(String policy) {

    65 

    66         switch (policy) {

    67             case "all":

    68                 permissions.add(new AllPermission());

    69                 break;

    70             case "setLog":

    71                 setMinimalPermissions();

    72                 permissions.add(new SQLPermission("setLog"));

    73                 break;

    74             case "deregisterDriver":

    75                 setMinimalPermissions();

    76                 permissions.add(new SQLPermission("deregisterDriver"));

    77                 break;

    78             case "setSyncFactory":

    79                 setMinimalPermissions();

    80                 permissions.add(new SQLPermission("setSyncFactory"));

    81                 break;

    82             default:

    83                 setMinimalPermissions();

    84         }

    85     }

    86 

    87     /*

    88      * Defines the minimal permissions required by testNG when running these

    89      * tests

    90      */

    91     private void setMinimalPermissions() {

    92         permissions.add(new SecurityPermission("getPolicy"));

    93         permissions.add(new SecurityPermission("setPolicy"));

    94         permissions.add(new RuntimePermission("getClassLoader"));

    95         permissions.add(new RuntimePermission("setSecurityManager"));

    96         permissions.add(new RuntimePermission("createSecurityManager"));

    97         permissions.add(new PropertyPermission("testng.show.stack.frames",

    98                 "read"));

    99         permissions.add(new PropertyPermission("line.separator", "read"));

   100         permissions.add(new PropertyPermission("fileStringBuffer", "read"));

   101         permissions.add(new PropertyPermission("dataproviderthreadcount", "read"));

   102         permissions.add(new PropertyPermission("java.io.tmpdir", "read"));

   103         permissions.add(new FilePermission("<<ALL FILES>>",

   104                 "read, write, delete"));

   105     }

   106 

   107     /*

   108      * Overloaded methods from the Policy class

   109      */

   110     @Override

   111     public String toString() {

   112         StringJoiner sj = new StringJoiner("\n", "policy: ", "");

   113         Enumeration<Permission> perms = permissions.elements();

   114         while (perms.hasMoreElements()) {

   115             sj.add(perms.nextElement().toString());

   116         }

   117         return sj.toString();

   118 

   119     }

   120 

   121     @Override

   122     public PermissionCollection getPermissions(ProtectionDomain domain) {

   123         return permissions;

   124     }

   125 

   126     @Override

   127     public PermissionCollection getPermissions(CodeSource codesource) {

   128         return permissions;

   129     }

   130 

   131     @Override

   132     public boolean implies(ProtectionDomain domain, Permission perm) {

   133         return permissions.implies(perm);

   134     }

   135 }