Home | History | Annotate | Download | only in prstat
      1  0  stevel /*
      2  0  stevel  * CDDL HEADER START
      3  0  stevel  *
      4  0  stevel  * The contents of this file are subject to the terms of the
      5  0  stevel  * Common Development and Distribution License, Version 1.0 only
      6  0  stevel  * (the "License").  You may not use this file except in compliance
      7  0  stevel  * with the License.
      8  0  stevel  *
      9  0  stevel  * You can obtain a copy of the license at usr/src/OPENSOLARIS.LICENSE
     10  0  stevel  * or http://www.opensolaris.org/os/licensing.
     11  0  stevel  * See the License for the specific language governing permissions
     12  0  stevel  * and limitations under the License.
     13  0  stevel  *
     14  0  stevel  * When distributing Covered Code, include this CDDL HEADER in each
     15  0  stevel  * file and include the License file at usr/src/OPENSOLARIS.LICENSE.
     16  0  stevel  * If applicable, add the following below this CDDL HEADER, with the
     17  0  stevel  * fields enclosed by brackets "[]" replaced with your own identifying
     18  0  stevel  * information: Portions Copyright [yyyy] [name of copyright owner]
     19  0  stevel  *
     20  0  stevel  * CDDL HEADER END
     21  0  stevel  */
     22  0  stevel /*
     23  0  stevel  * Copyright (c) 1999 by Sun Microsystems, Inc.
     24  0  stevel  * All rights reserved.
     25  0  stevel  */
     26  0  stevel 
     27  0  stevel #pragma ident	"%Z%%M%	%I%	%E% SMI"
     28  0  stevel 
     29  0  stevel #include <sys/stat.h>
     30  0  stevel #include <fcntl.h>
     31  0  stevel #include <unistd.h>
     32  0  stevel #include <stdlib.h>
     33  0  stevel #include <errno.h>
     34  0  stevel #include <string.h>
     35  0  stevel #include <strings.h>
     36  0  stevel 
     37  0  stevel #include "prtable.h"
     38  0  stevel #include "prutil.h"
     39  0  stevel #include "prfile.h"
     40  0  stevel 
     41  0  stevel #define	FDS_TABLE_SIZE	1024
     42  0  stevel 
     43  0  stevel static fd_t *fd_tbl = NULL;
     44  0  stevel static int fd_max;
     45  0  stevel static int fd_cnt;
     46  0  stevel static int fd_cnt_cur;
     47  0  stevel static int fd_cnt_old;
     48  0  stevel static fds_t *fds_tbl[FDS_TABLE_SIZE];
     49  0  stevel 
     50  0  stevel void
     51  0  stevel fd_init(int n)
     52  0  stevel {
     53  0  stevel 	fd_max = n;
     54  0  stevel 	fd_cnt = fd_cnt_cur = fd_cnt_old = 0;
     55  0  stevel 	fd_tbl = Zalloc(sizeof (fd_t) * n);
     56  0  stevel 	(void) memset(fds_tbl, 0, sizeof (fds_t *) * FDS_TABLE_SIZE);
     57  0  stevel }
     58  0  stevel 
     59  0  stevel void
     60  0  stevel fd_exit()
     61  0  stevel {
     62  0  stevel 	if (fd_tbl)
     63  0  stevel 		free(fd_tbl);
     64  0  stevel }
     65  0  stevel 
     66  0  stevel void
     67  0  stevel fd_close(fd_t *fdp)
     68  0  stevel {
     69  0  stevel 	if (fdp) {
     70  0  stevel 		if (fdp->fd_fd >= 0 && fdp->fd_name[0] != '\0') {
     71  0  stevel 			(void) close(fdp->fd_fd);
     72  0  stevel 			fd_cnt--;
     73  0  stevel 		}
     74  0  stevel 
     75  0  stevel 		(void) memset(fdp, 0, sizeof (fd_t));
     76  0  stevel 		fdp->fd_fd = -1;
     77  0  stevel 	}
     78  0  stevel }
     79  0  stevel 
     80  0  stevel void
     81  0  stevel fd_closeall()
     82  0  stevel {
     83  0  stevel 	fd_t *fdp = fd_tbl;
     84  0  stevel 	int i;
     85  0  stevel 
     86  0  stevel 	for (i = 0; i < fd_max; i++) {
     87  0  stevel 		fd_close(fdp);
     88  0  stevel 		fdp++;
     89  0  stevel 	}
     90  0  stevel }
     91  0  stevel 
     92  0  stevel static void
     93  0  stevel fd_recycle()
     94  0  stevel {
     95  0  stevel 	fd_t *fdp = fd_tbl;
     96  0  stevel 	int counter;
     97  0  stevel 	int i;
     98  0  stevel 
     99  0  stevel 	counter = abs(fd_cnt_old - fd_cnt) + NUM_RESERVED_FD;
    100  0  stevel 
    101  0  stevel 	for (i = 0; i < fd_max; i++, fdp++) {
    102  0  stevel 
    103  0  stevel 		if (fdp->fd_fd == -1)
    104  0  stevel 			continue;	/* skip recycled ones */
    105  0  stevel 
    106  0  stevel 		if (fdp->fd_name[0] != '\0') {	/* file has name */
    107  0  stevel 			(void) close(fdp->fd_fd);
    108  0  stevel 			fd_cnt--;
    109  0  stevel 			counter--;
    110  0  stevel 			fdp->fd_fd = -1;
    111  0  stevel 		}
    112  0  stevel 
    113  0  stevel 		if (counter == 0)
    114  0  stevel 			break;
    115  0  stevel 	}
    116  0  stevel }
    117  0  stevel 
    118  0  stevel fd_t *
    119  0  stevel fd_open(char *name, int flags, fd_t *fdp)
    120  0  stevel {
    121  0  stevel 	fd_t *fdp_new;
    122  0  stevel 	int fd;
    123  0  stevel 
    124  0  stevel 	if (fd_cnt > fd_max - NUM_RESERVED_FD)
    125  0  stevel 		fd_recycle();
    126  0  stevel 
    127  0  stevel 	if (fdp != NULL) {
    128  0  stevel 		if ((strcmp(fdp->fd_name, name) == 0) && (fdp->fd_fd >= 0)) {
    129  0  stevel 			fd_cnt_cur++;
    130  0  stevel 			return (fdp);
    131  0  stevel 		}
    132  0  stevel 	}
    133  0  stevel 
    134  0  stevel again:	fd = open(name, flags);
    135  0  stevel 
    136  0  stevel 	if (fd == -1) {
    137  0  stevel 		if ((errno == EMFILE) || (errno == ENFILE)) {
    138  0  stevel 			fd_recycle();
    139  0  stevel 			goto again;
    140  0  stevel 		}
    141  0  stevel 		fdp_new = NULL;
    142  0  stevel 	} else {
    143  0  stevel 		fdp_new = &fd_tbl[fd];
    144  0  stevel 		fdp_new->fd_fd = fd;
    145  0  stevel 		fdp_new->fd_flags = flags;
    146  0  stevel 		(void) strcpy(fdp_new->fd_name, name);
    147  0  stevel 		fd_cnt++;
    148  0  stevel 		fd_cnt_cur++;
    149  0  stevel 	}
    150  0  stevel 	return (fdp_new);
    151  0  stevel }
    152  0  stevel 
    153  0  stevel int
    154  0  stevel fd_getfd(fd_t *fdp)
    155  0  stevel {
    156  0  stevel 	return (fdp->fd_fd);
    157  0  stevel }
    158  0  stevel 
    159  0  stevel void
    160  0  stevel fd_update()
    161  0  stevel {
    162  0  stevel 	fd_cnt_old = fd_cnt_cur;
    163  0  stevel 	fd_cnt_cur = 0;
    164  0  stevel }
    165  0  stevel 
    166  0  stevel fds_t *
    167  0  stevel fds_get(pid_t pid)
    168  0  stevel {
    169  0  stevel 	fds_t *fdsp;
    170  0  stevel 	int hash = pid % FDS_TABLE_SIZE;
    171  0  stevel 
    172  0  stevel 	for (fdsp = fds_tbl[hash]; fdsp; fdsp = fdsp->fds_next)
    173  0  stevel 		if (fdsp->fds_pid == pid)	/* searching for pid */
    174  0  stevel 			return (fdsp);
    175  0  stevel 
    176  0  stevel 	fdsp = Zalloc(sizeof (fds_t));	/* adding new if pid was not found */
    177  0  stevel 	fdsp->fds_pid = pid;
    178  0  stevel 	fdsp->fds_next = fds_tbl[hash];
    179  0  stevel 	fds_tbl[hash] = fdsp;
    180  0  stevel 	return (fdsp);
    181  0  stevel }
    182  0  stevel 
    183  0  stevel void
    184  0  stevel fds_rm(pid_t pid)
    185  0  stevel {
    186  0  stevel 	fds_t *fds;
    187  0  stevel 	fds_t *fds_prev = NULL;
    188  0  stevel 	int hash = pid % FDS_TABLE_SIZE;
    189  0  stevel 
    190  0  stevel 	for (fds = fds_tbl[hash]; fds && fds->fds_pid != pid;
    191  0  stevel 	    fds = fds->fds_next)	/* finding pid */
    192  0  stevel 		fds_prev = fds;
    193  0  stevel 
    194  0  stevel 	if (fds) {			/* if pid was found */
    195  0  stevel 
    196  0  stevel 		fd_close(fds->fds_psinfo);
    197  0  stevel 		fd_close(fds->fds_usage);
    198  0  stevel 		fd_close(fds->fds_lpsinfo);
    199  0  stevel 		fd_close(fds->fds_lusage);
    200  0  stevel 
    201  0  stevel 		if (fds_prev)
    202  0  stevel 			fds_prev->fds_next = fds->fds_next;
    203  0  stevel 		else
    204  0  stevel 			fds_tbl[hash] = fds->fds_next;
    205  0  stevel 
    206  0  stevel 		free(fds);
    207  0  stevel 	}
    208  0  stevel }
    209