From 7ee5acb351f43fa650de138f8d2a3bda73ab27a0 Mon Sep 17 00:00:00 2001 From: Michael Krelin Date: Sun, 03 Feb 2008 16:18:47 +0000 Subject: added test OP Signed-off-by: Michael Krelin --- diff --git a/test/.gitignore b/test/.gitignore index d07884c..3d88495 100644 --- a/test/.gitignore +++ b/test/.gitignore @@ -5,3 +5,5 @@ /idiscover /RP.cgi /RP-db.cc +/OP.cgi +/OP-db.cc diff --git a/test/Makefile.am b/test/Makefile.am index 61e3787..8fedf48 100644 --- a/test/Makefile.am +++ b/test/Makefile.am @@ -1,4 +1,4 @@ -noinst_PROGRAMS = test idiscover RP.cgi +noinst_PROGRAMS = test idiscover RP.cgi OP.cgi AM_CPPFLAGS=${CPPFLAGS_DEBUG} DEFAULT_INCLUDES = -I${top_builddir} @@ -8,8 +8,8 @@ test_SOURCES = test.cc test_LDADD = ${top_builddir}/lib/libopkele.la EXTRA_DIST= \ - sqlite.h \ - RP-db.sql + sqlite.h kingate_openid_message.h \ + RP-db.sql OP-db.sql idiscover_SOURCES = idiscover.cc idiscover_LDADD = ${top_builddir}/lib/libopkele.la @@ -31,8 +31,20 @@ RP-db.cc: RP-db.sql echo ';' \ ) >$@ +OP_cgi_SOURCES = OP.cc +nodist_OP_cgi_SOURCES = OP-db.cc +OP_cgi_LDADD = ${RP_cgi_LDADD} +OP_cgi_CFLAGS = ${RP_cgi_CFLAGS} + +OP-db.cc: OP-db.sql + ( \ + echo 'const char * __OP_db_bootstrap = ' && \ + sed -e 's/^/"/' -e 's/$$/"/' $< && \ + echo ';' \ + ) >$@ + clean-local: - rm -f RP-db.cc + rm -f RP-db.cc OP-db.cc endif #HAVE_UUID endif #HAVE_KINGATE diff --git a/test/OP-db.sql b/test/OP-db.sql new file mode 100644 index 0000000..424b5ff --- a/dev/null +++ b/test/OP-db.sql @@ -0,0 +1,25 @@ +CREATE TABLE assoc ( + a_op text, + a_handle text NOT NULL, + a_type text DEFAULT 'HMAC-SHA1', + a_ctime text NOT NULL, + a_etime text NOT NULL, + a_secret text NOT NULL, + a_stateless integer NOT NULL DEFAULT 0, + a_itime integer, + UNIQUE(a_op,a_handle) +); + +CREATE TABLE nonces ( + n_once text NOT NULL PRIMARY KEY, + n_itime integer +); + +CREATE TABLE setup ( + s_password text +); + +CREATE TABLE ht_sessions ( + hts_id text NOT NULL PRIMARY KEY, + authorized integer NOT NULL DEFAULT 0 +); diff --git a/test/OP.cc b/test/OP.cc new file mode 100644 index 0000000..1196c0c --- a/dev/null +++ b/test/OP.cc @@ -0,0 +1,409 @@ +#include +#include +#include +#include +#include +using namespace std; +#include +#include +#include +#include +#include +#include +#include +#include +#include +#include + +#include "sqlite.h" +#include "kingate_openid_message.h" + +static const string get_self_url(const kingate::cgi_gateway& gw) { + bool s = gw.has_meta("SSL_PROTOCOL_VERSION"); + string rv = s?"https://":"http://"; + rv += gw.http_request_header("Host"); + const string& port = gw.get_meta("SERVER_PORT"); + if( port!=(s?"443":"80") ) { + rv += ':'; rv += port; + } + rv += gw.get_meta("REQUEST_URI"); + string::size_type q = rv.find('?'); + if(q!=string::npos) + rv.erase(q); + return rv; +} + +class opdb_t : public sqlite3_t { + public: + opdb_t() + : sqlite3_t("/tmp/OP.db") { + assert(_D); + char **resp; int nr,nc; char *errm; + if(sqlite3_get_table( + _D, "SELECT a_op FROM assoc LIMIT 0", + &resp,&nr,&nc,&errm)!=SQLITE_OK) { + extern const char *__OP_db_bootstrap; + DOUT_("Bootstrapping DB"); + if(sqlite3_exec(_D,__OP_db_bootstrap,NULL,NULL,&errm)!=SQLITE_OK) + throw opkele::exception(OPKELE_CP_ string("Failed to boostrap SQLite database: ")+errm); + }else + sqlite3_free_table(resp); + } +}; + +class example_op_t : public opkele::verify_op { + public: + kingate::cgi_gateway& gw; + opdb_t db; + kingate::cookie htc; + + + example_op_t(kingate::cgi_gateway& gw) + : gw(gw) { + try { + htc = gw.cookies.get_cookie("htop_session"); + sqlite3_mem_t S = sqlite3_mprintf( + "SELECT 1 FROM ht_sessions WHERE hts_id=%Q", + htc.get_value().c_str()); + sqlite3_table_t T; int nr,nc; + db.get_table(S,T,&nr,&nc); + if(nr<1) + throw kingate::exception_notfound(CODEPOINT,"forcing cookie generation"); + }catch(kingate::exception_notfound& kenf) { + uuid_t uuid; uuid_generate(uuid); + htc = kingate::cookie("htop_session",opkele::util::encode_base64(uuid,sizeof(uuid))); + sqlite3_mem_t S = sqlite3_mprintf( + "INSERT INTO ht_sessions (hts_id) VALUES (%Q)", + htc.get_value().c_str()); + db.exec(S); + } + } + + void set_authorized(bool a) { + sqlite3_mem_t + S = sqlite3_mprintf( + "UPDATE ht_sessions" + " SET authorized=%d" + " WHERE hts_id=%Q", + (int)a,htc.get_value().c_str()); + db.exec(S); + } + bool get_authorized() { + sqlite3_mem_t + S = sqlite3_mprintf( + "SELECT authorized" + " FROM ht_sessions" + " WHERE hts_id=%Q", + htc.get_value().c_str()); + sqlite3_table_t T; int nr,nc; + db.get_table(S,T,&nr,&nc); + assert(nr==1); assert(nc=1); + return opkele::util::string_to_long(T.get(1,0,nc)); + } + + ostream& cookie_header(ostream& o) const { + o << "Set-Cookie: " << htc.set_cookie_header() << "\n"; + return o; + } + + opkele::assoc_t alloc_assoc(const string& type,size_t klength,bool sl) { + uuid_t uuid; uuid_generate(uuid); + string a_handle = opkele::util::encode_base64(uuid,sizeof(uuid)); + opkele::secret_t a_secret; + generate_n( + back_insert_iterator(a_secret),klength, + rand ); + string ssecret; a_secret.to_base64(ssecret); + time_t now = time(0); + int expires_in = sl?3600*2:3600*24*7*2; + sqlite3_mem_t + S = sqlite3_mprintf( + "INSERT INTO assoc" + " (a_handle,a_type,a_ctime,a_etime,a_secret,a_stateless)" + " VALUES (" + " %Q,%Q,datetime('now')," + " datetime('now','+%d seconds')," + " %Q,%d );", + a_handle.c_str(), type.c_str(), + expires_in, + ssecret.c_str(), sl ); + db.exec(S); + return opkele::assoc_t(new opkele::association( + "", + a_handle, type, a_secret, + now+expires_in, sl )); + } + + opkele::assoc_t retrieve_assoc(const string& h) { + sqlite3_mem_t + S = sqlite3_mprintf( + "SELECT" + " a_handle,a_type,a_secret,a_stateless," + " strftime('%%s',a_etime) AS a_etime," + " a_itime" + " FROM assoc" + " WHERE a_handle=%Q AND a_itime IS NULL" + " AND datetime('now') < a_etime" + " LIMIT 1", + h.c_str() ); + sqlite3_table_t T; + int nr,nc; + db.get_table(S,T,&nr,&nc); + if(nr<1) + throw opkele::failed_lookup(OPKELE_CP_ + "couldn't retrieve valid unexpired assoc"); + assert(nr==1); assert(nc==6); + opkele::secret_t secret; opkele::util::decode_base64(T.get(1,2,nc),secret); + return opkele::assoc_t(new opkele::association( + "", h, T.get(1,1,nc), secret, + strtol(T.get(1,4,nc),0,0), + strtol(T.get(1,3,nc),0,0) )); + } + + string& alloc_nonce(string& nonce,bool stateless) { + uuid_t uuid; uuid_generate(uuid); + nonce += opkele::util::encode_base64(uuid,sizeof(uuid)); + sqlite3_mem_t + S = sqlite3_mprintf( + "INSERT INTO nonces" + " (n_once) VALUES (%Q)", + nonce.c_str() ); + db.exec(S); + return nonce; + } + bool check_nonce(const string& nonce) { + sqlite3_mem_t + S = sqlite3_mprintf( + "SELECT 1" + " FROM nonces" + " WHERE n_once=%Q AND n_itime IS NULL", + nonce.c_str()); + sqlite3_table_t T; + int nr,nc; + db.get_table(S,T,&nr,&nc); + return nr>=1; + } + void invalidate_nonce(const string& nonce) { + sqlite3_mem_t + S = sqlite3_mprintf( + "UPDATE nonces" + " SET n_itime=datetime('now')" + " WHERE n_once=%Q", + nonce.c_str()); + db.exec(S); + } + + const string get_op_endpoint() const { + return get_self_url(gw); + } + +}; + +int main(int argc,char *argv[]) { + try { + kingate::plaincgi_interface ci; + kingate::cgi_gateway gw(ci); + string op; + try { op = gw.get_param("op"); }catch(kingate::exception_notfound&) { } + string message; + if(op=="set_password") { + example_op_t OP(gw); + string password = gw.get_param("password"); + sqlite3_mem_t + Sget = sqlite3_mprintf("SELECT s_password FROM setup LIMIT 1"); + sqlite3_table_t T; int nr,nc; + OP.db.get_table(Sget,T,&nr,&nc); + if(nr>=1) + throw opkele::exception(OPKELE_CP_ "Password already set"); + sqlite3_mem_t + Sset = sqlite3_mprintf( + "INSERT INTO setup (s_password) VALUES (%Q)", + password.c_str()); + OP.db.exec(Sset); + op.clear(); + message = "password set"; + }else if(op=="login") { + example_op_t OP(gw); + string password = gw.get_param("password"); + sqlite3_mem_t + Sget = sqlite3_mprintf("SELECT s_password FROM setup LIMIT 1"); + sqlite3_table_t T; int nr,nc; + OP.db.get_table(Sget,T,&nr,&nc); + if(nr<1) + throw opkele::exception(OPKELE_CP_ "no password set"); + if(password!=T.get(1,0,nc)) + throw opkele::exception(OPKELE_CP_ "wrong password"); + OP.set_authorized(true); + op.clear(); + message = "logged in"; + OP.cookie_header(cout); + }else if(op=="logout") { + example_op_t OP(gw); + OP.set_authorized(false); + op.clear(); + message = "logged out"; + } + string om; + try { om = gw.get_param("openid.mode"); }catch(kingate::exception_notfound&) { } + if(op=="xrds") { + cout << + "Content-type: application/xrds+xml\n\n" + "" + "" + "" + "" + "" STURI_OPENID20 "" + "" << get_self_url(gw) << "" + ""; + if(gw.has_param("idsel")){ + cout << + "" + "" STURI_OPENID20_OP "" + "" << get_self_url(gw) << ""; + } + cout << + "" + ""; + }else if(op=="id_res" || op=="cancel") { + kingate_openid_message_t inm(gw); + example_op_t OP(gw); + if(gw.get_param("hts_id")!=OP.htc.get_value()) + throw opkele::exception(OPKELE_CP_ "toying around, huh?"); + OP.checkid_(inm,0); + OP.cookie_header(cout); + opkele::openid_message_t om; + if(op=="id_res") { + if(!OP.get_authorized()) + throw opkele::exception(OPKELE_CP_ "not logged in"); + if(OP.is_id_select()) { + OP.select_identity( get_self_url(gw), get_self_url(gw) ); + } + cout << + "Status: 302 Going back to RP with id_res\n" + "Location: " << OP.id_res(om).append_query(OP.get_return_to()) + << "\n\n"; + }else{ + cout << + "Status: 302 Going back to RP with cancel\n" + "Location: " << OP.cancel(om).append_query(OP.get_return_to()) + << "\n\n"; + } + om.to_keyvalues(clog); + }else if(om=="associate") { + kingate_openid_message_t inm(gw); + opkele::openid_message_t oum; + example_op_t OP(gw); + OP.associate(oum,inm); + cout << "Content-type: text/plain\n\n"; + oum.to_keyvalues(cout); + }else if(om=="checkid_setup") { + kingate_openid_message_t inm(gw); + example_op_t OP(gw); + OP.checkid_(inm,0); + OP.cookie_header(cout) << + "Content-type: text/html\n" + "\n" + + "" + "" + "test OP: confirm authentication" + "" + "" + "realm: " << OP.get_realm() << "
" + "return_to: " << OP.get_return_to() << "
" + "claimed_id: " << OP.get_claimed_id() << "
" + "identity: " << OP.get_identity() << "
"; + if(OP.is_id_select()) { + OP.select_identity( get_self_url(gw), get_self_url(gw) ); + cout << + "selected claimed_id: " << OP.get_claimed_id() << "
" + "selected identity: " << OP.get_identity() << "
"; + } + cout << + "
"; + inm.to_htmlhiddens(cout); + cout << + "" + "" + "" + "
" + "" + ""; + }else if(om=="check_authentication") { + kingate_openid_message_t inm(gw); + example_op_t OP(gw); + opkele::openid_message_t oum; + OP.check_authentication(oum,inm); + cout << "Content-type: text/plain\n\n"; + oum.to_keyvalues(cout); + oum.to_keyvalues(clog); + }else{ + example_op_t OP(gw); + string idsel; + if(gw.has_param("idsel")) + idsel = "&idsel=idsel"; + OP.cookie_header(cout) << + "Content-type: text/html\n" + "X-XRDS-Location: " << get_self_url(gw) << "?op=xrds" << idsel << "\n" + "\n" + + "" + "" + "test OP" + "" + "" + "" + "test openid 2.0 endpoint" + "
" + "XRDS document" + "
" + "

" << message << "

"; + sqlite3_mem_t + S = sqlite3_mprintf("SELECT s_password FROM setup LIMIT 1"); + sqlite3_table_t T; int nr,nc; + OP.db.get_table(S,T,&nr,&nc); + if(nr<1) { + cout << + "
" + "set password " + "" + "" + "" + "
"; + }else if(OP.get_authorized()) { + cout << + "
" + "logout"; + }else{ + cout << + "
" + "login " + "" + "" + "" + "
"; + } + cout << ""; + } +#ifdef OPKELE_HAVE_KONFORKA + }catch(konforka::exception& e) { +#else + }catch(std::exception& e){ +#endif + DOUT_("Oops: " << e.what()); + cout << "Content-Type: text/plain\n\n" + "Exception:\n" + " what: " << e.what() << endl; +#ifdef OPKELE_HAVE_KONFORKA + cout << " where: " << e.where() << endl; + if(!e._seen.empty()) { + cout << " seen:" << endl; + for(list::const_iterator + i=e._seen.begin();i!=e._seen.end();++i) { + cout << " " << i->c_str() << endl; + } + } +#endif + } +} -- cgit v0.9.0.2