|
0
|
1 /* test/hfile.c -- Test cases for low-level input/output streams.
|
|
|
2
|
|
|
3 Copyright (C) 2013-2014 Genome Research Ltd.
|
|
|
4
|
|
|
5 Author: John Marshall <jm18@sanger.ac.uk>
|
|
|
6
|
|
|
7 Permission is hereby granted, free of charge, to any person obtaining a copy
|
|
|
8 of this software and associated documentation files (the "Software"), to deal
|
|
|
9 in the Software without restriction, including without limitation the rights
|
|
|
10 to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
|
|
|
11 copies of the Software, and to permit persons to whom the Software is
|
|
|
12 furnished to do so, subject to the following conditions:
|
|
|
13
|
|
|
14 The above copyright notice and this permission notice shall be included in
|
|
|
15 all copies or substantial portions of the Software.
|
|
|
16
|
|
|
17 THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
|
|
|
18 IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
|
|
|
19 FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL
|
|
|
20 THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
|
|
|
21 LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
|
|
|
22 FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER
|
|
|
23 DEALINGS IN THE SOFTWARE. */
|
|
|
24
|
|
|
25 #include <stdarg.h>
|
|
|
26 #include <stdio.h>
|
|
|
27 #include <stdlib.h>
|
|
|
28 #include <errno.h>
|
|
|
29
|
|
|
30 #include <sys/stat.h>
|
|
|
31
|
|
|
32 #include "htslib/hfile.h"
|
|
|
33 #include "htslib/hts_defs.h"
|
|
|
34
|
|
|
35 void HTS_NORETURN fail(const char *format, ...)
|
|
|
36 {
|
|
|
37 int err = errno;
|
|
|
38 va_list args;
|
|
|
39 va_start(args, format);
|
|
|
40 vfprintf(stderr, format, args);
|
|
|
41 va_end(args);
|
|
|
42 if (err != 0) fprintf(stderr, ": %s", strerror(err));
|
|
|
43 fprintf(stderr, "\n");
|
|
|
44 exit(EXIT_FAILURE);
|
|
|
45 }
|
|
|
46
|
|
|
47 void check_offset(hFILE *f, off_t off, const char *message)
|
|
|
48 {
|
|
|
49 off_t ret = htell(f);
|
|
|
50 if (ret < 0) fail("htell(%s)", message);
|
|
|
51 if (ret == off) return;
|
|
|
52
|
|
|
53 fprintf(stderr, "%s offset incorrect: expected %ld but got %ld\n",
|
|
|
54 message, (long)off, (long)ret);
|
|
|
55 exit(EXIT_FAILURE);
|
|
|
56 }
|
|
|
57
|
|
|
58 char *slurp(const char *filename)
|
|
|
59 {
|
|
|
60 char *text;
|
|
|
61 struct stat sbuf;
|
|
|
62 size_t filesize;
|
|
|
63 FILE *f = fopen(filename, "r");
|
|
|
64 if (f == NULL) fail("fopen(\"%s\", \"r\")", filename);
|
|
|
65 if (fstat(fileno(f), &sbuf) != 0) fail("fstat(\"%s\")", filename);
|
|
|
66 filesize = sbuf.st_size;
|
|
|
67
|
|
|
68 text = (char *) malloc(filesize + 1);
|
|
|
69 if (text == NULL) fail("malloc(text)");
|
|
|
70
|
|
|
71 if (fread(text, 1, filesize, f) != filesize) fail("fread");
|
|
|
72 fclose(f);
|
|
|
73
|
|
|
74 text[filesize] = '\0';
|
|
|
75 return text;
|
|
|
76 }
|
|
|
77
|
|
|
78 hFILE *fin = NULL;
|
|
|
79 hFILE *fout = NULL;
|
|
|
80
|
|
|
81 void reopen(const char *infname, const char *outfname)
|
|
|
82 {
|
|
|
83 if (fin) { if (hclose(fin) != 0) fail("hclose(input)"); }
|
|
|
84 if (fout) { if (hclose(fout) != 0) fail("hclose(output)"); }
|
|
|
85
|
|
|
86 fin = hopen(infname, "r");
|
|
|
87 if (fin == NULL) fail("hopen(\"%s\")", infname);
|
|
|
88
|
|
|
89 fout = hopen(outfname, "w");
|
|
|
90 if (fout == NULL) fail("hopen(\"%s\")", outfname);
|
|
|
91 }
|
|
|
92
|
|
|
93 int main(void)
|
|
|
94 {
|
|
|
95 static const int size[] = { 1, 13, 403, 999, 30000 };
|
|
|
96
|
|
|
97 char buffer[40000];
|
|
|
98 char *original;
|
|
|
99 int c, i;
|
|
|
100 ssize_t n;
|
|
|
101 off_t off;
|
|
|
102
|
|
|
103 reopen("vcf.c", "test/hfile1.tmp");
|
|
|
104 while ((c = hgetc(fin)) != EOF) {
|
|
|
105 if (hputc(c, fout) == EOF) fail("hputc");
|
|
|
106 }
|
|
|
107 if (herrno(fin)) { errno = herrno(fin); fail("hgetc"); }
|
|
|
108
|
|
|
109 reopen("test/hfile1.tmp", "test/hfile2.tmp");
|
|
|
110 if (hpeek(fin, buffer, 50) < 0) fail("hpeek");
|
|
|
111 while ((n = hread(fin, buffer, 17)) > 0) {
|
|
|
112 if (hwrite(fout, buffer, n) != n) fail("hwrite");
|
|
|
113 }
|
|
|
114 if (n < 0) fail("hread");
|
|
|
115
|
|
|
116 reopen("test/hfile2.tmp", "test/hfile3.tmp");
|
|
|
117 while ((n = hread(fin, buffer, sizeof buffer)) > 0) {
|
|
|
118 if (hwrite(fout, buffer, n) != n) fail("hwrite");
|
|
|
119 if (hpeek(fin, buffer, 700) < 0) fail("hpeek");
|
|
|
120 }
|
|
|
121 if (n < 0) fail("hread");
|
|
|
122
|
|
|
123 reopen("test/hfile3.tmp", "test/hfile4.tmp");
|
|
|
124 i = 0;
|
|
|
125 off = 0;
|
|
|
126 while ((n = hread(fin, buffer, size[i++ % 5])) > 0) {
|
|
|
127 off += n;
|
|
|
128 buffer[n] = '\0';
|
|
|
129 check_offset(fin, off, "pre-peek");
|
|
|
130 if (hputs(buffer, fout) == EOF) fail("hputs");
|
|
|
131 if ((n = hpeek(fin, buffer, size[(i+3) % 5])) < 0) fail("hpeek");
|
|
|
132 check_offset(fin, off, "post-peek");
|
|
|
133 }
|
|
|
134 if (n < 0) fail("hread");
|
|
|
135
|
|
|
136 reopen("test/hfile4.tmp", "test/hfile5.tmp");
|
|
|
137 n = hread(fin, buffer, 200);
|
|
|
138 if (n < 0) fail("hread");
|
|
|
139 else if (n != 200) fail("hread only got %d", (int)n);
|
|
|
140 if (hwrite(fout, buffer, 1000) != 1000) fail("hwrite");
|
|
|
141 check_offset(fin, 200, "input/first200");
|
|
|
142 check_offset(fout, 1000, "output/first200");
|
|
|
143
|
|
|
144 if (hseek(fin, 800, SEEK_CUR) < 0) fail("hseek/cur");
|
|
|
145 check_offset(fin, 1000, "input/seek");
|
|
|
146 for (off = 1000; (n = hread(fin, buffer, sizeof buffer)) > 0; off += n)
|
|
|
147 if (hwrite(fout, buffer, n) != n) fail("hwrite");
|
|
|
148 if (n < 0) fail("hread");
|
|
|
149 check_offset(fin, off, "input/eof");
|
|
|
150 check_offset(fout, off, "output/eof");
|
|
|
151
|
|
|
152 if (hseek(fin, 200, SEEK_SET) < 0) fail("hseek/set");
|
|
|
153 if (hseek(fout, 200, SEEK_SET) < 0) fail("hseek(output)");
|
|
|
154 check_offset(fin, 200, "input/backto200");
|
|
|
155 check_offset(fout, 200, "output/backto200");
|
|
|
156 n = hread(fin, buffer, 800);
|
|
|
157 if (n < 0) fail("hread");
|
|
|
158 else if (n != 800) fail("hread only got %d", (int)n);
|
|
|
159 if (hwrite(fout, buffer, 800) != 800) fail("hwrite");
|
|
|
160 check_offset(fin, 1000, "input/wrote800");
|
|
|
161 check_offset(fout, 1000, "output/wrote800");
|
|
|
162
|
|
|
163 if (hflush(fout) == EOF) fail("hflush");
|
|
|
164
|
|
|
165 original = slurp("vcf.c");
|
|
|
166 for (i = 1; i <= 5; i++) {
|
|
|
167 char *text;
|
|
|
168 sprintf(buffer, "test/hfile%d.tmp", i);
|
|
|
169 text = slurp(buffer);
|
|
|
170 if (strcmp(original, text) != 0) {
|
|
|
171 fprintf(stderr, "%s differs from vcf.c\n", buffer);
|
|
|
172 return EXIT_FAILURE;
|
|
|
173 }
|
|
|
174 free(text);
|
|
|
175 }
|
|
|
176 free(original);
|
|
|
177
|
|
|
178 if (hclose(fin) != 0) fail("hclose(input)");
|
|
|
179 if (hclose(fout) != 0) fail("hclose(output)");
|
|
|
180
|
|
|
181 fout = hopen("test/hfile_chars.tmp", "w");
|
|
|
182 if (fout == NULL) fail("hopen(\"test/hfile_chars.tmp\")");
|
|
|
183 for (i = 0; i < 256; i++)
|
|
|
184 if (hputc(i, fout) != i) fail("chars: hputc (%d)", i);
|
|
|
185 if (hclose(fout) != 0) fail("hclose(test/hfile_chars.tmp)");
|
|
|
186
|
|
|
187 fin = hopen("test/hfile_chars.tmp", "r");
|
|
|
188 if (fin == NULL) fail("hopen(\"test/hfile_chars.tmp\") for reading");
|
|
|
189 for (i = 0; i < 256; i++)
|
|
|
190 if ((c = hgetc(fin)) != i)
|
|
|
191 fail("chars: hgetc (%d = 0x%x) returned %d = 0x%x", i, i, c, c);
|
|
|
192 if ((c = hgetc(fin)) != EOF) fail("chars: hgetc (EOF) returned %d", c);
|
|
|
193 if (hclose(fin) != 0) fail("hclose(test/hfile_chars.tmp) for reading");
|
|
|
194
|
|
|
195 fin = hopen("data:hello, world!\n", "r");
|
|
|
196 if (fin == NULL) fail("hopen(\"data:...\")");
|
|
|
197 n = hread(fin, buffer, 300);
|
|
|
198 if (n < 0) fail("hread");
|
|
|
199 buffer[n] = '\0';
|
|
|
200 if (strcmp(buffer, "hello, world!\n") != 0) fail("hread result");
|
|
|
201 if (hclose(fin) != 0) fail("hclose(\"data:...\")");
|
|
|
202
|
|
|
203 return EXIT_SUCCESS;
|
|
|
204 }
|
