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 }