#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, *v; int attid, 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((char *)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. */ attid = postid = userid = 0; while(p = nextparam(GET, 128)){ v = split(p); if(!attid && strcmp(p, "att") == 0) attid = atoi(v); else if(!postid && strcmp(p, "post") == 0) postid = atoi(v); else if(!userid && strcmp(p, "user") == 0) userid = atoi(v); } /* Handle request. */ if(attid) showatt(attid); else if(postid) showpost(postid); else if(userid) showuser(userid); else showfront(); end: sqlite3_close(db); return 0; }