dever 7 years ago
parent
commit
346a722c02
2 changed files with 6 additions and 2 deletions
  1. 5 1
      front/api/main.py
  2. 1 1
      service/convert.py

+ 5 - 1
front/api/main.py

@@ -48,6 +48,7 @@ class Common(object):
 
 		self.referer = self.request.headers.get("Referer")
 		self.host = Demeter.host(self.data['site']['link'])
+		uri = self.request.protocol + "://" + self.request.host + self.request.uri
 
 		if not self.referer:
 			self.out('验证失败')
@@ -55,6 +56,9 @@ class Common(object):
 		if self.host != Demeter.host(self.referer):
 			self.out('验证失败')
 
+		if self.referer == uri:
+			self.out('验证失败')
+
 
 		self.param = self.service.signature(self.data['site']['id'], appid, self.data['site']['appsecret'], timestamp, nonce, file, file_id, uid, status)
 
@@ -279,7 +283,7 @@ class view_path(Load):
 				url = url + '/main/view?path=' + str(path) + '&file=' + file['key'] + '&file_id=' + file_id + '&timestamp=' + timestamp + '&nonce=' + nonce + '&appid=' + appid + '&signature=' + signature + '&uid=' + uid + '&page='
 				content = content.replace('data-page-url="', 'data-page-url="' + url)
 
-				script = script + '<script>document.domain="'+self.host+'";</script>'
+				script = '<script>document.domain="'+self.host+'";</script>'
 				content = content.replace('</body>', script + '</body>')
 
 				style = '<style>body{zoom:100%;}#page-container{background-color: white;background-image: none;overflow: hidden;}#sidebar{background-color: white;background-image: none;}.pf{margin-top:0px;margin-bottom:0px;border:0px solid #bababa;box-shadow:none}</style>'

+ 1 - 1
service/convert.py

@@ -9,7 +9,7 @@ class Convert(object):
 		model.cdate.assign(num, exp='<=')
 		model.delete()
 
-	def signature(self, site_id, appid, appsecret, timestamp, nonce, file, file_id, uid, status):
+	def signature(self, site_id, appid, appsecret, timestamp, nonce, file, file_id, uid, status = 1):
 		sign = Demeter.sha1(appid + '&' + appsecret + '&' + str(timestamp) + '&' + str(nonce) + '&' + file + '&' + str(file_id) + '&' + str(uid));
 
 		time = Demeter.time()