1 /*

     2  * Copyright (c) 2003, 2013, Oracle and/or its affiliates. All rights reserved.

     3  *

     4  * Redistribution and use in source and binary forms, with or without

     5  * modification, are permitted provided that the following conditions

     6  * are met:

     7  *

     8  *   - Redistributions of source code must retain the above copyright

     9  *     notice, this list of conditions and the following disclaimer.

    10  *

    11  *   - Redistributions in binary form must reproduce the above copyright

    12  *     notice, this list of conditions and the following disclaimer in the

    13  *     documentation and/or other materials provided with the distribution.

    14  *

    15  *   - Neither the name of Oracle nor the names of its

    16  *     contributors may be used to endorse or promote products derived

    17  *     from this software without specific prior written permission.

    18  *

    19  * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS

    20  * IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO,

    21  * THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR

    22  * PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL THE COPYRIGHT OWNER OR

    23  * CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,

    24  * EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,

    25  * PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR

    26  * PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF

    27  * LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING

    28  * NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS

    29  * SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.

    30  */

    31 

    32 /*

    33  * This source code is provided to illustrate the usage of a given feature

    34  * or technique and has been deliberately simplified. Additional steps

    35  * required for a production-quality application, such as security checks,

    36  * input validation and proper error handling, might not be present in

    37  * this sample code.

    38  */

    39 

    40 

    41 #include <sys/types.h>

    42 #include <sys/stat.h>

    43 #include <fcntl.h>

    44 

    45 #if !defined(LINUX) && !defined(_ALLBSD_SOURCE) && !defined(AIX)

    46 #include <procfs.h>

    47 #endif

    48 

    49 #include <stdio.h>

    50 #include <stdlib.h>

    51 #include <string.h>

    52 #include <sys/socket.h>

    53 #include <sys/errno.h>

    54 #include <unistd.h>

    55 #include <errno.h>

    56 #include <dlfcn.h>

    57 #include <sys/time.h>

    58 

    59 #include <netdb.h>

    60 #include <netinet/in.h>

    61 #include <sys/param.h>

    62 #include <time.h>

    63 

    64 #include "jni.h"

    65 #include "jvm_md.h"

    66 #include "hprof.h"

    67 

    68 #ifdef AIX

    69 #include "porting_aix.h" /* For the 'dladdr' function. */

    70 #endif

    71 

    72 int

    73 md_getpid(void)

    74 {

    75     static int pid = -1;

    76 

    77     if ( pid >= 0 ) {

    78         return pid;

    79     }

    80     pid = getpid();

    81     return pid;

    82 }

    83 

    84 void

    85 md_sleep(unsigned seconds)

    86 {

    87     sleep(seconds);

    88 }

    89 

    90 void

    91 md_init(void)

    92 {

    93 #if defined(LINUX) || defined(_ALLBSD_SOURCE) || defined(AIX)

    94     /* No Hi-Res timer option? */

    95 #else

    96     if ( gdata->micro_state_accounting ) {

    97         char proc_ctl_fn[48];

    98         int  procfd;

    99 

   100         /* Turn on micro state accounting, once per process */

   101         (void)md_snprintf(proc_ctl_fn, sizeof(proc_ctl_fn),

   102                 "/proc/%d/ctl", md_getpid());

   103 

   104         procfd = open(proc_ctl_fn, O_WRONLY);

   105         if (procfd >= 0) {

   106             long ctl_op[2];

   107 

   108             ctl_op[0] = PCSET;

   109             ctl_op[1] = PR_MSACCT;

   110             (void)write(procfd, ctl_op, sizeof(ctl_op));

   111             (void)close(procfd);

   112         }

   113     }

   114 #endif

   115 }

   116 

   117 int

   118 md_connect(char *hostname, unsigned short port)

   119 {

   120     struct hostent *hentry;

   121     struct sockaddr_in s;

   122     int fd;

   123 

   124     /* create a socket */

   125     fd = socket(AF_INET, SOCK_STREAM, 0);

   126     if ( fd < 0 ) {

   127         return -1;

   128     }

   129 

   130     /* find remote host's addr from name */

   131     if ((hentry = gethostbyname(hostname)) == NULL) {

   132         (void)close(fd);

   133         return -1;

   134     }

   135     (void)memset((char *)&s, 0, sizeof(s));

   136     /* set remote host's addr; its already in network byte order */

   137     (void)memcpy(&s.sin_addr.s_addr, *(hentry->h_addr_list),

   138            (int)sizeof(s.sin_addr.s_addr));

   139     /* set remote host's port */

   140     s.sin_port = htons(port);

   141     s.sin_family = AF_INET;

   142 

   143     /* now try connecting */

   144     if (-1 == connect(fd, (struct sockaddr*)&s, sizeof(s))) {

   145         (void)close(fd);

   146         return 0;

   147     }

   148     return fd;

   149 }

   150 

   151 int

   152 md_recv(int f, char *buf, int len, int option)

   153 {

   154     return recv(f, buf, len, option);

   155 }

   156 

   157 int

   158 md_shutdown(int filedes, int option)

   159 {

   160     return shutdown(filedes, option);

   161 }

   162 

   163 int

   164 md_open(const char *filename)

   165 {

   166     return open(filename, O_RDONLY);

   167 }

   168 

   169 int

   170 md_open_binary(const char *filename)

   171 {

   172     return md_open(filename);

   173 }

   174 

   175 int

   176 md_creat(const char *filename)

   177 {

   178     return open(filename, O_WRONLY | O_CREAT | O_TRUNC,

   179             S_IRUSR | S_IWUSR | S_IRGRP | S_IROTH);

   180 }

   181 

   182 int

   183 md_creat_binary(const char *filename)

   184 {

   185     return md_creat(filename);

   186 }

   187 

   188 jlong

   189 md_seek(int filedes, jlong cur)

   190 {

   191     jlong new_pos;

   192 

   193     if ( cur == (jlong)-1 ) {

   194         new_pos = lseek(filedes, 0, SEEK_END);

   195     } else {

   196         new_pos = lseek(filedes, cur, SEEK_SET);

   197     }

   198     return new_pos;

   199 }

   200 

   201 void

   202 md_close(int filedes)

   203 {

   204     (void)close(filedes);

   205 }

   206 

   207 int

   208 md_send(int s, const char *msg, int len, int flags)

   209 {

   210     int res;

   211 

   212     do {

   213         res = send(s, msg, len, flags);

   214     } while ((res < 0) && (errno == EINTR));

   215 

   216     return res;

   217 }

   218 

   219 int

   220 md_write(int filedes, const void *buf, int nbyte)

   221 {

   222     int res;

   223 

   224     do {

   225         res = write(filedes, buf, nbyte);

   226     } while ((res < 0) && (errno == EINTR));

   227 

   228     return res;

   229 }

   230 

   231 int

   232 md_read(int filedes, void *buf, int nbyte)

   233 {

   234     int res;

   235 

   236     do {

   237         res = read(filedes, buf, nbyte);

   238     } while ((res < 0) && (errno == EINTR));

   239 

   240     return res;

   241 }

   242 

   243 /* Time of day in milli-seconds */

   244 static jlong

   245 md_timeofday(void)

   246 {

   247     struct timeval tv;

   248 

   249     if ( gettimeofday(&tv, (void *)0) != 0 ) {

   250         return (jlong)0; /* EOVERFLOW ? */

   251     }

   252     /*LINTED*/

   253     return ((jlong)tv.tv_sec * (jlong)1000) + (jlong)(tv.tv_usec / 1000);

   254 }

   255 

   256 /* Hi-res timer in micro-seconds */

   257 jlong

   258 md_get_microsecs(void)

   259 {

   260 #if defined(LINUX) || defined(_ALLBSD_SOURCE) || defined(AIX)

   261     return (jlong)(md_timeofday() * (jlong)1000); /* Milli to micro */

   262 #else

   263     return (jlong)(gethrtime()/(hrtime_t)1000); /* Nano seconds to micro seconds */

   264 #endif

   265 }

   266 

   267 /* Time of day in milli-seconds */

   268 jlong

   269 md_get_timemillis(void)

   270 {

   271     return md_timeofday();

   272 }

   273 

   274 /* Current CPU hi-res CPU time used */

   275 jlong

   276 md_get_thread_cpu_timemillis(void)

   277 {

   278 #if defined(LINUX) || defined(_ALLBSD_SOURCE) || defined(AIX)

   279     return md_timeofday();

   280 #else

   281     return (jlong)(gethrvtime()/1000); /* Nano seconds to milli seconds */

   282 #endif

   283 }

   284 

   285 void

   286 md_get_prelude_path(char *path, int path_len, char *filename)

   287 {

   288     void *addr;

   289     char libdir[FILENAME_MAX+1];

   290     Dl_info dlinfo;

   291 

   292     libdir[0] = 0;

   293 #if defined(LINUX) || defined(_ALLBSD_SOURCE) || defined(AIX)

   294     addr = (void*)&Agent_OnLoad;

   295 #else

   296     /* Just using &Agent_OnLoad will get the first external symbol with

   297      *   this name in the first .so, which may not be libhprof.so.

   298      *   On Solaris we can actually ask for the address of our Agent_OnLoad.

   299      */

   300     addr = dlsym(RTLD_SELF, "Agent_OnLoad");

   301     /* Just in case the above didn't work (missing linker patch?). */

   302     if ( addr == NULL ) {

   303         addr = (void*)&Agent_OnLoad;

   304     }

   305 #endif

   306 

   307     /* Use dladdr() to get the full path to libhprof.so, which we use to find

   308      *  the prelude file.

   309      */

   310     dlinfo.dli_fname = NULL;

   311     (void)dladdr(addr, &dlinfo);

   312     if ( dlinfo.dli_fname != NULL ) {

   313         char * lastSlash;

   314 

   315         /* Full path to library name, need to move up one directory to 'lib' */

   316         (void)strcpy(libdir, (char *)dlinfo.dli_fname);

   317         lastSlash = strrchr(libdir, '/');

   318         if ( lastSlash != NULL ) {

   319             *lastSlash = '\0';

   320         }

   321 #ifndef __APPLE__

   322         // not sure why other platforms have to go up two levels, but on macos we only need up one

   323         lastSlash = strrchr(libdir, '/');

   324         if ( lastSlash != NULL ) {

   325             *lastSlash = '\0';

   326         }

   327 #endif /* __APPLE__ */

   328     }

   329     (void)snprintf(path, path_len, "%s/%s", libdir, filename);

   330 }

   331 

   332 

   333 int

   334 md_vsnprintf(char *s, int n, const char *format, va_list ap)

   335 {

   336     return vsnprintf(s, n, format, ap);

   337 }

   338 

   339 int

   340 md_snprintf(char *s, int n, const char *format, ...)

   341 {

   342     int ret;

   343     va_list ap;

   344 

   345     va_start(ap, format);

   346     ret = md_vsnprintf(s, n, format, ap);

   347     va_end(ap);

   348     return ret;

   349 }

   350 

   351 void

   352 md_system_error(char *buf, int len)

   353 {

   354     char *p;

   355 

   356     buf[0] = 0;

   357     p = strerror(errno);

   358     if ( p != NULL ) {

   359         (void)strcpy(buf, p);

   360     }

   361 }

   362 

   363 unsigned

   364 md_htons(unsigned short s)

   365 {

   366     return htons(s);

   367 }

   368 

   369 unsigned

   370 md_htonl(unsigned l)

   371 {

   372     return htonl(l);

   373 }

   374 

   375 unsigned

   376 md_ntohs(unsigned short s)

   377 {

   378     return ntohs(s);

   379 }

   380 

   381 unsigned

   382 md_ntohl(unsigned l)

   383 {

   384     return ntohl(l);

   385 }

   386 

   387 static void dll_build_name(char* buffer, size_t buflen,

   388                            const char* paths, const char* fname) {

   389     char *path, *paths_copy, *next_token;

   390 

   391     paths_copy = strdup(paths);

   392     if (paths_copy == NULL) {

   393         return;

   394     }

   395 

   396     next_token = NULL;

   397     path = strtok_r(paths_copy, ":", &next_token);

   398 

   399     while (path != NULL) {

   400         snprintf(buffer, buflen, "%s/lib%s" JNI_LIB_SUFFIX, path, fname);

   401         if (access(buffer, F_OK) == 0) {

   402             break;

   403         }

   404         *buffer = '\0';

   405         path = strtok_r(NULL, ":", &next_token);

   406     }

   407 

   408     free(paths_copy);

   409 }

   410 

   411 /* Create the actual fill filename for a dynamic library.  */

   412 void

   413 md_build_library_name(char *holder, int holderlen, const char *pname, const char *fname)

   414 {

   415     int   pnamelen;

   416 

   417     /* Length of options directory location. */

   418     pnamelen = pname ? strlen(pname) : 0;

   419 

   420     *holder = '\0';

   421     /* Quietly truncate on buffer overflow.  Should be an error. */

   422     if (pnamelen + (int)strlen(fname) + 10 > holderlen) {

   423         return;

   424     }

   425 

   426     /* Construct path to library */

   427     if (pnamelen == 0) {

   428         (void)snprintf(holder, holderlen, "lib%s" JNI_LIB_SUFFIX, fname);

   429     } else {

   430       dll_build_name(holder, holderlen, pname, fname);

   431     }

   432 }

   433 

   434 /* Load this library (return NULL on error, and error message in err_buf) */

   435 void *

   436 md_load_library(const char *name, char *err_buf, int err_buflen)

   437 {

   438     void * result;

   439 

   440     result = dlopen(name, RTLD_LAZY);

   441     if (result == NULL) {

   442         (void)strncpy(err_buf, dlerror(), err_buflen-2);

   443         err_buf[err_buflen-1] = '\0';

   444     }

   445     return result;

   446 }

   447 

   448 /* Unload this library */

   449 void

   450 md_unload_library(void *handle)

   451 {

   452     (void)dlclose(handle);

   453 }

   454 

   455 /* Find an entry point inside this library (return NULL if not found) */

   456 void *

   457 md_find_library_entry(void *handle, const char *name)

   458 {

   459     void * sym;

   460 

   461     sym =  dlsym(handle, name);

   462     return sym;

   463 }

   464 

   465