core: import: check pid validity

1 year ago · 27833d6c53
parent de492550a6
commit 27833d6c53
1 changed files with 2 additions and 2 deletions
--- a/packages/hydrooj/src/model/problem.ts
+++ b/packages/hydrooj/src/model/problem.ts
@ -442,8 +442,8 @@ export class ProblemModel {
                if (!pdoc) continue;
                let pid = pdoc.pid;
                if (pid) {
-                    const current = await ProblemModel.get(domainId, pid);
-                    if (current) pid = undefined;
+                    if (!(/^[A-Za-z]+[0-9]*$/.test(pid))) pid = undefined;
+                    else if (await ProblemModel.get(domainId, pid)) pid = undefined;
                }
                const overrideContent = findOverrideContent(path.join(tmpdir, i));
                const docId = await ProblemModel.add(