From c4440be61d5ef7baffc1d1f5d66dd9cc4240b37b Mon Sep 17 00:00:00 2001
From: undefined <i@undefined.moe>
Date: Thu, 6 Oct 2022 19:57:50 +0800
Subject: [PATCH] core: sudo mode: save referer

---
 packages/hydrooj/src/handler/user.ts       | 11 +++++++----
 packages/hydrooj/src/service/decorators.ts |  2 ++
 2 files changed, 9 insertions(+), 4 deletions(-)

diff --git a/packages/hydrooj/src/handler/user.ts b/packages/hydrooj/src/handler/user.ts
index f5e7569f5..8f7fd04ea 100644
--- a/packages/hydrooj/src/handler/user.ts
+++ b/packages/hydrooj/src/handler/user.ts
@@ -142,14 +142,14 @@ class UserLoginHandler extends Handler {
 
 class UserSudoHandler extends Handler {
     async get() {
-        if (!this.session.sudoArgs) throw new ForbiddenError();
+        if (!this.session.sudoArgs?.method) throw new ForbiddenError();
         this.response.template = 'user_sudo.html';
     }
 
     @param('password', Types.String)
     @param('tfa', Types.String, true)
     async post(domainId: string, password: string, tfa = '') {
-        if (!this.session.sudoArgs) throw new ForbiddenError();
+        if (!this.session.sudoArgs?.method) throw new ForbiddenError();
         await Promise.all([
             this.limitRate('user_sudo', 60, 5, true),
             oplog.log(this, 'user.sudo', {}),
@@ -163,8 +163,8 @@ class UserSudoHandler extends Handler {
         if (this.session.sudoArgs.method.toLowerCase() !== 'get') {
             this.response.template = 'user_sudo_redirect.html';
             this.response.body = this.session.sudoArgs;
-            this.session.sudoArgs = null;
         } else this.response.redirect = this.session.sudoArgs.redirect;
+        this.session.sudoArgs.method = null;
     }
 }
 
@@ -197,7 +197,10 @@ export class UserRegisterHandler extends Handler {
             if (await user.getByEmail('system', mail)) throw new UserAlreadyExistError(mail);
             const mailDomain = mail.split('@')[1];
             if (await BlackListModel.get(`mail::${mailDomain}`)) throw new BlacklistedError(mailDomain);
-            await this.limitRate('send_mail', 3600, 30);
+            await Promise.all([
+                this.limitRate('send_mail', 3600, 30),
+                oplog.log(this, 'user.register', {}),
+            ]);
             const t = await token.add(
                 token.TYPE_REGISTRATION,
                 system.get('session.unsaved_expire_seconds'),
diff --git a/packages/hydrooj/src/service/decorators.ts b/packages/hydrooj/src/service/decorators.ts
index ca85b52dd..a0f9cee50 100644
--- a/packages/hydrooj/src/service/decorators.ts
+++ b/packages/hydrooj/src/service/decorators.ts
@@ -216,10 +216,12 @@ export function requireSudo(target: any, funcName: string, obj: any) {
     const originalMethod = obj.value;
     obj.value = function sudo(this: Handler, ...args: any[]) {
         if (this.session.sudo && Date.now() - this.session.sudo < Time.hour) {
+            this.request.headers.referer = this.session.sudoArgs.referer;
             return originalMethod.call(this, ...args);
         }
         this.session.sudoArgs = {
             method: this.request.method,
+            referer: this.request.headers.referer,
             args: this.args,
             redirect: this.request.path,
         };