summaryrefslogtreecommitdiff
path: root/src/truncfile.c
blob: cc1f3993d1435e065627927da8d3a4da24e2caa4 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
/*
 * Copyright (c) 2000-2001 Silicon Graphics, Inc.  All Rights Reserved.
 * 
 * This program is free software; you can redistribute it and/or modify it
 * under the terms of version 2 of the GNU General Public License as
 * published by the Free Software Foundation.
 * 
 * This program is distributed in the hope that it would be useful, but
 * WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.
 * 
 * Further, this software is distributed without any warranty that it is
 * free of the rightful claim of any third person regarding infringement
 * or the like.  Any license provided herein, whether implied or
 * otherwise, applies only to this software file.  Patent licenses, if
 * any, provided herein do not apply to combinations of this program with
 * other software, or any other product whatsoever.
 * 
 * You should have received a copy of the GNU General Public License along
 * with this program; if not, write the Free Software Foundation, Inc., 59
 * Temple Place - Suite 330, Boston MA 02111-1307, USA.
 * 
 * Contact information: Silicon Graphics, Inc., 1600 Amphitheatre Pkwy,
 * Mountain View, CA  94043, or:
 * 
 * http://www.sgi.com 
 * 
 * For further information regarding this notice, see: 
 * 
 * http://oss.sgi.com/projects/GenInfo/SGIGPLNoticeExplan/
 */
 
#include "global.h"

long filesize;
int blocksize;
int count;
int verbose;
off64_t fileoffset;

void usage(char *progname);
void writeblk(int fd);
void truncfile(int fd);

void
usage(char *progname)
{
	fprintf(stderr, "usage: %s [-l filesize] [-b blocksize] [-c count] [-s seed] [-o fileoffset (hex)] [-v] filename\n",
			progname);
	exit(1);
}

int
main(int argc, char *argv[])
{
	int seed, i, ch, fd;
	char *filename = NULL;

	filesize = 256*1024*1024;
	blocksize = 512;
	count = filesize/blocksize;
	verbose = 0;
	fileoffset = 0;
	seed = time(NULL);
	while ((ch = getopt(argc, argv, "b:l:s:c:o:v")) != EOF) {
		switch(ch) {
		case 'b':	blocksize  = atoi(optarg);	break;
		case 'l':	filesize   = atol(optarg);	break;
		case 's':	seed       = atoi(optarg);	break;
		case 'c':	count      = atoi(optarg);	break;
		case 'o':	fileoffset = strtoll(optarg, NULL, 16); break;
		case 'v':	verbose++;			break;
		default:	usage(argv[0]);			break;
		}
	}
	if (optind == argc-1)
		filename = argv[optind];
	else
		usage(argv[0]);
	printf("Seed = %d (use \"-s %d\" to re-execute this test)\n", seed, seed);
	srandom(seed);

	/*
	 * Open the file, write rand block in random places, truncate the file,
	 * repeat ad-nauseum, then close the file.
	 */
	if ((fd = open(filename, O_RDWR|O_CREAT|O_TRUNC, 0666)) < 0) {
		perror("open");
		return 1;
	}
	for (i = 0; i < count; i++) {
		writeblk(fd);
		truncfile(fd);
		if (verbose && ((i % 100) == 0)) {
			printf(".");
			fflush(stdout);
		}
	}
	if (close(fd) < 0) {
		perror("close");
		return 1;
	}
	return 0;
}

void
writeblk(int fd)
{
	off_t offset;
	static char *buffer = NULL;

	if ((buffer == NULL) && ((buffer = calloc(1, blocksize)) == NULL)) {
		perror("malloc");
		exit(1);
	}

	offset = random() % filesize;
	if (lseek64(fd, (off64_t)(fileoffset + offset), SEEK_SET) < 0) {
		perror("lseek");
		exit(1);
	}
	*(long *)buffer = *(long *)(buffer+256) = (long)offset;
	if (write(fd, buffer, blocksize) < blocksize) {
		perror("write");
		exit(1);
	}
	if (verbose > 1)
		printf("writing   data at offset=%llx\n",
		       (unsigned long long)(fileoffset + offset));
}

void
truncfile(int fd)
{
	off_t offset;

	offset = random() % filesize;
	if (ftruncate64(fd, (off64_t)(fileoffset + offset)) < 0) {
		perror("truncate");
		exit(1);
	}
	if (verbose > 1)
		printf("truncated file to offset %llx\n",
		       (unsigned long long)(fileoffset + offset));
}