1 /*
   2  * Copyright (c) 2000, 2013, 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.  Oracle designates this
   8  * particular file as subject to the "Classpath" exception as provided
   9  * by Oracle in the LICENSE file that accompanied this code.
  10  *
  11  * This code is distributed in the hope that it will be useful, but WITHOUT
  12  * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
  13  * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License
  14  * version 2 for more details (a copy is included in the LICENSE file that
  15  * accompanied this code).
  16  *
  17  * You should have received a copy of the GNU General Public License version
  18  * 2 along with this work; if not, write to the Free Software Foundation,
  19  * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
  20  *
  21  * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA
  22  * or visit www.oracle.com if you need additional information or have any
  23  * questions.
  24  */
  25 
  26 #include <jni.h>
  27 #include "com_sun_security_auth_module_SolarisSystem.h"
  28 #include <stdio.h>
  29 #include <pwd.h>
  30 #include <unistd.h>
  31 #include <stdlib.h>
  32 #include <string.h>
  33 #include <pwd.h>
  34 
  35 static void throwIllegalArgumentException(JNIEnv *env, const char *msg) {
  36     jclass clazz = (*env)->FindClass(env, "java/lang/IllegalArgumentException");
  37     if (clazz != NULL)
  38         (*env)->ThrowNew(env, clazz, msg);
  39 }
  40 
  41 JNIEXPORT void JNICALL
  42 Java_com_sun_security_auth_module_SolarisSystem_getSolarisInfo
  43                                                 (JNIEnv *env, jobject obj) {
  44 
  45     int i;
  46     char pwd_buf[1024];
  47     struct passwd pwd;
  48     jsize numSuppGroups = getgroups(0, NULL);
  49     jfieldID fid;
  50     jstring jstr;
  51     jlongArray jgroups;
  52     jlong *jgroupsAsArray;
  53     gid_t *groups;
  54     jclass cls;
  55 
  56     groups = (gid_t *)calloc(numSuppGroups, sizeof(gid_t));
  57 
  58     if (groups == NULL) {
  59         jclass cls = (*env)->FindClass(env,"java/lang/OutOfMemoryError");
  60         if (cls != NULL)
  61             (*env)->ThrowNew(env, cls, NULL);
  62         return;
  63     }
  64 
  65     cls = (*env)->GetObjectClass(env, obj);
  66 
  67     memset(pwd_buf, 0, sizeof(pwd_buf));
  68     if (getpwuid_r(getuid(), &pwd, pwd_buf, sizeof(pwd_buf)) != NULL &&
  69         getgroups(numSuppGroups, groups) != -1) {
  70 
  71         /*
  72          * set username
  73          */
  74         fid = (*env)->GetFieldID(env, cls, "username", "Ljava/lang/String;");
  75         if (fid == 0) {
  76             (*env)->ExceptionClear(env);
  77             throwIllegalArgumentException(env, "invalid field: username");
  78             goto cleanupAndReturn;
  79         }
  80         jstr = (*env)->NewStringUTF(env, pwd.pw_name);
  81         if (jstr == NULL) {
  82             goto cleanupAndReturn;
  83         }
  84         (*env)->SetObjectField(env, obj, fid, jstr);
  85 
  86         /*
  87          * set uid
  88          */
  89         fid = (*env)->GetFieldID(env, cls, "uid", "J");
  90         if (fid == 0) {
  91             (*env)->ExceptionClear(env);
  92             throwIllegalArgumentException(env, "invalid field: uid");
  93             goto cleanupAndReturn;
  94         }
  95         (*env)->SetLongField(env, obj, fid, pwd.pw_uid);
  96 
  97         /*
  98          * set gid
  99          */
 100         fid = (*env)->GetFieldID(env, cls, "gid", "J");
 101         if (fid == 0) {
 102             (*env)->ExceptionClear(env);
 103             throwIllegalArgumentException(env, "invalid field: gid");
 104             goto cleanupAndReturn;
 105         }
 106         (*env)->SetLongField(env, obj, fid, pwd.pw_gid);
 107 
 108         /*
 109          * set supplementary groups
 110          */
 111         fid = (*env)->GetFieldID(env, cls, "groups", "[J");
 112         if (fid == 0) {
 113             (*env)->ExceptionClear(env);
 114             throwIllegalArgumentException(env, "invalid field: groups");
 115             goto cleanupAndReturn;
 116         }
 117 
 118         jgroups = (*env)->NewLongArray(env, numSuppGroups);
 119         if (jgroups == NULL) {
 120             goto cleanupAndReturn;
 121         }
 122         jgroupsAsArray = (*env)->GetLongArrayElements(env, jgroups, 0);
 123         if (jgroupsAsArray == NULL) {
 124             goto cleanupAndReturn;
 125         }
 126         for (i = 0; i < numSuppGroups; i++)
 127             jgroupsAsArray[i] = groups[i];
 128         (*env)->ReleaseLongArrayElements(env, jgroups, jgroupsAsArray, 0);
 129         (*env)->SetObjectField(env, obj, fid, jgroups);
 130     }
 131 cleanupAndReturn:
 132     free(groups);
 133 
 134     return;
 135 }