1 /*
   2  * Copyright (c) 2000, 2016, 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 "jni_util.h"
  28 #include "com_sun_security_auth_module_UnixSystem.h"
  29 #include <stdio.h>
  30 #include <sys/types.h>
  31 #include <unistd.h>
  32 #include <stdlib.h>
  33 #include <string.h>
  34 
  35 /* For POSIX-compliant getpwuid_r on Solaris */
  36 #if defined(__solaris__)
  37 #define _POSIX_PTHREAD_SEMANTICS
  38 #endif
  39 #include <pwd.h>
  40 
  41 /*
  42  * Declare library specific JNI_Onload entry if static build
  43  */
  44 DEF_STATIC_JNI_OnLoad
  45 
  46 JNIEXPORT void JNICALL
  47 Java_com_sun_security_auth_module_UnixSystem_getUnixInfo
  48                                                 (JNIEnv *env, jobject obj) {
  49 
  50     int i;
  51     char pwd_buf[1024];
  52     struct passwd *pwd;
  53     struct passwd resbuf;
  54     jfieldID userNameID;
  55     jfieldID userID;
  56     jfieldID groupID;
  57     jfieldID supplementaryGroupID;
  58 
  59     jstring jstr;
  60     jlongArray jgroups;
  61     jlong *jgroupsAsArray;
  62     jsize numSuppGroups;
  63     gid_t *groups;
  64     jclass cls;
  65 
  66     numSuppGroups = getgroups(0, NULL);
  67     groups = (gid_t *)calloc(numSuppGroups, sizeof(gid_t));
  68     if (groups == NULL) {
  69         jclass cls = (*env)->FindClass(env,"java/lang/OutOfMemoryError");
  70         if (cls != NULL)
  71             (*env)->ThrowNew(env, cls, NULL);
  72         return;
  73     }
  74 
  75     cls = (*env)->GetObjectClass(env, obj);
  76 
  77     memset(pwd_buf, 0, sizeof(pwd_buf));
  78 
  79     if (getpwuid_r(getuid(), &resbuf, pwd_buf, sizeof(pwd_buf), &pwd) == 0 &&
  80         pwd != NULL &&
  81         getgroups(numSuppGroups, groups) != -1) {
  82 
  83         userNameID = (*env)->GetFieldID(env, cls, "username", "Ljava/lang/String;");
  84         if (userNameID == 0)
  85             goto cleanUpAndReturn;
  86 
  87         userID = (*env)->GetFieldID(env, cls, "uid", "J");
  88         if (userID == 0)
  89             goto cleanUpAndReturn;
  90 
  91         groupID = (*env)->GetFieldID(env, cls, "gid", "J");
  92         if (groupID == 0)
  93             goto cleanUpAndReturn;
  94 
  95         supplementaryGroupID = (*env)->GetFieldID(env, cls, "groups", "[J");
  96         if (supplementaryGroupID == 0)
  97             goto cleanUpAndReturn;
  98 
  99         jstr = (*env)->NewStringUTF(env, pwd->pw_name);
 100         if (jstr == NULL)
 101             goto cleanUpAndReturn;
 102         (*env)->SetObjectField(env, obj, userNameID, jstr);
 103 
 104         (*env)->SetLongField(env, obj, userID, pwd->pw_uid);
 105 
 106         (*env)->SetLongField(env, obj, groupID, pwd->pw_gid);
 107 
 108         jgroups = (*env)->NewLongArray(env, numSuppGroups);
 109         if (jgroups == NULL)
 110             goto cleanUpAndReturn;
 111         jgroupsAsArray = (*env)->GetLongArrayElements(env, jgroups, 0);
 112         if (jgroupsAsArray == NULL)
 113             goto cleanUpAndReturn;
 114         for (i = 0; i < numSuppGroups; i++)
 115             jgroupsAsArray[i] = groups[i];
 116         (*env)->ReleaseLongArrayElements(env, jgroups, jgroupsAsArray, 0);
 117         (*env)->SetObjectField(env, obj, supplementaryGroupID, jgroups);
 118     }
 119 cleanUpAndReturn:
 120     free(groups);
 121     return;
 122 }