@ -17,147 +17,147 @@ from pyi2phosts.lib.validation import validate_hostname
from pyi2phosts . lib . validation import validate_b64hash from pyi2phosts . lib . validation import validate_b64hash
class AddForm ( forms . ModelForm ) : class AddForm ( forms . ModelForm ) :
""" """
This is our class for host - add form . It ' s based on django ' s ModelForm This is our class for host - add form . It ' s based on django ' s ModelForm
and uses our model " i2phost " ( see postkey / models . py ) and uses our model " i2phost " ( see postkey / models . py )
""" """
class Meta : class Meta :
model = i2phost model = i2phost
fields = ( ' name ' , ' b64hash ' , ' description ' ) fields = ( ' name ' , ' b64hash ' , ' description ' )
widgets = { widgets = {
' name ' : forms . TextInput ( attrs = { ' size ' : ' 67 ' } ) , ' name ' : forms . TextInput ( attrs = { ' size ' : ' 67 ' } ) ,
' b64hash ' : forms . Textarea ( attrs = { ' rows ' : ' 1 ' , ' cols ' : ' 100 ' } ) , ' b64hash ' : forms . Textarea ( attrs = { ' rows ' : ' 1 ' , ' cols ' : ' 100 ' } ) ,
' description ' : forms . Textarea ( attrs = { ' rows ' : ' 2 ' , ' cols ' : ' 72 ' } ) ' description ' : forms . Textarea ( attrs = { ' rows ' : ' 2 ' , ' cols ' : ' 72 ' } )
} }
def clean_name ( self ) : def clean_name ( self ) :
""" Validate hostname """ """ Validate hostname """
data = self . cleaned_data [ ' name ' ] data = self . cleaned_data [ ' name ' ]
log . debug ( u ' hostname: %s ' , self . data [ ' name ' ] ) log . debug ( u ' hostname: %s ' , self . data [ ' name ' ] )
data = validate_hostname ( data ) data = validate_hostname ( data )
# Another set of reserved hostnames (suggested by zzz) # Another set of reserved hostnames (suggested by zzz)
if re . search ( r ' (^| \ .)(i2p|i2p2|geti2p|mail|project|i2project|i2pproject|i2p-project).i2p$ ' , data ) : if re . search ( r ' (^| \ .)(i2p|i2p2|geti2p|mail|project|i2project|i2pproject|i2p-project).i2p$ ' , data ) :
raise forms . ValidationError ( _ ( ' Trying to use hostname from additional reserved set ' ) ) raise forms . ValidationError ( _ ( ' Trying to use hostname from additional reserved set ' ) )
return data return data
def clean_b64hash ( self ) : def clean_b64hash ( self ) :
""" Validate base64 hash """ """ Validate base64 hash """
data = self . cleaned_data [ ' b64hash ' ] data = self . cleaned_data [ ' b64hash ' ]
log . debug ( u ' hash: %s ' , self . data [ ' b64hash ' ] ) log . debug ( u ' hash: %s ' , self . data [ ' b64hash ' ] )
data = validate_b64hash ( data ) data = validate_b64hash ( data )
return data return data
def is_valid ( self ) : def is_valid ( self ) :
""" Log validation errors """ """ Log validation errors """
is_valid = super ( AddForm , self ) . is_valid ( ) is_valid = super ( AddForm , self ) . is_valid ( )
if not is_valid : if not is_valid :
for field in self . errors . keys ( ) : for field in self . errors . keys ( ) :
log . info ( ' ValidationError: [ %s ]: \" %s \" %s ' , log . info ( ' ValidationError: [ %s ]: \" %s \" %s ' ,
field , self . data [ field ] , self . errors [ field ] . as_text ( ) ) field , self . data [ field ] , self . errors [ field ] . as_text ( ) )
return is_valid return is_valid
class SubdomainVerifyForm ( forms . Form ) : class SubdomainVerifyForm ( forms . Form ) :
""" Form for displaying verification filename and code when verifying a subdomain """ """ Form for displaying verification filename and code when verifying a subdomain """
filename = forms . CharField ( label = _ ( ' Filename ' ) , widget = forms . TextInput ( attrs = { filename = forms . CharField ( label = _ ( ' Filename ' ) , widget = forms . TextInput ( attrs = {
' size ' : ' 20 ' , ' size ' : ' 20 ' ,
' readonly ' : ' readonly ' , ' readonly ' : ' readonly ' ,
' onclick ' : ' this.select(); ' , ' onclick ' : ' this.select(); ' ,
} ) ) } ) )
def save_host ( request ) : def save_host ( request ) :
""" Function for saving hosts after validation or subdomain verification """ """ Function for saving hosts after validation or subdomain verification """
# avoid race conditions # avoid race conditions
try : try :
h = i2phost . objects . get ( name = request . session [ ' hostname ' ] ) h = i2phost . objects . get ( name = request . session [ ' hostname ' ] )
except i2phost . DoesNotExist : except i2phost . DoesNotExist :
host = i2phost ( name = request . session [ ' hostname ' ] , host = i2phost ( name = request . session [ ' hostname ' ] ,
b64hash = request . session [ ' b64hash ' ] , b64hash = request . session [ ' b64hash ' ] ,
description = request . session [ ' description ' ] , description = request . session [ ' description ' ] ,
date_added = datetime . datetime . utcnow ( ) ) date_added = datetime . datetime . utcnow ( ) )
host . save ( ) host . save ( )
return redirect ( ' pyi2phosts.postkey.views.success ' ) return redirect ( ' pyi2phosts.postkey.views.success ' )
else : else :
log . warning ( ' refusing to save already existed host: %s ' , request . session [ ' hostname ' ] ) log . warning ( ' refusing to save already existed host: %s ' , request . session [ ' hostname ' ] )
request . session . flush ( ) request . session . flush ( )
return redirect ( ' / ' ) return redirect ( ' / ' )
def addkey ( request ) : def addkey ( request ) :
if request . method == ' POST ' : if request . method == ' POST ' :
form = AddForm ( request . POST ) form = AddForm ( request . POST )
if form . is_valid ( ) : if form . is_valid ( ) :
request . session [ ' hostname ' ] = form . cleaned_data [ ' name ' ] request . session [ ' hostname ' ] = form . cleaned_data [ ' name ' ]
request . session [ ' b64hash ' ] = form . cleaned_data [ ' b64hash ' ] request . session [ ' b64hash ' ] = form . cleaned_data [ ' b64hash ' ]
request . session [ ' description ' ] = form . cleaned_data [ ' description ' ] request . session [ ' description ' ] = form . cleaned_data [ ' description ' ]
if form . cleaned_data [ ' name ' ] . count ( ' . ' ) > 1 : if form . cleaned_data [ ' name ' ] . count ( ' . ' ) > 1 :
return redirect ( ' pyi2phosts.postkey.views.subdomain ' ) return redirect ( ' pyi2phosts.postkey.views.subdomain ' )
else : else :
log . debug ( ' submit is valid, saving ' ) log . debug ( ' submit is valid, saving ' )
s = save_host ( request ) s = save_host ( request )
return s return s
else : else :
form = AddForm ( ) form = AddForm ( )
return render_to_response ( ' postkey.html ' , { return render_to_response ( ' postkey.html ' , {
' title ' : settings . SITE_NAME , ' title ' : settings . SITE_NAME ,
' form ' : form , ' form ' : form ,
} , context_instance = RequestContext ( request ) ) } , context_instance = RequestContext ( request ) )
def success ( request ) : def success ( request ) :
if ' hostname ' in request . session : if ' hostname ' in request . session :
hn = request . session [ ' hostname ' ] hn = request . session [ ' hostname ' ]
request . session . flush ( ) request . session . flush ( )
return render_to_response ( ' success_submission.html ' , { return render_to_response ( ' success_submission.html ' , {
' title ' : settings . SITE_NAME , ' title ' : settings . SITE_NAME ,
' hostname ' : hn , ' hostname ' : hn ,
} , context_instance = RequestContext ( request ) ) } , context_instance = RequestContext ( request ) )
else : else :
return redirect ( ' / ' ) return redirect ( ' / ' )
def subdomain ( request ) : def subdomain ( request ) :
""" Subdomain verification """ """ Subdomain verification """
if request . method == ' POST ' : if request . method == ' POST ' :
form = SubdomainVerifyForm ( request . POST ) form = SubdomainVerifyForm ( request . POST )
if form . is_valid ( ) : if form . is_valid ( ) :
# do verification here, then redirect to success # do verification here, then redirect to success
proxy_handler = urllib2 . ProxyHandler ( { ' http ' : settings . EEPROXY_URL } ) proxy_handler = urllib2 . ProxyHandler ( { ' http ' : settings . EEPROXY_URL } )
opener = urllib2 . build_opener ( proxy_handler ) opener = urllib2 . build_opener ( proxy_handler )
if ' topdomain ' in request . session and ' v_filename ' in request . session : if ' topdomain ' in request . session and ' v_filename ' in request . session :
url = ' http:// ' + request . session [ ' topdomain ' ] + ' / ' + request . session [ ' v_filename ' ] url = ' http:// ' + request . session [ ' topdomain ' ] + ' / ' + request . session [ ' v_filename ' ]
else : else :
log . warning ( ' trying to call subdomain validation without a session ' ) log . warning ( ' trying to call subdomain validation without a session ' )
return redirect ( ' / ' ) return redirect ( ' / ' )
log . info ( ' starting http-verification of subdomain: %s ' , request . session [ ' hostname ' ] ) log . info ( ' starting http-verification of subdomain: %s ' , request . session [ ' hostname ' ] )
try : try :
log . debug ( ' trying to open %s ' , url ) log . debug ( ' trying to open %s ' , url )
resp = opener . open ( url , timeout = 60 ) resp = opener . open ( url , timeout = 60 )
except urllib2 . URLError , e : except urllib2 . URLError , e :
if hasattr ( e , ' reason ' ) : if hasattr ( e , ' reason ' ) :
log . warning ( ' %s : failed to reach server, reason: %s ' , request . session [ ' topdomain ' ] , e . reason ) log . warning ( ' %s : failed to reach server, reason: %s ' , request . session [ ' topdomain ' ] , e . reason )
elif hasattr ( e , ' code ' ) : elif hasattr ( e , ' code ' ) :
log . warning ( ' %s can \' t finish the request, error code: %s ' , log . warning ( ' %s can \' t finish the request, error code: %s ' ,
request . session [ ' topdomain ' ] , e . code ) request . session [ ' topdomain ' ] , e . code )
return render_to_response ( ' subdomain_http_verify_failure.html ' , { return render_to_response ( ' subdomain_http_verify_failure.html ' , {
' title ' : settings . SITE_NAME , ' title ' : settings . SITE_NAME ,
' code ' : e . code , ' code ' : e . code ,
} , context_instance = RequestContext ( request ) ) } , context_instance = RequestContext ( request ) )
else : else :
log . debug ( ' subdomain verification success, saving host ' ) log . debug ( ' subdomain verification success, saving host ' )
s = save_host ( request ) s = save_host ( request )
return s return s
else : else :
# generate verification code and display info page to user # generate verification code and display info page to user
v_filename = ' ' . join ( [ random . choice ( string . letters + string . digits ) for x in xrange ( 16 ) ] ) v_filename = ' ' . join ( [ random . choice ( string . letters + string . digits ) for x in xrange ( 16 ) ] )
if ' hostname ' in request . session : if ' hostname ' in request . session :
m = re . match ( ' .+ \ .(.+ \ .i2p$) ' , request . session [ ' hostname ' ] ) m = re . match ( ' .+ \ .(.+ \ .i2p$) ' , request . session [ ' hostname ' ] )
topdomain = m . group ( 1 ) topdomain = m . group ( 1 )
else : else :
return redirect ( ' / ' ) return redirect ( ' / ' )
# save needed variables in session data because otherwise it will be lost # save needed variables in session data because otherwise it will be lost
request . session [ ' v_filename ' ] = v_filename request . session [ ' v_filename ' ] = v_filename
request . session [ ' topdomain ' ] = topdomain request . session [ ' topdomain ' ] = topdomain
form = SubdomainVerifyForm ( { ' filename ' : v_filename } ) form = SubdomainVerifyForm ( { ' filename ' : v_filename } )
return render_to_response ( ' subdomain_http_verify.html ' , { return render_to_response ( ' subdomain_http_verify.html ' , {
' title ' : settings . SITE_NAME , ' title ' : settings . SITE_NAME ,
' hostname ' : request . session [ ' hostname ' ] , ' hostname ' : request . session [ ' hostname ' ] ,
' topdomain ' : topdomain , ' topdomain ' : topdomain ,
' form ' : form , ' form ' : form ,
} , context_instance = RequestContext ( request ) ) } , context_instance = RequestContext ( request ) )
log = get_logger ( filename = settings . LOG_FILE , log_level = settings . LOG_LEVEL ) log = get_logger ( filename = settings . LOG_FILE , log_level = settings . LOG_LEVEL )
Hidden Z
9 years ago
