mirror of
https://github.com/ajayyy/SponsorBlockServer.git
synced 2024-11-10 01:02:30 +01:00
more lenient privateIDUsername checks
- disallow username = privateID - disallow username = other privateID on username table if length > minLength
This commit is contained in:
parent
b855eea349
commit
a384079562
3 changed files with 125 additions and 64 deletions
|
@ -33,14 +33,9 @@ export async function setUsername(req: Request, res: Response): Promise<Response
|
|||
userName = userName.replace(/[\u0000-\u001F\u007F-\u009F]/g, "");
|
||||
|
||||
// check privateID against publicID
|
||||
/*
|
||||
if (!await checkPrivateUsername(userName, userID)) {
|
||||
return res.sendStatus(400);
|
||||
}
|
||||
*/
|
||||
if (userName == userID) {
|
||||
return res.sendStatus(400);
|
||||
}
|
||||
|
||||
if (adminUserIDInput != undefined) {
|
||||
//this is the admin controlling the other users account, don't hash the controling account's ID
|
||||
|
@ -100,11 +95,10 @@ export async function setUsername(req: Request, res: Response): Promise<Response
|
|||
}
|
||||
|
||||
async function checkPrivateUsername(username: string, userID: string): Promise<boolean> {
|
||||
const userIDHash = await getHashCache(userID);
|
||||
if (username == userID) return false;
|
||||
if (username.length <= config.minUserIDLength) return true; // don't check for cross matches <= 30 characters
|
||||
const userNameHash = await getHashCache(username);
|
||||
if (userIDHash == userNameHash) return false;
|
||||
const sponsorTimeRow = await db.prepare("get", `SELECT "userID" FROM "sponsorTimes" WHERE "userID" = ? LIMIT 1`, [userNameHash]);
|
||||
const userNameRow = await db.prepare("get", `SELECT "userID" FROM "userNames" WHERE "userID" = ? LIMIT 1`, [userNameHash]);
|
||||
if ((sponsorTimeRow || userNameRow)?.userID) return false;
|
||||
if (userNameRow?.userID) return false;
|
||||
return true;
|
||||
}
|
|
@ -22,16 +22,6 @@ const user07PrivateUserID = "setUsername_07";
|
|||
const username07 = "Username 07";
|
||||
const user08PrivateUserID = "setUsername_08";
|
||||
|
||||
// private = public cases
|
||||
// user09 - username === privateID
|
||||
const user09PrivateUserID = "setUsername_09";
|
||||
// user 10/11 - user 11 username === user 10 privateID
|
||||
const user10PrivateUserID = "setUsername_10_collision";
|
||||
const username10 = "setUsername_10";
|
||||
const user11PrivateUserID = "setUsername_11";
|
||||
const user12PrivateUserID = "setUsername_12";
|
||||
const username12 = "Username 12";
|
||||
|
||||
async function addUsername(userID: string, userName: string, locked = 0) {
|
||||
await db.prepare("run", 'INSERT INTO "userNames" ("userID", "userName", "locked") VALUES(?, ?, ?)', [userID, userName, locked]);
|
||||
await addLogUserNameChange(userID, userName);
|
||||
|
@ -40,7 +30,7 @@ async function addUsername(userID: string, userName: string, locked = 0) {
|
|||
async function getUsernameInfo(userID: string): Promise<{ userName: string, locked: string}> {
|
||||
const row = await db.prepare("get", 'SELECT "userName", "locked" FROM "userNames" WHERE "userID" = ?', [userID]);
|
||||
if (!row) {
|
||||
return null;
|
||||
throw new Error("No username found");
|
||||
}
|
||||
return row;
|
||||
}
|
||||
|
@ -98,9 +88,6 @@ describe("setUsername", () => {
|
|||
await addUsername(getHash(user05PrivateUserID), username05, 0);
|
||||
await addUsername(getHash(user06PrivateUserID), username06, 0);
|
||||
await addUsername(getHash(user07PrivateUserID), username07, 1);
|
||||
await addUsername(getHash(user10PrivateUserID), username10, 0);
|
||||
// user11 skipped
|
||||
await addUsername(getHash(user12PrivateUserID), username12, 0);
|
||||
});
|
||||
|
||||
it("Should be able to set username that has never been set", (done) => {
|
||||
|
@ -249,47 +236,10 @@ describe("setUsername", () => {
|
|||
it("Should delete row if new username is same as publicID", (done) => {
|
||||
const publicID = getHash(user08PrivateUserID);
|
||||
postSetUserName(getHash(user08PrivateUserID), publicID)
|
||||
.then(async () => {
|
||||
const usernameInfo = await getUsernameInfo(getHash(user08PrivateUserID));
|
||||
assert.strictEqual(usernameInfo, null);
|
||||
done();
|
||||
})
|
||||
.catch((err) => done(err));
|
||||
});
|
||||
|
||||
it("Should return error if trying to set username to privateID", (done) => {
|
||||
const privateID = user09PrivateUserID;
|
||||
postSetUserName(privateID, privateID)
|
||||
.then(async (res) => {
|
||||
assert.strictEqual(res.status, 400);
|
||||
const usernameInfo = await getUsernameInfo(getHash(privateID));
|
||||
assert.strictEqual(usernameInfo, null);
|
||||
done();
|
||||
})
|
||||
.catch((err) => done(err));
|
||||
});
|
||||
|
||||
it("Should return error if trying to set username to someone else's privateID", (done) => {
|
||||
const privateID = user11PrivateUserID;
|
||||
postSetUserName(privateID, user10PrivateUserID)
|
||||
.then(async (res) => {
|
||||
assert.strictEqual(res.status, 400);
|
||||
const usernameInfo = await getUsernameInfo(getHash(privateID)); // user 10's privateID
|
||||
assert.strictEqual(usernameInfo, null);
|
||||
done();
|
||||
})
|
||||
.catch((err) => done(err));
|
||||
});
|
||||
|
||||
it("Should not return error if trying to set username to someone else's publicID", (done) => {
|
||||
const privateID = user12PrivateUserID;
|
||||
const user10PublicID = getHash(user10PrivateUserID);
|
||||
postSetUserName(privateID, user10PublicID)
|
||||
.then(async (res) => {
|
||||
assert.strictEqual(res.status, 200);
|
||||
const usernameInfo = await getUsernameInfo(getHash(privateID)); // user 10's publicID
|
||||
assert.strictEqual(usernameInfo.userName, user10PublicID);
|
||||
done();
|
||||
.then(() => {
|
||||
getUsernameInfo(getHash(user08PrivateUserID))
|
||||
.then(usernameinfo => done(`Username should be deleted - ${usernameinfo})`))
|
||||
.catch(() => done());
|
||||
})
|
||||
.catch((err) => done(err));
|
||||
});
|
||||
|
|
117
test/cases/setUsernamePrivate.ts
Normal file
117
test/cases/setUsernamePrivate.ts
Normal file
|
@ -0,0 +1,117 @@
|
|||
import { db } from "../../src/databases/databases";
|
||||
import { getHash } from "../../src/utils/getHash";
|
||||
import assert from "assert";
|
||||
import { client } from "../utils/httpClient";
|
||||
import { config } from "../../src/config";
|
||||
import sinon from "sinon";
|
||||
|
||||
const USERID_LIMIT = 30;
|
||||
|
||||
// preexisting username with userid < Limit
|
||||
const preExisting_underLimit = "preExisting_under";
|
||||
// preexisting username with userid > Limit
|
||||
const preExisting_overLimit = `preExisting_over${"*".repeat(USERID_LIMIT)}`;
|
||||
// new username to privateID < Limit
|
||||
const newUser_underLimit = "newUser_under";
|
||||
// new username to privateID > Limit
|
||||
const newUser_overLimit = `newUser_over${"*".repeat(USERID_LIMIT)}`;
|
||||
// new username to someone else'e privateID
|
||||
const otherUser = `otherUser${"*".repeat(USERID_LIMIT)}`;
|
||||
|
||||
|
||||
const addUsername = async (userID: string, userName: string, locked = 0) =>
|
||||
await db.prepare("run", 'INSERT INTO "userNames" ("userID", "userName", "locked") VALUES(?, ?, ?)', [userID, userName, locked]);
|
||||
|
||||
async function hasSetUsername(userID: string): Promise<boolean> {
|
||||
const row = await db.prepare("get", 'SELECT "userName", "locked" FROM "userNames" WHERE "userID" = ?', [userID]);
|
||||
return Boolean(row);
|
||||
}
|
||||
|
||||
const endpoint = "/api/setUsername";
|
||||
const postSetUserName = (userID: string, username: string) => client({
|
||||
method: "POST",
|
||||
url: endpoint,
|
||||
params: {
|
||||
userID,
|
||||
username,
|
||||
}
|
||||
});
|
||||
|
||||
describe("setUsernamePrivate tests", () => {
|
||||
// add preexisitng usernames
|
||||
before(async () => {
|
||||
await addUsername(getHash(preExisting_underLimit), preExisting_underLimit, 0);
|
||||
await addUsername(getHash(preExisting_overLimit), preExisting_overLimit, 0);
|
||||
});
|
||||
// stub minUserIDLength
|
||||
before(() => sinon.stub(config, "minUserIDLength").value(USERID_LIMIT));
|
||||
after(() => sinon.restore());
|
||||
|
||||
it("Existing privateID = username under Limit should retreive successfully", (done) => {
|
||||
const privateID = preExisting_underLimit;
|
||||
hasSetUsername(getHash(privateID))
|
||||
.then((usernameInfo) => {
|
||||
assert.ok(usernameInfo);
|
||||
done();
|
||||
});
|
||||
});
|
||||
|
||||
it("Existing privateID = username over Limit should retreive successfully", (done) => {
|
||||
const privateID = preExisting_overLimit;
|
||||
hasSetUsername(getHash(privateID))
|
||||
.then((usernameInfo) => {
|
||||
assert.ok(usernameInfo);
|
||||
done();
|
||||
});
|
||||
});
|
||||
|
||||
it("Should return error if trying to set userID = username under limit", (done) => {
|
||||
const privateID = newUser_underLimit;
|
||||
postSetUserName(privateID, privateID)
|
||||
.then(async (res) => {
|
||||
assert.strictEqual(res.status, 400);
|
||||
const usernameInfo = await hasSetUsername(getHash(privateID));
|
||||
assert.ok(!usernameInfo);
|
||||
done();
|
||||
})
|
||||
.catch((err) => done(err));
|
||||
});
|
||||
|
||||
it("Should return error if trying to set username = other privateID over limit", (done) => {
|
||||
const privateID = newUser_overLimit;
|
||||
postSetUserName(privateID, privateID)
|
||||
.then(async (res) => {
|
||||
assert.strictEqual(res.status, 400);
|
||||
const usernameInfo = await hasSetUsername(getHash(privateID));
|
||||
assert.ok(!usernameInfo);
|
||||
done();
|
||||
})
|
||||
.catch((err) => done(err));
|
||||
});
|
||||
|
||||
it("Should return error if trying to set username = other privateID over limit", (done) => {
|
||||
const privateID = otherUser;
|
||||
const otherUserPrivate = preExisting_overLimit;
|
||||
postSetUserName(privateID, otherUserPrivate)
|
||||
.then(async (res) => {
|
||||
assert.strictEqual(res.status, 400);
|
||||
const usernameInfo = await hasSetUsername(getHash(privateID));
|
||||
assert.ok(!usernameInfo);
|
||||
done();
|
||||
})
|
||||
.catch((err) => done(err));
|
||||
});
|
||||
|
||||
it("Should not return error if trying to set username = other privateID under limit", (done) => {
|
||||
const privateID = otherUser;
|
||||
const otherUserPrivate = preExisting_underLimit;
|
||||
postSetUserName(privateID, otherUserPrivate)
|
||||
.then(async (res) => {
|
||||
assert.strictEqual(res.status, 200);
|
||||
const usernameInfo = await hasSetUsername(getHash(privateID));
|
||||
assert.ok(usernameInfo);
|
||||
done();
|
||||
})
|
||||
.catch((err) => done(err));
|
||||
});
|
||||
});
|
Loading…
Reference in a new issue