You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
109 lines
3.4 KiB
109 lines
3.4 KiB
/* Copyright 2014-2017 Alex 'AdUser' Z (ad_user@runbox.com) |
|
* |
|
* This file is part of libsimdb |
|
* |
|
* This program is free software; you can redistribute it and/or modify |
|
* it under the terms of the GNU General Public License as published by |
|
* the Free Software Foundation; either version 2 of the License, or |
|
* (at your option) any later version. |
|
*/ |
|
|
|
#include <errno.h> |
|
#include <fcntl.h> |
|
#include <stdio.h> |
|
#include <stdlib.h> |
|
#include <string.h> |
|
#include <sys/stat.h> |
|
#include <sys/types.h> |
|
#include <unistd.h> |
|
|
|
#define SIMDB_REC_LEN 48 |
|
#define BLK_SIZE 1000 |
|
|
|
void usage(const char *message) { |
|
if (message) |
|
printf("error: %s\n", message); |
|
printf("Usage: simdb_convert_1to2 <infile> <outfile>\n"); |
|
exit(EXIT_FAILURE); |
|
} |
|
|
|
int main(int argc, char **argv) { |
|
int in, out; |
|
long unsigned int simdb_rec_total, rec_first, rec_last, records; |
|
unsigned char in_buf[SIMDB_REC_LEN * BLK_SIZE]; |
|
unsigned char out_buf[SIMDB_REC_LEN * BLK_SIZE]; |
|
unsigned char header[SIMDB_REC_LEN]; |
|
unsigned char *src, *dst; |
|
struct stat st; |
|
ssize_t bytes; |
|
|
|
memset(header, 0x0, SIMDB_REC_LEN); |
|
snprintf((char *) header, SIMDB_REC_LEN, "IMDB v%02u, CAPS: %s;", 2, "M-R"); |
|
|
|
if (argc < 2) |
|
usage(NULL); |
|
|
|
errno = 0; |
|
if ((in = open(argv[1], O_RDONLY)) < 0) |
|
usage(strerror(errno)); |
|
|
|
if ((out = open(argv[2], O_WRONLY | O_TRUNC | O_CREAT, 0644)) < 0) |
|
usage(strerror(errno)); |
|
|
|
if (fstat(in, &st) != 0) |
|
usage(strerror(errno)); |
|
|
|
if ((st.st_size % SIMDB_REC_LEN) != 0) |
|
usage("database size expected to be multiples to 48"); |
|
|
|
if (lseek(in, 0, SEEK_SET) < 0) |
|
usage(strerror(errno)); |
|
|
|
if (read(in, in_buf, SIMDB_REC_LEN) != SIMDB_REC_LEN) |
|
usage("can't read header of database"); |
|
|
|
if (memcmp(in_buf, "DB of image fingerprints (ver 1)", 32) != 0) |
|
usage("wrong database header / version mismatch"); |
|
|
|
simdb_rec_total = (st.st_size / SIMDB_REC_LEN); |
|
printf("Processing %lu records\n", simdb_rec_total - 1); |
|
|
|
if (lseek(in, SIMDB_REC_LEN, SEEK_SET) < 0) |
|
usage(strerror(errno)); |
|
|
|
if (lseek(out, 0, SEEK_SET) < 0) |
|
usage(strerror(errno)); |
|
if (write(out, header, SIMDB_REC_LEN) != SIMDB_REC_LEN) |
|
usage(strerror(errno)); |
|
|
|
for (unsigned int block = 0; block <= (simdb_rec_total / BLK_SIZE); block++) { |
|
rec_first = rec_last = (block * BLK_SIZE) + 1; |
|
records = ((rec_first + BLK_SIZE) > simdb_rec_total) ? simdb_rec_total - rec_first : BLK_SIZE; |
|
rec_last = rec_first + records; |
|
printf("* block %u, %4lu records [%lu, %lu]\n", |
|
block + 1, records, rec_first, rec_last - 1); |
|
lseek(in, rec_first * SIMDB_REC_LEN, SEEK_SET); |
|
lseek(out, rec_first * SIMDB_REC_LEN, SEEK_SET); |
|
bytes = read(in, in_buf, SIMDB_REC_LEN * records); |
|
if (bytes != (ssize_t) records * SIMDB_REC_LEN) { |
|
printf("Read size mismatch, expected %lu, got: %zi\n", |
|
SIMDB_REC_LEN * records, bytes); |
|
exit(EXIT_FAILURE); |
|
} |
|
memset(out_buf, 0x0, records * SIMDB_REC_LEN); |
|
for (unsigned int i = 0; i < records; i++) { |
|
src = &in_buf[i * SIMDB_REC_LEN]; |
|
dst = &out_buf[i * SIMDB_REC_LEN]; |
|
memcpy(dst + 0, src + 0, sizeof(char) * 1); // usage flag |
|
memcpy(dst + 16, src + 2, sizeof(char) * 32); // image bitmap |
|
} |
|
bytes = write(out, out_buf, SIMDB_REC_LEN * records); |
|
if (bytes != (ssize_t) records * SIMDB_REC_LEN) { |
|
printf("Write size mismatch, expected %lu, got: %zi\n", |
|
SIMDB_REC_LEN * records, bytes); |
|
exit(EXIT_FAILURE); |
|
} |
|
} |
|
|
|
exit(EXIT_SUCCESS); |
|
}
|
|
|