1996-12-09 14:05:35 +00:00
|
|
|
/*-
|
1996-12-10 23:59:04 +00:00
|
|
|
* Copyright (C) 1996
|
|
|
|
* David L. Nugent. All rights reserved.
|
1996-12-09 14:05:35 +00:00
|
|
|
*
|
|
|
|
* Redistribution and use in source and binary forms, with or without
|
|
|
|
* modification, are permitted provided that the following conditions
|
|
|
|
* are met:
|
|
|
|
* 1. Redistributions of source code must retain the above copyright
|
1996-12-10 23:59:04 +00:00
|
|
|
* notice, this list of conditions and the following disclaimer.
|
1996-12-09 14:05:35 +00:00
|
|
|
* 2. Redistributions in binary form must reproduce the above copyright
|
|
|
|
* notice, this list of conditions and the following disclaimer in the
|
|
|
|
* documentation and/or other materials provided with the distribution.
|
|
|
|
*
|
1996-12-10 23:59:04 +00:00
|
|
|
* THIS SOFTWARE IS PROVIDED BY DAVID L. NUGENT AND CONTRIBUTORS ``AS IS'' AND
|
1996-12-09 14:05:35 +00:00
|
|
|
* ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
|
|
|
|
* IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
|
1996-12-10 23:59:04 +00:00
|
|
|
* ARE DISCLAIMED. IN NO EVENT SHALL DAVID L. NUGENT OR CONTRIBUTORS BE LIABLE
|
1996-12-09 14:05:35 +00:00
|
|
|
* FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
|
|
|
|
* DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
|
|
|
|
* OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
|
|
|
|
* HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
|
|
|
|
* LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
|
|
|
|
* OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
|
|
|
|
* SUCH DAMAGE.
|
|
|
|
*/
|
|
|
|
|
1997-10-10 06:23:42 +00:00
|
|
|
#ifndef lint
|
|
|
|
static const char rcsid[] =
|
1999-08-28 01:35:59 +00:00
|
|
|
"$FreeBSD$";
|
1997-10-10 06:23:42 +00:00
|
|
|
#endif /* not lint */
|
|
|
|
|
1996-12-09 14:05:35 +00:00
|
|
|
#include <stdio.h>
|
|
|
|
#include <fcntl.h>
|
|
|
|
#include <stdlib.h>
|
|
|
|
#include <string.h>
|
|
|
|
#include <sys/types.h>
|
|
|
|
#include <sys/stat.h>
|
|
|
|
#include <sys/param.h>
|
|
|
|
#include <errno.h>
|
|
|
|
#include <unistd.h>
|
|
|
|
|
|
|
|
#include "pwupd.h"
|
|
|
|
|
1996-12-21 15:35:45 +00:00
|
|
|
int
|
|
|
|
extendline(char **buf, int * buflen, int needed)
|
|
|
|
{
|
|
|
|
if (needed > *buflen) {
|
|
|
|
char *tmp = realloc(*buf, needed);
|
|
|
|
if (tmp == NULL)
|
|
|
|
return -1;
|
|
|
|
*buf = tmp;
|
|
|
|
*buflen = needed;
|
|
|
|
}
|
|
|
|
return *buflen;
|
|
|
|
}
|
|
|
|
|
|
|
|
int
|
|
|
|
extendarray(char ***buf, int * buflen, int needed)
|
|
|
|
{
|
|
|
|
if (needed > *buflen) {
|
|
|
|
char **tmp = realloc(*buf, needed * sizeof(char *));
|
|
|
|
if (tmp == NULL)
|
|
|
|
return -1;
|
|
|
|
*buf = tmp;
|
|
|
|
*buflen = needed;
|
|
|
|
}
|
|
|
|
return *buflen;
|
|
|
|
}
|
|
|
|
|
|
|
|
|
1996-12-09 14:05:35 +00:00
|
|
|
int
|
|
|
|
fileupdate(char const * filename, mode_t fmode, char const * newline, char const * prefix, int pfxlen, int updmode)
|
|
|
|
{
|
|
|
|
int rc = 0;
|
|
|
|
|
|
|
|
if (pfxlen <= 1)
|
|
|
|
errno = EINVAL;
|
|
|
|
else {
|
1998-07-16 17:18:25 +00:00
|
|
|
int infd = open(filename, O_RDWR | O_CREAT, fmode);
|
1996-12-09 14:05:35 +00:00
|
|
|
|
|
|
|
if (infd != -1) {
|
|
|
|
FILE *infp = fdopen(infd, "r+");
|
|
|
|
|
|
|
|
if (infp == NULL)
|
|
|
|
close(infd);
|
|
|
|
else {
|
|
|
|
int outfd;
|
|
|
|
char file[MAXPATHLEN];
|
|
|
|
|
|
|
|
strcpy(file, filename);
|
|
|
|
strcat(file, ".new");
|
|
|
|
outfd = open(file, O_RDWR | O_CREAT | O_TRUNC | O_EXLOCK, fmode);
|
|
|
|
if (outfd != -1) {
|
|
|
|
FILE *outfp = fdopen(outfd, "w+");
|
|
|
|
|
|
|
|
if (outfp == NULL)
|
|
|
|
close(outfd);
|
|
|
|
else {
|
|
|
|
int updated = UPD_CREATE;
|
1996-12-21 15:35:45 +00:00
|
|
|
int linesize = PWBUFSZ;
|
|
|
|
char *line = malloc(linesize);
|
1996-12-09 14:05:35 +00:00
|
|
|
|
1996-12-21 15:35:45 +00:00
|
|
|
nextline:
|
|
|
|
while (fgets(line, linesize, infp) != NULL) {
|
1996-12-09 14:05:35 +00:00
|
|
|
char *p = strchr(line, '\n');
|
|
|
|
|
1996-12-21 15:35:45 +00:00
|
|
|
while ((p = strchr(line, '\n')) == NULL) {
|
|
|
|
int l;
|
|
|
|
if (extendline(&line, &linesize, linesize + PWBUFSZ) == -1) {
|
|
|
|
int ch;
|
|
|
|
fputs(line, outfp);
|
|
|
|
while ((ch = fgetc(infp)) != EOF) {
|
|
|
|
fputc(ch, outfp);
|
|
|
|
if (ch == '\n')
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
goto nextline;
|
1996-12-09 14:05:35 +00:00
|
|
|
}
|
1996-12-21 15:35:45 +00:00
|
|
|
l = strlen(line);
|
|
|
|
if (fgets(line + l, linesize - l, infp) == NULL)
|
|
|
|
break;
|
1996-12-09 14:05:35 +00:00
|
|
|
}
|
|
|
|
if (*line != '#' && *line != '\n') {
|
|
|
|
if (!updated && strncmp(line, prefix, pfxlen) == 0) {
|
|
|
|
updated = updmode == UPD_REPLACE ? UPD_REPLACE : UPD_DELETE;
|
|
|
|
|
|
|
|
/*
|
|
|
|
* Only actually write changes if updating
|
|
|
|
*/
|
|
|
|
if (updmode == UPD_REPLACE)
|
|
|
|
strcpy(line, newline);
|
|
|
|
else if (updmode == UPD_DELETE)
|
|
|
|
continue;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
fputs(line, outfp);
|
|
|
|
}
|
|
|
|
|
|
|
|
/*
|
|
|
|
* Now, we need to decide what to do: If we are in
|
|
|
|
* update mode, and no record was updated, then error If
|
|
|
|
* we are in insert mode, and record already exists,
|
|
|
|
* then error
|
|
|
|
*/
|
|
|
|
if (updmode != updated)
|
|
|
|
errno = (updmode == UPD_CREATE) ? EEXIST : ENOENT;
|
|
|
|
else {
|
|
|
|
|
|
|
|
/*
|
|
|
|
* If adding a new record, append it to the end
|
|
|
|
*/
|
|
|
|
if (updmode == UPD_CREATE)
|
|
|
|
fputs(newline, outfp);
|
|
|
|
|
|
|
|
/*
|
|
|
|
* Flush the file and check for the result
|
|
|
|
*/
|
|
|
|
rc = fflush(outfp) != EOF;
|
|
|
|
if (rc) {
|
|
|
|
|
|
|
|
/*
|
|
|
|
* Copy data back into the
|
|
|
|
* original file and truncate
|
|
|
|
*/
|
|
|
|
rewind(infp);
|
|
|
|
rewind(outfp);
|
1996-12-21 15:35:45 +00:00
|
|
|
while (fgets(line, linesize, outfp) != NULL)
|
1996-12-09 14:05:35 +00:00
|
|
|
fputs(line, infp);
|
|
|
|
|
|
|
|
/*
|
1998-07-16 17:18:25 +00:00
|
|
|
* If there was a problem with copying
|
|
|
|
* we will just rename 'file.new'
|
|
|
|
* to 'file'.
|
1996-12-09 14:05:35 +00:00
|
|
|
* This is a gross hack, but we may have
|
|
|
|
* corrupted the original file
|
1998-07-16 17:18:25 +00:00
|
|
|
* Unfortunately, it will lose the inode
|
|
|
|
* and hence the lock.
|
1996-12-09 14:05:35 +00:00
|
|
|
*/
|
1999-02-23 07:15:11 +00:00
|
|
|
if (fflush(infp) == EOF || ferror(infp)) {
|
|
|
|
rc = errno; /* Preserve errno for return */
|
|
|
|
rename(file, filename);
|
|
|
|
}
|
1996-12-09 14:05:35 +00:00
|
|
|
else
|
|
|
|
ftruncate(infd, ftell(infp));
|
|
|
|
}
|
|
|
|
}
|
1996-12-21 15:35:45 +00:00
|
|
|
free(line);
|
1996-12-09 14:05:35 +00:00
|
|
|
fclose(outfp);
|
|
|
|
}
|
|
|
|
remove(file);
|
|
|
|
}
|
|
|
|
fclose(infp);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
return rc;
|
|
|
|
}
|