xref: /titanic_50/usr/src/lib/libmail/common/maillock.c (revision 7257d1b4d25bfac0c802847390e98a464fd787ac)
17c478bd9Sstevel@tonic-gate /*
27c478bd9Sstevel@tonic-gate  * CDDL HEADER START
37c478bd9Sstevel@tonic-gate  *
47c478bd9Sstevel@tonic-gate  * The contents of this file are subject to the terms of the
5*7257d1b4Sraf  * Common Development and Distribution License (the "License").
6*7257d1b4Sraf  * You may not use this file except in compliance with the License.
77c478bd9Sstevel@tonic-gate  *
87c478bd9Sstevel@tonic-gate  * You can obtain a copy of the license at usr/src/OPENSOLARIS.LICENSE
97c478bd9Sstevel@tonic-gate  * or http://www.opensolaris.org/os/licensing.
107c478bd9Sstevel@tonic-gate  * See the License for the specific language governing permissions
117c478bd9Sstevel@tonic-gate  * and limitations under the License.
127c478bd9Sstevel@tonic-gate  *
137c478bd9Sstevel@tonic-gate  * When distributing Covered Code, include this CDDL HEADER in each
147c478bd9Sstevel@tonic-gate  * file and include the License file at usr/src/OPENSOLARIS.LICENSE.
157c478bd9Sstevel@tonic-gate  * If applicable, add the following below this CDDL HEADER, with the
167c478bd9Sstevel@tonic-gate  * fields enclosed by brackets "[]" replaced with your own identifying
177c478bd9Sstevel@tonic-gate  * information: Portions Copyright [yyyy] [name of copyright owner]
187c478bd9Sstevel@tonic-gate  *
197c478bd9Sstevel@tonic-gate  * CDDL HEADER END
207c478bd9Sstevel@tonic-gate  */
21e8031f0aSraf 
227c478bd9Sstevel@tonic-gate /*
23*7257d1b4Sraf  * Copyright 2008 Sun Microsystems, Inc.  All rights reserved.
24e8031f0aSraf  * Use is subject to license terms.
257c478bd9Sstevel@tonic-gate  */
267c478bd9Sstevel@tonic-gate 
27*7257d1b4Sraf /*	Copyright (c) 1984, 1986, 1987, 1988, 1989 AT&T	*/
28*7257d1b4Sraf /*	  All Rights Reserved  	*/
297c478bd9Sstevel@tonic-gate 
30*7257d1b4Sraf #pragma ident	"%Z%%M%	%I%	%E% SMI"
31*7257d1b4Sraf 
327c478bd9Sstevel@tonic-gate #include "maillock.h"
337c478bd9Sstevel@tonic-gate #include <sys/types.h>
347c478bd9Sstevel@tonic-gate #include <fcntl.h>
357c478bd9Sstevel@tonic-gate #include <stdio.h>
367c478bd9Sstevel@tonic-gate #include <string.h>
377c478bd9Sstevel@tonic-gate #include <unistd.h>
387c478bd9Sstevel@tonic-gate #include <stdlib.h>
397c478bd9Sstevel@tonic-gate #include <utime.h>
407c478bd9Sstevel@tonic-gate 
417c478bd9Sstevel@tonic-gate #include <sys/stat.h>
427c478bd9Sstevel@tonic-gate 
437c478bd9Sstevel@tonic-gate static	char	*lockext = ".lock";	/* Lock suffix for mailname */
447c478bd9Sstevel@tonic-gate static	char	curlock[PATHSIZE];	/* Last used name of lock */
457c478bd9Sstevel@tonic-gate static	int	locked;			/* To note that we locked it */
467c478bd9Sstevel@tonic-gate static	time_t	locktime;		/* time lock file was touched */
477c478bd9Sstevel@tonic-gate static time_t	lock1(char *, char *);
487c478bd9Sstevel@tonic-gate 
497c478bd9Sstevel@tonic-gate /*
507c478bd9Sstevel@tonic-gate  * Lock the specified mail file by setting the file mailfile.lock.
517c478bd9Sstevel@tonic-gate  * We must, of course, be careful to remove the lock file by a call
527c478bd9Sstevel@tonic-gate  * to unlock before we stop.  The algorithm used here is to see if
537c478bd9Sstevel@tonic-gate  * the lock exists, and if it does, to check its modify time.  If it
547c478bd9Sstevel@tonic-gate  * is older than 5 minutes, we assume error and set our own file.
557c478bd9Sstevel@tonic-gate  * Otherwise, we wait for 5 seconds and try again.
567c478bd9Sstevel@tonic-gate  */
577c478bd9Sstevel@tonic-gate 
587c478bd9Sstevel@tonic-gate /*ARGSUSED*/
597c478bd9Sstevel@tonic-gate int
maillock(char * user,int retrycnt)607c478bd9Sstevel@tonic-gate maillock(char *user, int retrycnt)
617c478bd9Sstevel@tonic-gate {
627c478bd9Sstevel@tonic-gate 	time_t t;
637c478bd9Sstevel@tonic-gate 	struct stat sbuf;
647c478bd9Sstevel@tonic-gate 	int statfailed;
657c478bd9Sstevel@tonic-gate 	char locktmp[PATHSIZE];	/* Usable lock temporary */
667c478bd9Sstevel@tonic-gate 	char file[PATHSIZE];
677c478bd9Sstevel@tonic-gate 
687c478bd9Sstevel@tonic-gate 	if (locked)
697c478bd9Sstevel@tonic-gate 		return (0);
707c478bd9Sstevel@tonic-gate 	(void) strcpy(file, MAILDIR);
717c478bd9Sstevel@tonic-gate 	(void) strcat(file, user);
727c478bd9Sstevel@tonic-gate 	(void) strcpy(curlock, file);
737c478bd9Sstevel@tonic-gate 	(void) strcat(curlock, lockext);
747c478bd9Sstevel@tonic-gate 	(void) strcpy(locktmp, file);
757c478bd9Sstevel@tonic-gate 	(void) strcat(locktmp, "XXXXXX");
767c478bd9Sstevel@tonic-gate 	(void) mktemp(locktmp);
777c478bd9Sstevel@tonic-gate 	(void) remove(locktmp);
787c478bd9Sstevel@tonic-gate 	statfailed = 0;
797c478bd9Sstevel@tonic-gate 	for (;;) {
807c478bd9Sstevel@tonic-gate 		t = lock1(locktmp, curlock);
817c478bd9Sstevel@tonic-gate 		if (t == (time_t)0) {
827c478bd9Sstevel@tonic-gate 			locked = 1;
837c478bd9Sstevel@tonic-gate 			locktime = time(0);
847c478bd9Sstevel@tonic-gate 			return (0);
857c478bd9Sstevel@tonic-gate 		}
867c478bd9Sstevel@tonic-gate 		if (stat(curlock, &sbuf) < 0) {
877c478bd9Sstevel@tonic-gate 			if (statfailed++ > 5)
887c478bd9Sstevel@tonic-gate 				return (-1);
897c478bd9Sstevel@tonic-gate 			(void) sleep(5);
907c478bd9Sstevel@tonic-gate 			continue;
917c478bd9Sstevel@tonic-gate 		}
927c478bd9Sstevel@tonic-gate 		statfailed = 0;
937c478bd9Sstevel@tonic-gate 
947c478bd9Sstevel@tonic-gate 		/*
957c478bd9Sstevel@tonic-gate 		 * Compare the time of the temp file with the time
967c478bd9Sstevel@tonic-gate 		 * of the lock file, rather than with the current
977c478bd9Sstevel@tonic-gate 		 * time of day, since the files may reside on
987c478bd9Sstevel@tonic-gate 		 * another machine whose time of day differs from
997c478bd9Sstevel@tonic-gate 		 * ours.  If the lock file is less than 5 minutes
1007c478bd9Sstevel@tonic-gate 		 * old, keep trying.
1017c478bd9Sstevel@tonic-gate 		 */
1027c478bd9Sstevel@tonic-gate 		if (t < sbuf.st_ctime + 300) {
1037c478bd9Sstevel@tonic-gate 			(void) sleep(5);
1047c478bd9Sstevel@tonic-gate 			continue;
1057c478bd9Sstevel@tonic-gate 		}
1067c478bd9Sstevel@tonic-gate 		(void) remove(curlock);
1077c478bd9Sstevel@tonic-gate 	}
1087c478bd9Sstevel@tonic-gate }
1097c478bd9Sstevel@tonic-gate 
1107c478bd9Sstevel@tonic-gate /*
1117c478bd9Sstevel@tonic-gate  * Remove the mail lock, and note that we no longer
1127c478bd9Sstevel@tonic-gate  * have it locked.
1137c478bd9Sstevel@tonic-gate  */
1147c478bd9Sstevel@tonic-gate void
mailunlock(void)1157c478bd9Sstevel@tonic-gate mailunlock(void)
1167c478bd9Sstevel@tonic-gate {
1177c478bd9Sstevel@tonic-gate 	(void) remove(curlock);
1187c478bd9Sstevel@tonic-gate 	locked = 0;
1197c478bd9Sstevel@tonic-gate }
1207c478bd9Sstevel@tonic-gate 
1217c478bd9Sstevel@tonic-gate /*
1227c478bd9Sstevel@tonic-gate  * Attempt to set the lock by creating the temporary file,
1237c478bd9Sstevel@tonic-gate  * then doing a link/unlink.  If it succeeds, return 0,
1247c478bd9Sstevel@tonic-gate  * else return a guess of the current time on the machine
1257c478bd9Sstevel@tonic-gate  * holding the file.
1267c478bd9Sstevel@tonic-gate  */
1277c478bd9Sstevel@tonic-gate static time_t
lock1(char tempfile[],char name[])1287c478bd9Sstevel@tonic-gate lock1(char tempfile[], char name[])
1297c478bd9Sstevel@tonic-gate {
1307c478bd9Sstevel@tonic-gate 	int fd;
1317c478bd9Sstevel@tonic-gate 	struct stat sbuf;
1327c478bd9Sstevel@tonic-gate 
1337c478bd9Sstevel@tonic-gate 	fd = open(tempfile, O_RDWR|O_CREAT|O_EXCL, 0600);
1347c478bd9Sstevel@tonic-gate 	if (fd < 0)
1357c478bd9Sstevel@tonic-gate 		return (time(0));
1367c478bd9Sstevel@tonic-gate 	(void) fstat(fd, &sbuf);
1377c478bd9Sstevel@tonic-gate 	/*
1387c478bd9Sstevel@tonic-gate 	 * Write the string "0" into the lock file to give us some
1397c478bd9Sstevel@tonic-gate 	 * interoperability with SVR4 mailers.  SVR4 mailers expect
1407c478bd9Sstevel@tonic-gate 	 * a process ID to be written into the lock file and then
1417c478bd9Sstevel@tonic-gate 	 * use kill() to see if the process is alive or not.  We write
1427c478bd9Sstevel@tonic-gate 	 * 0 into it so that SVR4 mailers will always think our lock file
1437c478bd9Sstevel@tonic-gate 	 * is valid.
1447c478bd9Sstevel@tonic-gate 	 */
1457c478bd9Sstevel@tonic-gate 	(void) write(fd, "0", 2);
1467c478bd9Sstevel@tonic-gate 	(void) close(fd);
1477c478bd9Sstevel@tonic-gate 	if (link(tempfile, name) < 0) {
1487c478bd9Sstevel@tonic-gate 		(void) remove(tempfile);
1497c478bd9Sstevel@tonic-gate 		return (sbuf.st_ctime);
1507c478bd9Sstevel@tonic-gate 	}
1517c478bd9Sstevel@tonic-gate 	(void) remove(tempfile);
1527c478bd9Sstevel@tonic-gate 	return ((time_t)0);
1537c478bd9Sstevel@tonic-gate }
1547c478bd9Sstevel@tonic-gate 
1557c478bd9Sstevel@tonic-gate /*
1567c478bd9Sstevel@tonic-gate  * Update the change time on the lock file so
1577c478bd9Sstevel@tonic-gate  * others will know we're still using it.
1587c478bd9Sstevel@tonic-gate  */
1597c478bd9Sstevel@tonic-gate void
touchlock(void)1607c478bd9Sstevel@tonic-gate touchlock(void)
1617c478bd9Sstevel@tonic-gate {
1627c478bd9Sstevel@tonic-gate 	struct stat sbuf;
1637c478bd9Sstevel@tonic-gate 	time_t t;
1647c478bd9Sstevel@tonic-gate 	struct utimbuf tp;
1657c478bd9Sstevel@tonic-gate 
1667c478bd9Sstevel@tonic-gate 	if (!locked)
1677c478bd9Sstevel@tonic-gate 		return;
1687c478bd9Sstevel@tonic-gate 
1697c478bd9Sstevel@tonic-gate 	/* if it hasn't been at least 3 minutes, don't bother */
1707c478bd9Sstevel@tonic-gate 	if (time(&t) < locktime + 180)
1717c478bd9Sstevel@tonic-gate 		return;
1727c478bd9Sstevel@tonic-gate 	locktime = t;
1737c478bd9Sstevel@tonic-gate 
1747c478bd9Sstevel@tonic-gate 	if (stat(curlock, &sbuf) < 0)
1757c478bd9Sstevel@tonic-gate 		return;
1767c478bd9Sstevel@tonic-gate 	/*
1777c478bd9Sstevel@tonic-gate 	 * Don't actually change the times, we just want the
1787c478bd9Sstevel@tonic-gate 	 * side effect that utime causes st_ctime to be set
1797c478bd9Sstevel@tonic-gate 	 * to the current time.
1807c478bd9Sstevel@tonic-gate 	 */
1817c478bd9Sstevel@tonic-gate 	tp.actime = sbuf.st_atime;
1827c478bd9Sstevel@tonic-gate 	tp.modtime = sbuf.st_mtime;
1837c478bd9Sstevel@tonic-gate 	(void) utime(curlock, &tp);
1847c478bd9Sstevel@tonic-gate }
185