def sample(p): if request.method == "POST": button = request.form['button'] if button == "Return Home": return redirect(url_for("home")) project = database.getProject(session['user'],p) return render_template("sample.html", project=project)
def lambda_handler(event, context): print(event) try: projectId = event['pathParameters']['projectId'] project = database.getProject(projectId) return response_formatter.formatSuccessfulResponse(project) except Exception as e: error_handler.handleError(e)
def creator2(prname): user = session['user'] if request.method == "POST": button = request.form['button'] if button == "Save": pname = prname user = session["user"] html = request.form["pagehtml"] database.addProject(user,pname,html) return redirect(url_for("creator2",prname=prname)) if button == "Return to Homepage": return redirect(url_for("home")) if button == "Publish to Public Projects": pname = prname user = session['user'] html = request.form["pagehtml"] database.addPublicProject(user,pname,html) if button == "Find User": pname = prname use = session["user"] tar = request.form["colabname"] if database.nameInSys(tar) == 0: tar = "ERR0RZ" html = database.getProject(use,pname) else: html = database.getProject(use,pname) database.makeCollab(use,pname,tar) database.addProject(tar,pname,html) database.makeCollab(tar,pname,use) return render_template("creator.html",p=html,colab=[tar,pname]) if button == str(database.whoCollab(session['user'],prname)): use = session["user"] pname = prname database.stopCollab(use,pname) project = database.getProject(user,prname) if database.isCollab(user,prname): return render_template("creator.html",p=project,colab=[database.whoCollab(user,prname),prname]) return render_template("creator.html",p=project)
def lambda_handler(event, context): print(event) message = json.loads(event['Records'][0]["Sns"]["Message"]) project_id = message['project_id'] scan_id = message['scan_id'] target_id = message['target_id'] resource_id = message['resource_id'] url = message['url'] url_hash = hashlib.md5(url.encode('utf-8')).hexdigest() width = "1200" height = "1024" timeout = "5" try: cmd = [ "./phantomjs/phantomjs_linux-x86_64", "--debug=yes", "--ignore-ssl-errors=true", "./phantomjs/screenshot.js", url, "/tmp/" + url_hash + ".png", width, height, timeout ] results = subprocess.Popen(cmd, stdout=subprocess.PIPE) print(results.stdout.read()) thumbnail_path = create_thumbnail("/tmp/" + url_hash + ".png", url_hash) project_record = database.getProject(project_id) target_record = database.get_target_record(target_id) bucket = os.environ['CoreBucketRef'] file_name = '/tmp/' + url_hash + ".png" key_name = url_hash + '.png' thumbnail_key_name = url_hash + '-320.png' target = target_record["host"] if target_record[ "host"] else target_record["ip"] obj_path = "{0}/{1}/{2}/{3}".format( project_record["name"], target, urlparse(url).path.rstrip('/').strip('/'), key_name) thumbnail_obj_path = "{0}/{1}/{2}/{3}".format( project_record["name"], target, urlparse(url).path.rstrip('/').strip('/'), thumbnail_key_name) s3 = boto3.client('s3') print(os.listdir('/tmp')) print(file_name) s3.upload_file(file_name, bucket, obj_path) s3.upload_file(thumbnail_path, bucket, thumbnail_obj_path) database.create_screenshot_record(project_id, scan_id, target_id, resource_id, obj_path, thumbnail_obj_path) except Exception as e: error_handler.handleError(e)
def creator3(usname,prname): user = usname if request.method == "POST": button = request.form['button'] if button == "Save": pname = prname use = session["user"] return redirect(url_for("creator2",prname=prname)) if button == "Return to Homepage": return redirect(url_for("home")) if button == "Publish to Public Projects": pname = prname use = session['user'] html = request.form["pagehtml"] database.addPublicProject(use,pname,html) project = database.getProject(user,prname) return render_template("creator.html",p=project)