From 27833d6c53277f6cf496d390a3e175b3ebc0d338 Mon Sep 17 00:00:00 2001 From: undefined Date: Thu, 18 May 2023 15:00:44 +0800 Subject: [PATCH] core: import: check pid validity --- packages/hydrooj/src/model/problem.ts | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/packages/hydrooj/src/model/problem.ts b/packages/hydrooj/src/model/problem.ts index d3405802d..72b8336cb 100644 --- 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(