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