#include #include #include #include #include "ctl.h" /* Controllers. */ #include "db.h" /* Database. Defines global variable db. */ #include "err.h" /* HTTP errors. */ #include "site.h" /* Site settings. Defines global struct site. */ #include "query.h" /* Query functions. Defines global struct query. */ #define MAXERR 300 int main(int argc, char *argv[]) { char err[MAXERR], *p, *qs, *v; int postid, userid; sqlite3_stmt *stmt; /* * The database is opened or a server error is generated. * In the rest of the program, the database is always * assumed to be opened. */ if(sqlite3_open("db", &db) != SQLITE_OK){ snprintf(err, MAXERR, "The database could not be opened: %s\n", sqlite3_errmsg(db)); srverr(err); sqlite3_close(db); return 1; } /* * The site name is retrieved from the database. This early on, * it is appropriate to die with a server error on failure. */ if(sqlite3_prepare(db, "SELECT value FROM settings WHERE key = 'name'", -1, &stmt, 0) != SQLITE_OK){ snprintf(err, MAXERR, "The site name could not be retrieved: %s\n", sqlite3_errmsg(db)); srverr(err); sqlite3_close(db); return 1; } if(sqlite3_step(stmt) == SQLITE_ROW) site.name = strdup(sqlite3_column_text(stmt, 0)); else{ snprintf(err, MAXERR, "The site name is not set.\n"); srverr(err); sqlite3_finalize(stmt); sqlite3_close(db); return 1; } sqlite3_finalize(stmt); /* * The global struct query is set, or the program dies. * From now on, query is assumed to be set. */ setquery(); /* Handle empty request. */ if(!*query.string){ showfront(); goto end; } /* Parse query string. */ postid = userid = -1; while(p = nextparam(GET, 128)){ v = split(p); if(strcmp(p, "post") == 0) postid = atoi(v); else if(strcmp(p, "user") == 0) userid = atoi(v); } /* Handle request. */ if(postid != -1) showpost(postid); else if(userid != -1) showuser(userid); else showfront(); end: sqlite3_close(db); return 0; }