ericonr
/
util-linux
mirror of https://github.com/ericonr/util-linux.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

mcookie: change coding style 

Now in align with README.devel.

Signed-off-by: Sami Kerola <kerolasa@iki.fi>
This commit is contained in:
Sami Kerola 2011-06-11 19:14:59 +02:00
parent 177f48fa0c
commit e6ebab66d4
1 changed files with 139 additions and 129 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

268
misc-utils/mcookie.c
View File

@ -18,167 +18,177 @@
*
*/
#include "c.h"
#include "md5.h"
#include "nls.h"
#include <fcntl.h>
#include <getopt.h>
#include <stddef.h>
#include <stdio.h>
#include <stdlib.h>
#include <fcntl.h>
#include "md5.h"
#include <sys/time.h>
#include <unistd.h>
#include "nls.h"
#include <getopt.h>
#include "c.h"
#define BUFFERSIZE 4096
#define BUFFERSIZE 4096
struct rngs {
const char *path;
int minlength, maxlength;
const char *path;
int minlength, maxlength;
} rngs[] = {
{ "/dev/random", 16, 16 }, /* 16 bytes = 128 bits suffice */
{ "/proc/interrupts", 0, 0 },
{ "/proc/slabinfo", 0, 0 },
{ "/proc/stat", 0, 0 },
{ "/dev/urandom", 32, 64 },
{"/dev/random", 16, 16}, /* 16 bytes = 128 bits suffice */
{"/proc/interrupts", 0, 0},
{"/proc/slabinfo", 0, 0},
{"/proc/stat", 0, 0},
{"/dev/urandom", 32, 64},
};
#define RNGS (sizeof(rngs)/sizeof(struct rngs))
int Verbose = 0;
/* The basic function to hash a file */
static off_t
hash_file(struct MD5Context *ctx, int fd)
static off_t hash_file(struct MD5Context *ctx, int fd)
{
off_t count = 0;
ssize_t r;
unsigned char buf[BUFFERSIZE];
off_t count = 0;
ssize_t r;
unsigned char buf[BUFFERSIZE];
while ((r = read(fd, buf, sizeof(buf))) > 0) {
MD5Update(ctx, buf, r);
count += r;
}
/* Separate files with a null byte */
buf[0] = 0;
MD5Update(ctx, buf, 1);
return count;
while ((r = read(fd, buf, sizeof(buf))) > 0) {
MD5Update(ctx, buf, r);
count += r;
}
/* Separate files with a null byte */
buf[0] = '\0';
MD5Update(ctx, buf, 1);
return count;
}
static void __attribute__ ((__noreturn__)) usage(FILE * out)
{
fprintf(out, _("Usage: %s [options]\n"),
program_invocation_short_name);
fprintf(out, _("Usage: %s [options]\n"),
program_invocation_short_name);
fprintf(out, _("\nOptions:\n"
" -f, --file=FILE use file as a cookie seed\n"
" -v, --verbose explain what is being done\n"
" -V, --version output version information and exit\n"
" -h, --help display this help and exit\n"));
fprintf(out, _("\nOptions:\n"
" -f, --file=FILE use file as a cookie seed\n"
" -v, --verbose explain what is being done\n"
" -V, --version output version information and exit\n"
" -h, --help display this help and exit\n"));
exit(out == stderr ? EXIT_FAILURE : EXIT_SUCCESS);
exit(out == stderr ? EXIT_FAILURE : EXIT_SUCCESS);
}
int main( int argc, char **argv )
int main(int argc, char **argv)
{
size_t i;
struct MD5Context ctx;
unsigned char digest[16];
unsigned char buf[BUFFERSIZE];
int fd;
int c;
pid_t pid;
char *file = NULL;
int r;
struct timeval tv;
struct timezone tz;
size_t i;
struct MD5Context ctx;
unsigned char digest[16];
unsigned char buf[BUFFERSIZE];
int fd;
int c;
pid_t pid;
char *file = NULL;
int verbose = 0;
int r;
struct timeval tv;
struct timezone tz;
static const struct option longopts[] = {
{"file", required_argument, NULL, 'f'},
{"verbose", no_argument, NULL, 'v'},
{"version", no_argument, NULL, 'V'},
{"help", no_argument, NULL, 'h'},
{NULL, 0, NULL, 0}
};
static const struct option longopts[] = {
{"file", required_argument, NULL, 'f'},
{"verbose", no_argument, NULL, 'v'},
{"version", no_argument, NULL, 'V'},
{"help", no_argument, NULL, 'h'},
{NULL, 0, NULL, 0}
};
setlocale(LC_ALL, "");
bindtextdomain(PACKAGE, LOCALEDIR);
textdomain(PACKAGE);
setlocale(LC_ALL, "");
bindtextdomain(PACKAGE, LOCALEDIR);
textdomain(PACKAGE);
while ((c = getopt_long(argc, argv, "f:vVh", longopts, NULL)) != -1)
switch (c) {
case 'v':
Verbose = 1;
break;
case 'f':
file = optarg;
break;
case 'V':
printf(_("%s from %s\n"), program_invocation_short_name,
PACKAGE_STRING);
return EXIT_SUCCESS;
case 'h':
usage(stdout);
default:
usage(stderr);
}
while ((c =
getopt_long(argc, argv, "f:vVh", longopts, NULL)) != -1)
switch (c) {
case 'v':
verbose = 1;
break;
case 'f':
file = optarg;
break;
case 'V':
printf(_("%s from %s\n"),
program_invocation_short_name,
PACKAGE_STRING);
return EXIT_SUCCESS;
case 'h':
usage(stdout);
default:
usage(stderr);
}
MD5Init( &ctx );
gettimeofday( &tv, &tz );
MD5Update( &ctx, (unsigned char *)&tv, sizeof( tv ) );
MD5Init(&ctx);
gettimeofday(&tv, &tz);
MD5Update(&ctx, (unsigned char *) &tv, sizeof(tv));
pid = getppid();
MD5Update( &ctx, (unsigned char *)&pid, sizeof( pid ));
pid = getpid();
MD5Update( &ctx, (unsigned char *)&pid, sizeof( pid ));
pid = getppid();
MD5Update(&ctx, (unsigned char *) &pid, sizeof(pid));
pid = getpid();
MD5Update(&ctx, (unsigned char *) &pid, sizeof(pid));
if (file) {
int count = 0;
if (file[0] == '-' && !file[1])
fd = STDIN_FILENO;
else
fd = open( file, O_RDONLY );
if (file) {
int count = 0;
if (fd < 0) {
warn( _("Could not open %s"), file );
} else {
count = hash_file( &ctx, fd );
if (Verbose)
fprintf( stderr, _("Got %d bytes from %s\n"), count, file );
if (file[0] == '-' && !file[1])
fd = STDIN_FILENO;
else
fd = open(file, O_RDONLY);
if (fd != STDIN_FILENO)
if(close( fd ))
err(EXIT_FAILURE, _("closing %s failed"), file);
}
}
if (fd < 0) {
warn(_("Could not open %s"), file);
} else {
count = hash_file(&ctx, fd);
if (verbose)
fprintf(stderr,
_("Got %d bytes from %s\n"), count,
file);
for (i = 0; i < RNGS; i++) {
if ((fd = open( rngs[i].path, O_RDONLY|O_NONBLOCK )) >= 0) {
int count = sizeof(buf);
if (fd != STDIN_FILENO)
if (close(fd))
err(EXIT_FAILURE,
_("closing %s failed"), file);
}
}
if (rngs[i].maxlength && count > rngs[i].maxlength)
count = rngs[i].maxlength;
r = read( fd, buf, count );
if (r > 0)
MD5Update( &ctx, buf, r );
else
r = 0;
close( fd );
if (Verbose)
fprintf( stderr, _("Got %d bytes from %s\n"), r, rngs[i].path );
if (rngs[i].minlength && r >= rngs[i].minlength)
break;
} else if (Verbose)
warn( _("Could not open %s"), rngs[i].path );
}
for (i = 0; i < RNGS; i++) {
if ((fd = open(rngs[i].path, O_RDONLY | O_NONBLOCK)) >= 0) {
int count = sizeof(buf);
MD5Final( digest, &ctx );
for (i = 0; i < 16; i++) printf( "%02x", digest[i] );
putchar ( '\n' );
/*
* The following is important for cases like disk full, so shell scripts
* can bomb out properly rather than think they succeeded.
*/
if (fflush(stdout) < 0 || fclose(stdout) < 0)
return EXIT_FAILURE;
if (rngs[i].maxlength && count > rngs[i].maxlength)
count = rngs[i].maxlength;
r = read(fd, buf, count);
if (r > 0)
MD5Update(&ctx, buf, r);
else
r = 0;
close(fd);
if (verbose)
fprintf(stderr,
_("Got %d bytes from %s\n"), r,
rngs[i].path);
if (rngs[i].minlength && r >= rngs[i].minlength)
break;
} else if (verbose)
warn(_("Could not open %s"), rngs[i].path);
}
return EXIT_SUCCESS;
MD5Final(digest, &ctx);
for (i = 0; i < 16; i++)
printf("%02x", digest[i]);
putchar('\n');
/*
* The following is important for cases like disk full,
* so shell scripts can bomb out properly rather than
* think they succeeded.
*/
if (fflush(stdout) < 0 || fclose(stdout) < 0)
return EXIT_FAILURE;
return EXIT_SUCCESS;
}