move file handling routines into helper

This commit is contained in:
Duncan Laurie 2004-09-01 00:19:36 +00:00
parent 6da8379fe4
commit a745d83af2
4 changed files with 81 additions and 80 deletions

View File

@ -50,10 +50,12 @@ unsigned short buf2short(unsigned char * buf);
uint32_t buf2long(unsigned char * buf);
const char * buf2str(unsigned char * buf, int len);
void printbuf(const unsigned char * buf, int len, const char * desc);
void signal_handler(int sig, void * handler);
unsigned char ipmi_csum(unsigned char * d, int s);
int ipmi_open_file(const char * file, int flags);
#define ipmi_open_file_read(file) ipmi_open_file(file, 0)
#define ipmi_open_file_write(file) ipmi_open_file(file, 1)
#define SIG_IGNORE(s) ((void)signal((s), SIG_IGN))
#define SIG_DEFAULT(s) ((void)signal((s), SIG_DFL))

View File

@ -39,6 +39,12 @@
#include <inttypes.h>
#include <signal.h>
#include <string.h>
#include <sys/types.h>
#include <sys/stat.h>
#include <unistd.h>
#include <fcntl.h>
#include <errno.h>
#include <ipmitool/helper.h>
uint32_t buf2long(unsigned char * buf)
@ -145,3 +151,67 @@ unsigned char ipmi_csum(unsigned char * d, int s)
return -c;
}
/* safely open a file for reading or writing
* file: filename
* rw: read-write flag, 1=write
*/
int ipmi_open_file(const char * file, int rw)
{
struct stat st1, st2;
int fp;
/* verify existance */
if (lstat(file, &st1) < 0) {
if (rw) {
/* does not exist, ok to create */
fp = open(file, O_WRONLY|O_TRUNC|O_EXCL|O_CREAT, 0600);
if (fp < 0) {
printf("ERROR: Unable to open file %s for write: %s\n",
file, strerror(errno));
return -1;
}
return fp;
} else {
printf("ERROR: File %s does not exist\n", file);
return -1;
}
}
/* it exists - only regular files, not links */
if (!S_ISREG(st1.st_mode)) {
printf("ERROR: File %s has invalid mode: %d\n", file, st1.st_mode);
return -1;
}
/* allow only files with 1 link (itself) */
if (st1.st_nlink != 1) {
printf("ERROR: File %s has invalid link count: %d != 1\n",
file, (int)st1.st_nlink);
return -1;
}
fp = open(file, rw ? (O_WRONLY|O_TRUNC) : O_RDONLY, 0600);
if (fp < 0) {
printf("ERROR: Unable to open file %s: %s\n",
file, strerror(errno));
return -1;
}
/* stat again */
if (fstat(fp, &st2) < 0) {
close(fp);
return -1;
}
/* verify inode, owner, link count */
if (st2.st_ino != st1.st_ino ||
st2.st_uid != st1.st_uid ||
st2.st_nlink != 1) {
printf("ERROR: Unable to verify file %s\n", file);
close(fp);
return -1;
}
return fp;
}

View File

@ -38,11 +38,8 @@
#include <math.h>
#include <stdio.h>
#include <fcntl.h>
#include <unistd.h>
#include <errno.h>
#include <sys/types.h>
#include <sys/stat.h>
#include <time.h>
#include <ipmitool/ipmi.h>
@ -1388,51 +1385,13 @@ ipmi_sdr_print_info(struct ipmi_intf * intf)
static int ipmi_sdr_dump_bin(struct ipmi_intf * intf, const char * ofile)
{
struct stat st1, st2;
struct sdr_get_rs * header;
struct ipmi_sdr_iterator * itr;
int fd;
/* open file for writing */
if (lstat(ofile, &st1) < 0) {
/* does not exist, ok to create */
fd = open(ofile, O_WRONLY|O_TRUNC|O_EXCL|O_CREAT, 0600);
if (fd < 1) {
printf("ERROR: Unable to open file '%s' for write: %s\n",
ofile, strerror(errno));
return -1;
}
} else {
/* it exists - allow only regular file, not links */
if (!S_ISREG(st1.st_mode)) {
printf("ERROR: file '%s' has invalid mode: %d\n",
ofile, st1.st_mode);
return -1;
}
/* allow only files with 1 link (itself) */
if (st1.st_nlink != 1) {
printf("ERROR: file '%s' has invalid link count: %d != 1\n",
ofile, (int)st1.st_nlink);
return -1;
}
/* open it for write, overwrite existing file */
fd = open(ofile, O_WRONLY | O_TRUNC, 0600);
if (fd < 1) {
printf("ERROR: unable to overwrite file '%s': %s\n",
ofile, strerror(errno));
return -1;
}
/* stat again and verify inode/owner/link count */
fstat(fd, &st2);
if (st2.st_ino != st1.st_ino || st2.st_uid != st1.st_uid ||
st2.st_nlink != 1) {
printf("ERROR: unable to verify file '%s'\n", ofile);
close(fd);
return -1;
}
fd = ipmi_open_file_write(ofile);
if (fd < 0) {
return -1;
}
/* open connection to SDR */

View File

@ -35,14 +35,10 @@
*/
#include <stdio.h>
#include <fcntl.h>
#include <unistd.h>
#include <sys/ioctl.h>
#include <errno.h>
#include <stdlib.h>
#include <string.h>
#include <sys/types.h>
#include <sys/stat.h>
#include <ipmitool/helper.h>
#include <ipmitool/log.h>
@ -160,38 +156,14 @@ static void usage(void)
}
static char * ipmi_password_file_read(char * file)
static char * ipmi_password_file_read(char * filename)
{
struct stat st1, st2;
int fp;
char * pass = NULL;
/* verify existance */
if (lstat(file, &st1) < 0) {
return NULL;
}
/* only regular files: no links */
if (!S_ISREG(st1.st_mode) || st1.st_nlink != 1) {
return NULL;
}
/* open it read-only */
if ((fp = open(file, O_RDONLY, 0)) < 0) {
return NULL;
}
/* stat again */
if (fstat(fp, &st2) < 0) {
close(fp);
return NULL;
}
/* verify inode, owner, link count */
if (st2.st_ino != st1.st_ino ||
st2.st_uid != st1.st_uid ||
st2.st_nlink != 1) {
close(fp);
fp = ipmi_open_file_read((const char *)filename);
if (fp < 0) {
printf("Unable to open password file %s\n", filename);
return NULL;
}
@ -207,9 +179,7 @@ static char * ipmi_password_file_read(char * file)
return NULL;
}
/* close file */
close(fp);
return pass;
}