xref: /titanic_41/usr/src/cmd/sendmail/src/stats.c (revision f52e84dfce89ba2ea0ebb6a191b4466da3012efd)
1 /*
2  * Copyright (c) 1998-2002 Sendmail, Inc. and its suppliers.
3  *	All rights reserved.
4  * Copyright (c) 1983, 1995-1997 Eric P. Allman.  All rights reserved.
5  * Copyright (c) 1988, 1993
6  *	The Regents of the University of California.  All rights reserved.
7  *
8  * By using this file, you agree to the terms and conditions set
9  * forth in the LICENSE file which can be found at the top level of
10  * the sendmail distribution.
11  *
12  */
13 
14 #pragma ident	"%Z%%M%	%I%	%E% SMI"
15 
16 #include <sendmail.h>
17 
18 SM_RCSID("@(#)$Id: stats.c,v 8.57 2006/08/15 23:24:58 ca Exp $")
19 
20 #include <sendmail/mailstats.h>
21 
22 static struct statistics	Stat;
23 
24 static bool	GotStats = false;	/* set when we have stats to merge */
25 
26 /* See http://physics.nist.gov/cuu/Units/binary.html */
27 #define ONE_K		1000		/* one thousand (twenty-four?) */
28 #define KBYTES(x)	(((x) + (ONE_K - 1)) / ONE_K)
29 /*
30 **  MARKSTATS -- mark statistics
31 **
32 **	Parameters:
33 **		e -- the envelope.
34 **		to -- to address.
35 **		type -- type of stats this represents.
36 **
37 **	Returns:
38 **		none.
39 **
40 **	Side Effects:
41 **		changes static Stat structure
42 */
43 
44 void
45 markstats(e, to, type)
46 	register ENVELOPE *e;
47 	register ADDRESS *to;
48 	int type;
49 {
50 	switch (type)
51 	{
52 	  case STATS_QUARANTINE:
53 		if (e->e_from.q_mailer != NULL)
54 			Stat.stat_nq[e->e_from.q_mailer->m_mno]++;
55 		break;
56 
57 	  case STATS_REJECT:
58 		if (e->e_from.q_mailer != NULL)
59 		{
60 			if (bitset(EF_DISCARD, e->e_flags))
61 				Stat.stat_nd[e->e_from.q_mailer->m_mno]++;
62 			else
63 				Stat.stat_nr[e->e_from.q_mailer->m_mno]++;
64 		}
65 		Stat.stat_cr++;
66 		break;
67 
68 	  case STATS_CONNECT:
69 		if (to == NULL)
70 			Stat.stat_cf++;
71 		else
72 			Stat.stat_ct++;
73 		break;
74 
75 	  case STATS_NORMAL:
76 		if (to == NULL)
77 		{
78 			if (e->e_from.q_mailer != NULL)
79 			{
80 				Stat.stat_nf[e->e_from.q_mailer->m_mno]++;
81 				Stat.stat_bf[e->e_from.q_mailer->m_mno] +=
82 					KBYTES(e->e_msgsize);
83 			}
84 		}
85 		else
86 		{
87 			Stat.stat_nt[to->q_mailer->m_mno]++;
88 			Stat.stat_bt[to->q_mailer->m_mno] += KBYTES(e->e_msgsize);
89 		}
90 		break;
91 
92 	  default:
93 		/* Silently ignore bogus call */
94 		return;
95 	}
96 
97 
98 	GotStats = true;
99 }
100 /*
101 **  CLEARSTATS -- clear statistics structure
102 **
103 **	Parameters:
104 **		none.
105 **
106 **	Returns:
107 **		none.
108 **
109 **	Side Effects:
110 **		clears the Stat structure.
111 */
112 
113 void
114 clearstats()
115 {
116 	/* clear the structure to avoid future disappointment */
117 	memset(&Stat, '\0', sizeof(Stat));
118 	GotStats = false;
119 }
120 /*
121 **  POSTSTATS -- post statistics in the statistics file
122 **
123 **	Parameters:
124 **		sfile -- the name of the statistics file.
125 **
126 **	Returns:
127 **		none.
128 **
129 **	Side Effects:
130 **		merges the Stat structure with the sfile file.
131 */
132 
133 void
134 poststats(sfile)
135 	char *sfile;
136 {
137 	int fd;
138 	static bool entered = false;
139 	long sff = SFF_REGONLY|SFF_OPENASROOT;
140 	struct statistics stats;
141 	extern off_t lseek();
142 
143 	if (sfile == NULL || *sfile == '\0' || !GotStats || entered)
144 		return;
145 	entered = true;
146 
147 	(void) time(&Stat.stat_itime);
148 	Stat.stat_size = sizeof(Stat);
149 	Stat.stat_magic = STAT_MAGIC;
150 	Stat.stat_version = STAT_VERSION;
151 
152 	if (!bitnset(DBS_WRITESTATSTOSYMLINK, DontBlameSendmail))
153 		sff |= SFF_NOSLINK;
154 	if (!bitnset(DBS_WRITESTATSTOHARDLINK, DontBlameSendmail))
155 		sff |= SFF_NOHLINK;
156 
157 	fd = safeopen(sfile, O_RDWR, 0600, sff);
158 	if (fd < 0)
159 	{
160 		if (LogLevel > 12)
161 			sm_syslog(LOG_INFO, NOQID, "poststats: %s: %s",
162 				  sfile, sm_errstring(errno));
163 		errno = 0;
164 		entered = false;
165 		return;
166 	}
167 	if (read(fd, (char *) &stats, sizeof(stats)) == sizeof(stats) &&
168 	    stats.stat_size == sizeof(stats) &&
169 	    stats.stat_magic == Stat.stat_magic &&
170 	    stats.stat_version == Stat.stat_version)
171 	{
172 		/* merge current statistics into statfile */
173 		register int i;
174 
175 		for (i = 0; i < MAXMAILERS; i++)
176 		{
177 			stats.stat_nf[i] += Stat.stat_nf[i];
178 			stats.stat_bf[i] += Stat.stat_bf[i];
179 			stats.stat_nt[i] += Stat.stat_nt[i];
180 			stats.stat_bt[i] += Stat.stat_bt[i];
181 			stats.stat_nr[i] += Stat.stat_nr[i];
182 			stats.stat_nd[i] += Stat.stat_nd[i];
183 			stats.stat_nq[i] += Stat.stat_nq[i];
184 		}
185 		stats.stat_cr += Stat.stat_cr;
186 		stats.stat_ct += Stat.stat_ct;
187 		stats.stat_cf += Stat.stat_cf;
188 	}
189 	else
190 		memmove((char *) &stats, (char *) &Stat, sizeof(stats));
191 
192 	/* write out results */
193 	(void) lseek(fd, (off_t) 0, 0);
194 	(void) write(fd, (char *) &stats, sizeof(stats));
195 	(void) close(fd);
196 
197 	/* clear the structure to avoid future disappointment */
198 	clearstats();
199 	entered = false;
200 }
201