HTTP registry browser. Added: trunk/rosapps/net/netreg/ Added: trunk/rosapps/net/netreg/makefile Added: trunk/rosapps/net/netreg/netreg.cpp Added: trunk/rosapps/net/netreg/netreg.rc _____
Added: trunk/rosapps/net/netreg/makefile --- trunk/rosapps/net/netreg/makefile 2005-01-18 04:14:13 UTC (rev 13111) +++ trunk/rosapps/net/netreg/makefile 2005-01-18 04:25:27 UTC (rev 13112) @@ -0,0 +1,22 @@
+ +PATH_TO_TOP=../../../reactos + +TARGET_TYPE = program + +TARGET_APPTYPE = console + +TARGET_NAME = netreg + +TARGET_SDKLIBS = + +TARGET_OBJECTS = $(TARGET_NAME).o + +TARGET_GCCLIBS = ws2_32 stdc++ + +include $(PATH_TO_TOP)/rules.mak + +include $(TOOLS_PATH)/helper.mk + +TARGET_CPPFLAGS = -I$(PATH_TO_TOP)/w32api/include -D__USE_W32API + +# EOF _____
Added: trunk/rosapps/net/netreg/netreg.cpp --- trunk/rosapps/net/netreg/netreg.cpp 2005-01-18 04:14:13 UTC (rev 13111) +++ trunk/rosapps/net/netreg/netreg.cpp 2005-01-18 04:25:27 UTC (rev 13112) @@ -0,0 +1,522 @@
+/* + * COPYRIGHT: See COPYING in the top level directory + * PROJECT: ReactOS test application + * FILE: apps/net/netreg/netreg.cpp + * PURPOSE: HTTP Registry Server + * PROGRAMMERS: Art Yerkes (arty@users.sf.net) + * REVISIONS: + * 01-17-2005 arty -- initial + */ +#include <windows.h> +#include <winsock.h> +#include <stdlib.h> +#include <map> +#include <string> +#include <sstream> +#include <iomanip> + +using std::hex; +using std::setw; +using std::setfill; +using std::map; +using std::string; +using std::ostringstream; + +const char *root_entries[] = { + "HKEY_LOCAL_MACHINE", + "HKEY_CURRENT_USER", + "HKEY_CLASSES_ROOT", + "HKEY_CURRENT_CONFIG", + "HKEY_USERS", + 0 +}; + +const HKEY root_handles[] = { + HKEY_LOCAL_MACHINE, + HKEY_CURRENT_USER, + HKEY_CLASSES_ROOT, + HKEY_CURRENT_CONFIG, + HKEY_USERS +}; + +class RequestHandler { +public: + RequestHandler( SOCKET s ) : socket( s ), state( NO_REQUEST_YET ) {} + ~RequestHandler() { closesocket( socket ); } + void RecvData( string input ) { + full_input += input; + if( full_input.find( "\r\n\r\n" ) != string::npos ) { + // Full request received... + size_t space_pos = full_input.find( ' ' ); + if( space_pos == string::npos ) { state = SHOULD_DIE; return; } + string method = full_input.substr( 0, space_pos ); + if( method != "GET" ) { state = SHOULD_DIE; return; } + space_pos++; + if( full_input[space_pos] != '/' ) { state = SHOULD_DIE; return; } + space_pos++; + string reg_key_and_remainder = + full_input.substr( space_pos, full_input.size() - space_pos ); + space_pos = reg_key_and_remainder.find( ' ' ); + if( space_pos == string::npos ) { state = SHOULD_DIE; return; } + string reg_key_name = reg_key_and_remainder.substr( 0, space_pos ); + process_request( urldec( reg_key_name ) ); + state = REQUEST_RECVD_SENDING_REPLY; + } + } + void OkToSend() { + int rv = send( socket, + remaining_output.c_str(), + remaining_output.size(), 0 ); + if( rv < 0 ) { + state = SHOULD_DIE; + return; + } else { + remaining_output = + remaining_output.substr( rv, remaining_output.size() - rv ); + if( remaining_output.size() == 0 ) { + state = SHOULD_DIE; + } + } + } + + SOCKET GetSocket() const { return socket; } + + bool ShouldDie() const { + return state == SHOULD_DIE; + } + + bool WantPollout() const { + return state == REQUEST_RECVD_SENDING_REPLY; + } + + +private: + string urlenc( string in ) { + ostringstream out; + + for( string::iterator i = in.begin(); + i != in.end(); + i++ ) { + if( isalnum( *i ) || *i == '/' ) + out << *i; + else { + char minibuf[10]; + sprintf( minibuf, "%02x", *i ); + out << "%" << minibuf; + } + } + + return out.str(); + } + + string urldec( string in ) { + string out; + + for( string::iterator i = in.begin(); + i != in.end(); + i++ ) { + if( *i == '%' ) { + char buf[3]; + int res = ' '; + + i++; + if( i != in.end() ) { + buf[0] = *i; + i++; + if( i != in.end() ) { + buf[1] = *i; + buf[2] = 0; + sscanf( buf, "%x", &res ); + fprintf( stderr, "Interpreting %c%c as %02x\n", + buf[0], buf[1], + res ); + out += (char)res; + } + } + } else out += *i; + } + + return out; + } + + string dump_one_line( const char *data, int llen, int len, int addr ) { + ostringstream out; + int i; + + out << setw( 8 ) << setfill( '0' ) << hex << addr << ": "; + + for( i = 0; i < llen; i++ ) { + if( i < len ) out << setw( 2 ) << setfill( '0' ) << hex << + (data[i] & 0xff) << " "; + else out << " "; + } + + out << " : "; + + for( i = 0; i < llen; i++ ) { + if( i < len && i < llen && + data[i] >= ' ' && data[i] < 0x7f ) out << data[i]; else out << '.'; + } + + out << "\n"; + + return out.str(); + } + + string bindump( const char *data, int len ) { + const char *end = data + len; + string out; + int addr = 0; + + out += "<pre>"; + + while( data < end ) { + out += dump_one_line( data, 16, end - data, addr ); + addr += 16; + data += 16; + } + + out += "</pre>"; + + return out; + } + + string present_value( DWORD type, const char *data, DWORD len ) { + switch( type ) { + default: + return bindump( data, len ); + } + } + + void process_valid_request( HKEY open_reg_key, string key_name ) { + size_t ending_slash; + string up_level; + ostringstream text_out; + + DWORD num_sub_keys; + DWORD max_subkey_len; + DWORD num_values; + DWORD max_value_name_len; + DWORD max_value_len; + + char *value_name_buf; + char *value_buf; + char *key_name_buf; + + if( RegQueryInfoKey( open_reg_key, + NULL, + NULL, + NULL, + &num_sub_keys, + &max_subkey_len, + NULL, + &num_values, + &max_value_name_len, + &max_value_len, + NULL, + NULL ) != ERROR_SUCCESS ) { + process_invalid_request( key_name ); + return; + } + + value_name_buf = new char [max_value_name_len+1]; + value_buf = new char [max_value_len+1]; + key_name_buf = new char [max_subkey_len+1]; + + ending_slash = key_name.rfind( '/' ); + if( ending_slash != string::npos ) + up_level = key_name.substr( 0, ending_slash ); + + text_out << "HTTP/1.0 200 OK\r\n" + << "Content-Type: text/html\r\n" + << "\r\n" + << "<html><head><title>Registry Key `" + << key_name + << "'</title></head><body>\r\n" + << "<h1>Registry Key `" << key_name << "'</h1>\r\n" + << "<a href='/" << urlenc(up_level) + << "'>(Up one level)</a><p>\r\n" + << "<h2>Subkeys:</h2><table border='1'>\r\n"; + + DWORD which_index; + DWORD key_name_size; + + for( which_index = 0; which_index < num_sub_keys; which_index++ ) { + key_name_size = max_subkey_len+1; + RegEnumKeyEx( open_reg_key, + which_index, + key_name_buf, + &key_name_size, + NULL, + NULL, + NULL, + NULL ); + text_out << "<tr><td><a href='/" << urlenc(key_name) << "/" + << urlenc(string(key_name_buf,key_name_size)) << "'>" + << string(key_name_buf,key_name_size) + << "</a></td></tr>\r\n"; + } + + text_out << "</table><h2>Values:</h2><table border='1'>\r\n"; + + DWORD value_name_size; + DWORD value_data_size; + DWORD value_type; + + for( which_index = 0; which_index < num_values; which_index++ ) { + value_name_size = max_value_name_len+1; + value_data_size = max_value_len+1; + + RegEnumValue( open_reg_key, + which_index, + value_name_buf, + &value_name_size, + NULL, + &value_type, + (BYTE *)value_buf, + &value_data_size ); + + text_out << "<tr><td><b>" << string(value_name_buf,value_name_size) + << "</b></td><td>" + << present_value( value_type, value_buf, value_data_size ) + << "</td></tr>"; + } + + text_out << "</ul></body></html>\r\n"; + + delete [] key_name_buf; + delete [] value_name_buf; + delete [] value_buf; + + remaining_output = text_out.str(); + } + + void process_invalid_request( string reg_key ) { + ostringstream text_out; + text_out << "HTTP/1.0 404 Not Found\r\n" + << "Content-Type: text/html\r\n" + << "\r\n" + << "<html><head><title>Can't find registry key `" + << reg_key + << "'</title></head><body>\r\n" + << "<H1>Can't find registry key `" + << reg_key + << "'</H1>\r\n" + << "The registry key doesn't exist in the local registry.\r\n" + << "</body></html>\r\n"; + + remaining_output = text_out.str(); + } + + void process_root_request() { + ostringstream text_out; + int i; + + text_out << "HTTP/1.0 200 OK\r\n" + << "Content-Type: text/html\r\n" + << "\r\n" + << "<html><head><title>Registry Browser</title></head>\r\n" + << "<body>\r\n" + << "<H1>Registry Browser</H1>" + << "You can use this interface to browse the registry." + << "You will be presented with one registry key at a time and " + << "the decendents.\r\n" + << "<h2>Root Level</h2>\r\n" + << "Subkeys:<ul>\r\n"; + + for( i = 0; root_entries[i]; i++ ) + text_out << "<li>" + << "<a href='/" << urlenc(root_entries[i]) + << "'>" << root_entries[i] + << "</a></li>\r\n"; + + text_out << "</ul></body></html>\r\n"; + + remaining_output = text_out.str(); + } + + void process_request( string reg_key ) { + int i; + bool is_predefined_key = true; + + if( reg_key == "" ) { process_root_request(); return; } + HKEY hRegKey = 0; + + // Parse the key name... + size_t slash = reg_key.find( '/' ); + string reg_initial = ""; + + if( slash == string::npos ) // A root key... + reg_initial = reg_key; + else // Any other key + reg_initial = reg_key.substr( 0, slash ); + + fprintf( stderr, "reg_init = %s, reg_key = %s\n", + reg_initial.c_str(), + reg_key.c_str() ); + + for( i = 0; root_entries[i]; i++ ) + if( reg_initial == root_entries[i] ) hRegKey = root_handles[i]; + + if( hRegKey != 0 && reg_initial != reg_key ) { + size_t start_of_reg_path = reg_initial.size() + 1; + string reg_path = reg_key.substr( start_of_reg_path, + reg_key.size() - start_of_reg_path ); + + string reg_open_path = reg_path; + do { + slash = reg_open_path.find( '/' ); + string reg_single_key = reg_open_path; + + if( slash != string::npos ) { + reg_single_key = reg_open_path.substr( 0, slash ); + reg_open_path = reg_open_path.substr( slash+1, + reg_open_path.size() ); + } + + HKEY oldKey = hRegKey; + + fprintf( stderr, "Opening %s\n", reg_single_key.c_str() ); + + if( RegOpenKey( hRegKey, reg_single_key.c_str(), &hRegKey ) != + ERROR_SUCCESS ) { + hRegKey = 0; + break; + } else RegCloseKey( oldKey ); + + is_predefined_key = false; + } while( slash != string::npos ); + } + + if( hRegKey == 0 ) process_invalid_request( reg_key ); + else { + process_valid_request( hRegKey, reg_key ); + if( !is_predefined_key ) RegCloseKey( hRegKey ); + } + } + + typedef enum _RHState { + NO_REQUEST_YET, + REQUEST_RECVD_SENDING_REPLY, + SHOULD_DIE + } RHState; + + string full_input; + string remaining_output; + RHState state; + SOCKET socket; +}; + +SOCKET make_listening_socket( int port ) { + struct sockaddr_in sa; + + ZeroMemory( &sa, sizeof( sa ) ); + + sa.sin_family = PF_INET; + sa.sin_port = ntohs( port ); + + fprintf( stderr, "Creating the listener\n" ); + SOCKET l = socket( PF_INET, SOCK_STREAM, 0 ); + fprintf( stderr, "Socket %x\n", l ); + + if( l == INVALID_SOCKET ) return l; + if( bind( l, (struct sockaddr *)&sa, sizeof( sa ) ) < 0 ) { + fprintf( stderr, "Bad response from bind: %d\n", WSAGetLastError() ); + closesocket( l ); return INVALID_SOCKET; + } + if( listen( l, 5 ) < 0 ) { + fprintf( stderr, "Listening: %d\n", WSAGetLastError() ); + closesocket( l ); + return INVALID_SOCKET; + } + + return l; +} + +int main( int argc, char **argv ) { + WSADATA wdWinsockData; + map<SOCKET,RequestHandler *> requests; + fd_set pollin,pollout,pollerr; + SOCKET listen_socket; + int i; + int port_to_listen = 80; + int active_fds = 0; + + for( i = 1; i < argc; i++ ) { + if( string( "-p" ) == argv[i] ) { + i++; + if( i < argc ) port_to_listen = atoi( argv[i] ); + } + } + + WSAStartup( 0x0101, &wdWinsockData ); + + listen_socket = make_listening_socket( port_to_listen ); + if( listen_socket == INVALID_SOCKET ) return 1; + + while( true ) { + FD_ZERO( &pollin ); + FD_ZERO( &pollout ); + FD_ZERO( &pollerr ); + active_fds = listen_socket + 1; + + for( std::map<SOCKET,RequestHandler *>::iterator i = requests.begin(); + i != requests.end(); + i++ ) { + if( i->second->ShouldDie() ) { + delete i->second; + requests.erase( i ); + i = requests.begin(); + break; + } + + FD_SET(i->first,&pollin); + FD_SET(i->first,&pollerr); + + if( i->first > active_fds ) active_fds = i->first + 1; + + if( i->second->WantPollout() ) FD_SET(i->first,&pollout); + } + + FD_SET(listen_socket,&pollin); + + struct timeval tv; + + active_fds = select( active_fds, &pollin, &pollout, &pollerr, NULL ); + + if( active_fds > 0 ) { + if( FD_ISSET(listen_socket,&pollin) ) { + SOCKET ns = accept( listen_socket, NULL, NULL ); + if( ns != INVALID_SOCKET ) { + requests.insert( std::make_pair( ns, new RequestHandler( ns ) ) ); + } + } + + for( std::map<SOCKET,RequestHandler *>::iterator i = requests.begin(); + i != requests.end(); + i++ ) { + if( FD_ISSET(i->first,&pollin) ) { + char inbuf[1024]; + int rv = recv(i->first,inbuf,1024,0); + if( rv < 0 ) { + delete i->second; + requests.erase( i ); + i = requests.begin(); + break; + } else i->second->RecvData( string( inbuf, rv ) ); + } + if( FD_ISSET(i->first,&pollout) ) { + i->second->OkToSend(); + } + if( FD_ISSET(i->first,&pollerr) ) { + delete i->second; + requests.erase( i ); + i = requests.begin(); + break; + } + } + } + } + + WSACleanup(); +} _____
Added: trunk/rosapps/net/netreg/netreg.rc --- trunk/rosapps/net/netreg/netreg.rc 2005-01-18 04:14:13 UTC (rev 13111) +++ trunk/rosapps/net/netreg/netreg.rc 2005-01-18 04:25:27 UTC (rev 13112) @@ -0,0 +1,6 @@
+/* $Id: niclist.rc 11299 2004-10-16 22:30:19Z gvg $ */ + +#define REACTOS_STR_FILE_DESCRIPTION "ReactOS HTTP Registry Browser\0" +#define REACTOS_STR_INTERNAL_NAME "netreg\0" +#define REACTOS_STR_ORIGINAL_FILENAME "netreg.exe\0" +#include <reactos/version.rc>
-
arty＠svn.reactos.com