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
|
/*
* readbyte.c -- read a byte from a given file, using open()/read() or fopen()/fread()
*
* Copyright (C) 2018 Martin Pitt <martin@piware.de>
*
* umockdev is free software; you can redistribute it and/or
* modify it under the terms of the GNU Lesser General Public
* License as published by the Free Software Foundation; either
* version 2.1 of the License, or (at your option) any later version.
*
* umockdev 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
* Lesser General Public License for more details.
*
* You should have received a copy of the GNU Lesser General Public License
* along with this program; If not, see <http://www.gnu.org/licenses/>.
*/
#include <stdio.h>
#include <unistd.h>
#include <fcntl.h>
#include <string.h>
#define writestr(s) assert(write(fd, s, strlen(s)) >= 0)
int
main(int argc, char **argv)
{
char buf;
if (argc < 2 || argc > 3) {
fprintf(stderr, "Usage: %s file [open|fopen]\n", argv[0]);
return 1;
}
if (argc == 2 || strcmp(argv[2], "open") == 0) {
int fd;
ssize_t len;
fd = open(argv[1], O_RDWR);
if (fd < 0) {
perror("open");
return 1;
}
len = read(fd, &buf, 1);
if (len < 0) {
perror("read");
return 1;
}
close(fd);
}
else if (strcmp(argv[2], "fopen") == 0) {
FILE *f;
size_t len;
f = fopen(argv[1], "rb");
if (f == NULL) {
perror("fopen");
return 1;
}
len = fread(&buf, 1, 1, f);
if (len != 1 && ferror(f) != 0) {
perror("fread");
return 1;
}
fclose(f);
}
else {
fprintf(stderr, "ERROR: Unknown mode %s\n", argv[2]);
return 1;
}
return 0;
}
|