1 /* 2 * This file and its contents are supplied under the terms of the 3 * Common Development and Distribution License ("CDDL"), version 1.0. 4 * You may only use this file in accordance with the terms of version 5 * 1.0 of the CDDL. 6 * 7 * A full copy of the text of the CDDL should have accompanied this 8 * source. A copy of the CDDL is also available via the Internet at 9 * http://www.illumos.org/license/CDDL. 10 */ 11 12 /* 13 * Copyright 2012 DEY Storage Systems, Inc. All rights reserved. 14 */ 15 /* 16 * Copyright (c) 2013 Joyent, Inc. All Rights reserved. 17 */ 18 19 #include <limits.h> 20 #include <stdio.h> 21 #include <errno.h> 22 #include <unistd.h> 23 #include <dirent.h> 24 #include <ctype.h> 25 #include <string.h> 26 #include <sys/mkdev.h> 27 28 #include "libproc.h" 29 #include "Pcontrol.h" 30 31 /* 32 * Pfdinfo.c - obtain open file information. 33 */ 34 35 /* 36 * Allocate an fd_info structure and stick it on the list. 37 * (Unless one already exists.) The list is sorted in 38 * reverse order. We will traverse it in that order later. 39 * This makes the usual ordered insert *fast*. 40 */ 41 fd_info_t * 42 Pfd2info(struct ps_prochandle *P, int fd) 43 { 44 fd_info_t *fip = list_next(&P->fd_head); 45 fd_info_t *next; 46 int i; 47 48 if (fip == NULL) { 49 list_link(&P->fd_head, NULL); 50 fip = list_next(&P->fd_head); 51 } 52 53 for (i = 0; i < P->num_fd; i++, fip = list_next(fip)) { 54 if (fip->fd_info.pr_fd == fd) { 55 return (fip); 56 } 57 if (fip->fd_info.pr_fd < fd) { 58 break; 59 } 60 } 61 62 next = fip; 63 if ((fip = calloc(1, sizeof (*fip))) == NULL) 64 return (NULL); 65 66 fip->fd_info.pr_fd = fd; 67 list_link(fip, next ? next : (void *)&(P->fd_head)); 68 P->num_fd++; 69 return (fip); 70 } 71 72 /* 73 * Attempt to load the open file information from a live process. 74 */ 75 static void 76 load_fdinfo(struct ps_prochandle *P) 77 { 78 /* 79 * In the unlikely case there are *no* file descriptors open, 80 * we will keep rescanning the proc directory, which will be empty. 81 * This is an edge case it isn't worth adding additional state to 82 * to eliminate. 83 */ 84 if (P->num_fd > 0) { 85 return; 86 } 87 88 if (P->state != PS_DEAD && P->state != PS_IDLE) { 89 char dir_name[PATH_MAX]; 90 char path[PATH_MAX]; 91 struct dirent *ent; 92 DIR *dirp; 93 int fd; 94 95 /* 96 * Try to get the path information first. 97 */ 98 (void) snprintf(dir_name, sizeof (dir_name), 99 "%s/%d/path", procfs_path, (int)P->pid); 100 dirp = opendir(dir_name); 101 if (dirp == NULL) { 102 return; 103 } 104 ent = NULL; 105 while ((ent = readdir(dirp)) != NULL) { 106 fd_info_t *fip; 107 prfdinfo_t *info; 108 int len; 109 struct stat64 stat; 110 111 if (!isdigit(ent->d_name[0])) 112 continue; 113 114 fd = atoi(ent->d_name); 115 116 fip = Pfd2info(P, fd); 117 info = &fip->fd_info; 118 info->pr_fd = fd; 119 120 if (pr_fstat64(P, fd, &stat) == 0) { 121 info->pr_mode = stat.st_mode; 122 info->pr_uid = stat.st_uid; 123 info->pr_gid = stat.st_gid; 124 info->pr_major = major(stat.st_dev); 125 info->pr_minor = minor(stat.st_dev); 126 info->pr_rmajor = major(stat.st_rdev); 127 info->pr_rminor = minor(stat.st_rdev); 128 info->pr_size = stat.st_size; 129 info->pr_ino = stat.st_ino; 130 } 131 132 info->pr_fileflags = pr_fcntl(P, fd, F_GETXFL, 0); 133 info->pr_fdflags = pr_fcntl(P, fd, F_GETFD, 0); 134 info->pr_offset = pr_llseek(P, fd, 0, SEEK_CUR); 135 136 /* attempt to determine the path to it */ 137 (void) snprintf(path, sizeof (path), 138 "%s/%d/path/%d", procfs_path, (int)P->pid, fd); 139 len = readlink(path, info->pr_path, 140 sizeof (info->pr_path) - 1); 141 142 if (len < 0) { 143 info->pr_path[0] = 0; 144 } else { 145 info->pr_path[len] = 0; 146 } 147 } 148 (void) closedir(dirp); 149 150 } 151 } 152 153 int 154 Pfdinfo_iter(struct ps_prochandle *P, proc_fdinfo_f *func, void *cd) 155 { 156 fd_info_t *fip; 157 int rv; 158 159 /* Make sure we have live data, if appropriate */ 160 load_fdinfo(P); 161 162 /* NB: We walk the list backwards. */ 163 164 for (fip = list_prev(&P->fd_head); 165 fip != (void *)&P->fd_head && fip != NULL; 166 fip = list_prev(fip)) { 167 if ((rv = func(cd, &fip->fd_info)) != 0) 168 return (rv); 169 } 170 return (0); 171 } 172