diff options
Diffstat (limited to 'build_rom.c')
| -rw-r--r-- | build_rom.c | 128 |
1 files changed, 127 insertions, 1 deletions
diff --git a/build_rom.c b/build_rom.c index 4346154..70a1583 100644 --- a/build_rom.c +++ b/build_rom.c @@ -7,6 +7,9 @@ #include <string.h> #include <time.h> #include <unistd.h> +#ifndef OLD_DB_FORMAT +# include <sqlite3.h> +#endif #define __packed __attribute__((__packed__)) @@ -65,7 +68,12 @@ int rand_range(int max) { return (int)(rand()/(double)RAND_MAX*max); } unsigned date(void) { return time(NULL) / 86400; } static char *next_serial = "next_serial.txt"; + +#ifdef OLD_DB_FORMAT static char *db = "db.bin"; +#else +static char *db = "db.sqlite"; +#endif static void print_usage(char *name) { @@ -80,7 +88,8 @@ static void print_usage(char *name) "\t-m MAC\tUse this MAC address + the 2 next\n" "\t-f MAC\tUse random MAC addresses (start of range)\n" "\t-t MAC\tUse random MAC addresses (end of range)\n" - "\t-n NUM\tUse this number for the serial\n", name); + "\t-n NUM\tUse this number for the serial\n" + , name); } int main(int argc, char *argv[]) @@ -302,8 +311,61 @@ unsigned get_next_serial(void) return cur; } +#ifndef OLD_DB_FORMAT +# define TEST_IN_SQLITE3(cond, bind) \ + sqlite3 *db; \ + sqlite3_stmt *stmt; \ + int rc, res; \ +\ + rc = sqlite3_open(fname, &db); \ + if (SQLITE_OK != rc) { \ + fprintf(stderr, "could not open sqlite database \"%s\" " \ + "(error %d)\n", fname, rc); \ + sqlite3_close(db); \ + res = -2; \ + goto opened; \ + } \ +\ + rc = sqlite3_prepare_v2(db, "SELECT 1 FROM xioh_data " \ + "WHERE " cond, \ + -1, &stmt, NULL); \ + if (SQLITE_OK != rc) { \ + fprintf(stderr, "could not compile statement (error %d)\n", rc); \ + sqlite3_close(db); \ + res = -4; \ + goto prepared; \ + } \ +\ + bind; \ +\ + rc = sqlite3_step(stmt); \ +\ + switch (rc) { \ + case SQLITE_DONE: \ + res = false; \ + break; \ + case SQLITE_ROW: \ + res = true; \ + break; \ + default: \ + fprintf(stderr, "could not execute statement (error %d)\n", rc); \ + res = -5; \ + } \ +\ +prepared: \ + sqlite3_finalize(stmt); \ +opened: \ + sqlite3_close(db); \ +\ + if (res < 0) \ + exit(-res); \ + else \ + return res; +#endif + bool mac_used(const char *fname, mac *addr) { +#ifdef OLD_DB_FORMAT FILE *f = fopen(fname, "r"); struct xioh_data xioh_data; unsigned i; @@ -325,10 +387,19 @@ bool mac_used(const char *fname, mac *addr) fclose(f); return false; +#else + TEST_IN_SQLITE3("mac1 == ? OR mac2 == ? OR mac3 == ?", + int i; + for (i = 1; i <= 3; i++) + sqlite3_bind_blob(stmt, i, addr, sizeof(*addr), + SQLITE_STATIC)) + +#endif } bool serial_used(const char *fname, int serial) { +#ifdef OLD_DB_FORMAT FILE *f = fopen(fname, "r"); struct xioh_data xioh_data; @@ -347,10 +418,14 @@ bool serial_used(const char *fname, int serial) fclose(f); return false; +#else + TEST_IN_SQLITE3("serial_number == ?", sqlite3_bind_int(stmt, 1, serial)) +#endif } void record_use(char *fname, struct xioh_data *xioh_data) { +#ifdef OLD_DB_FORMAT FILE *f = fopen(fname, "a"); if (!f) { printf("Can't open file \"%s\": %s\n", fname, strerror(errno)); @@ -359,6 +434,57 @@ void record_use(char *fname, struct xioh_data *xioh_data) fwrite(xioh_data, sizeof(*xioh_data), 1, f); fclose(f); +#else + int rc, i; + sqlite3 *db; + sqlite3_stmt *stmt; + + rc = sqlite3_open(fname, &db); + if (SQLITE_OK != rc) { + fprintf(stderr, "could not open sqlite database \"%s\" (error %d)\n", fname, rc); + sqlite3_close(db); + exit(2); + } + + rc = sqlite3_prepare_v2(db, + "INSERT INTO xioh_data (xioh_data, version, date," + "hardware_version, serial_number, mac1, mac2, mac3 ) " + "VALUES (?, ?, ?, ?, ?, ?, ?, ?)", -1, &stmt, NULL); + + if (SQLITE_OK != rc) { + fprintf(stderr, "could not compile statement (error %d)\n", rc); + sqlite3_close(db); + exit(5); + } + + sqlite3_bind_blob(stmt, 1, xioh_data, sizeof(*xioh_data), + SQLITE_TRANSIENT); + sqlite3_bind_int(stmt, 2, xioh_data->version); + sqlite3_bind_int(stmt, 3, xioh_data->serial.date); + sqlite3_bind_int(stmt, 4, xioh_data->serial.version); + sqlite3_bind_int(stmt, 5, xioh_data->serial.number); + for (i = 0; i < MAC_NUM; i++) + sqlite3_bind_blob(stmt, 6, &xioh_data->addr[i], + sizeof(*xioh_data->addr), + SQLITE_TRANSIENT); + + rc = sqlite3_step(stmt); + if (SQLITE_DONE != rc) { + fprintf(stderr, "could not insert into database: (error %d)\n", rc); + dump_xioh_data(xioh_data); + sqlite3_finalize(stmt); + sqlite3_close(db); + exit(4); + } + + sqlite3_finalize(stmt); + rc = sqlite3_close(db); + + if (SQLITE_OK != rc) { + fprintf(stderr, "Error on closing database (error %d)\n", rc); + exit(2); + } +#endif } mac mac_between(mac from, mac to) |