lp_utils.c 8.99 KiB
/* -*- mode: c; c-basic-offset: 8; indent-tabs-mode: nil; -*-
* vim:expandtab:shiftwidth=8:tabstop=8:
*
* GPL HEADER START
*
* DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
*
* This program is free software; you can redistribute it and/or modify
* it under the terms of the GNU General Public License version 2 only,
* as published by the Free Software Foundation.
*
* This program is distributed in the hope that it will be useful, but
* WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
* General Public License version 2 for more details (a copy is included
* in the LICENSE file that accompanied this code).
*
* You should have received a copy of the GNU General Public License
* version 2 along with this program; If not, see
* http://www.sun.com/software/products/lustre/docs/GPLv2.pdf
*
* Please contact Sun Microsystems, Inc., 4150 Network Circle, Santa Clara,
* CA 95054 USA or visit www.sun.com if you need additional information or
* have any questions.
*
* GPL HEADER END
*/
/*
* Copyright 2008 Sun Microsystems, Inc. All rights reserved
* Use is subject to license terms.
*/
/*
* This file is part of Lustre, http://www.lustre.org/
* Lustre is a trademark of Sun Microsystems, Inc.
*
* lustre/tests/lp_utils.c
*
* Author: You Feng <youfeng@clusterfs.com>
*/
#include <mpi.h>
#include <stdio.h>
#include <string.h>
#include <time.h>
#include <sys/time.h>
#include <sys/types.h>
#include <sys/stat.h>
#include <unistd.h>
#include <sys/ioctl.h>
#include <fcntl.h>
#include <errno.h>
#include "lustre/lustre_user.h"
#include "lustre/tests/lp_utils.h"
#define MAX_PROCESSES 8
int verbose = 0;
int debug = 0;
char hostname[1024];
struct timeval t1, t2;
char *timestamp() {
static char datestring[80];
time_t timestamp;
fflush(stdout);
timestamp = time(NULL);
strftime(datestring, 80, "%T", localtime(×tamp));
return datestring;
}
inline void begin(char *str) {
if (verbose > 0 && rank == 0) {
gettimeofday(&t1, NULL);
printf("%s:\tBeginning %s\n", timestamp(), str);
fflush(stdout);
}
}
inline void end(char *str) {
float elapsed;
MPI_Barrier(MPI_COMM_WORLD);
if (verbose > 0 && rank == 0) {
gettimeofday(&t2, NULL);
elapsed = (t2.tv_sec + ((float)t2.tv_usec/1000000))
- (t1.tv_sec + ((float)t1.tv_usec/1000000));
if (elapsed >= 60) {
printf("%s:\tFinished %-15s(%.2f min)\n",
timestamp(), str, elapsed / 60);
} else {
printf("%s:\tFinished %-15s(%.3f sec)\n",
timestamp(), str, elapsed);
}
fflush(stdout);
}
}
void dump_diff(char *orig_buf, char *buf, int size, long _off)
{
int i, diff, off;
char *p, *end;
printf("commpared buf size %d, at offset %lu\n\n", size, _off);
if (orig_buf) {
printf("original buf:\n");
p = orig_buf;
end = orig_buf + size;
i = 1;
while (p < end) {
printf(" %8lx", *(long *)p);
p += sizeof(long);
if (i++%8 == 0)
printf("\n");
}
if (i%8) printf("\n\n");
else printf("\n");
}
if (buf) {
printf("different data: diff_data(orig_data)\n");
diff = 0;
off = 0;
i = 1;
p = buf;
end = buf + size;
while (p < end) {
if (memcmp(p, orig_buf + off, sizeof(long)) != 0) {
printf("\toff: %5d,\tdata: %8lx (%8lx)\n", off,
*(unsigned long *)p,
*(unsigned long *)(orig_buf + off));
diff++;
}
off += sizeof(long);
p += sizeof(long); }
printf("\n %d total differents found\n\n", diff);
}
}
void lp_gethostname(void)
{
if (gethostname(hostname, 1024) == -1) {
fprintf(stderr, "gethostname: (%d)%s", errno, strerror(errno));
MPI_Abort(MPI_COMM_WORLD, 2);
}
}
/* This function does not FAIL if the requested "name" does not exit.
* This is just to clean up any files or directories left over from
* previous runs
*/
void remove_file_or_dir(char *name)
{
struct stat statbuf;
char errmsg[MAX_FILENAME_LEN + 20];
if (stat(name, &statbuf) != -1) {
if (S_ISREG(statbuf.st_mode)) {
printf("stale file found\n");
if (unlink(name) == -1) {
sprintf(errmsg, "unlink of %s", name);
FAIL(errmsg);
}
}
if (S_ISDIR(statbuf.st_mode)) {
printf("stale directory found\n");
if (rmdir(name) == -1) {
sprintf(errmsg, "rmdir of %s", name);
FAIL(errmsg);
}
}
}
}
void create_file(char *name, long filesize, int fill)
{
static char filename[MAX_FILENAME_LEN];
char errmsg[MAX_FILENAME_LEN + 20];
char buf[1024 * 8];
char c = 'A' + size;
int fd, rc;
short zero = 0;
long left = filesize;
/* Process 0 creates the test file(s) */
if (rank == 0) {
sprintf(filename, "%s/%s", testdir, name);
remove_file_or_dir(filename);
if ((fd = creat(filename, FILEMODE)) == -1) {
sprintf(errmsg, "create of file %s", filename);
FAIL(errmsg);
}
if (filesize > 0) {
if (lseek(fd, filesize - 1, SEEK_SET) == -1) {
close(fd);
sprintf(errmsg, "lseek of file %s", filename);
FAIL(errmsg);
}
if (write(fd, &zero, 1) == -1) {
close(fd);
sprintf(errmsg, "write of file %s", filename);
FAIL(errmsg);
}
} if (filesize > 0 && fill) {
if (lseek(fd, 0, SEEK_SET) == -1) {
close(fd);
sprintf(errmsg, "lseek of file %s", filename);
FAIL(errmsg);
}
memset(buf, c, 1024);
while (left > 0) {
if ((rc = write(fd, buf,
left > (1024 * 8) ? (1024 * 8) : left))
== -1) {
close(fd);
sprintf(errmsg, "write of file %s", filename);
FAIL(errmsg);
}
left -= rc;
}
}
if (close(fd) == -1) {
sprintf(errmsg, "close of file %s", filename);
FAIL(errmsg);
}
}
}
void check_stat(char *filename, struct stat *state, struct stat *old_state)
{
char errmsg[MAX_FILENAME_LEN+20];
if (stat(filename, state) == -1) {
sprintf(errmsg, "stat of file %s", filename);
FAIL(errmsg);
}
if (memcmp(state, old_state, sizeof(struct stat)) != 0) {
errno = 0;
sprintf(errmsg, LP_STAT_FMT, LP_STAT_ARGS);
FAIL(errmsg);
}
}
void remove_file(char *name)
{
char filename[MAX_FILENAME_LEN];
char errmsg[MAX_FILENAME_LEN + 20];
/* Process 0 remove the file(s) */
if (rank == 0) {
sprintf(filename, "%s/%s", testdir, name);
if (unlink(filename) == -1) {
sprintf(errmsg, "unlink of file %s", filename);
FAIL(errmsg);
}
}
}
void fill_stride(char *buf, int buf_size, long long rank, long long _off)
{
char *p = buf;
long long off, data[2];
int cp, left = buf_size;
data[0] = rank;
off = _off;
while (left > 0) {
data[1] = off;
cp = left > sizeof(data) ? sizeof(data) : left;
memcpy(p, data, cp);
off += cp;
p += cp; left -= cp;
}
}