/api/mek/register, /api/mek/share Endpoint 삭제 및 MEK 서명 매커니즘 구현

2025년 첫 커밋! Happy New Year~
This commit is contained in:
static
2025-01-01 05:24:13 +09:00
parent e8e4022bc2
commit 363f809d02
12 changed files with 112 additions and 259 deletions

View File

@@ -1,8 +1,13 @@
import { and, or, eq, lt, desc } from "drizzle-orm";
import { and, or, eq } from "drizzle-orm";
import db from "./drizzle";
import { mek, clientMek, userClient } from "./schema";
import { mek, clientMek } from "./schema";
export const registerInitialMek = async (userId: number, createdBy: number, encMek: string) => {
export const registerInitialMek = async (
userId: number,
createdBy: number,
encMek: string,
encMekSig: string,
) => {
await db.transaction(async (tx) => {
await tx.insert(mek).values({
userId,
@@ -16,70 +21,11 @@ export const registerInitialMek = async (userId: number, createdBy: number, encM
clientId: createdBy,
mekVersion: 1,
encMek,
encMekSig,
});
});
};
export const registerActiveMek = async (
userId: number,
version: number,
createdBy: number,
clientMeks: {
clientId: number;
encMek: string;
}[],
) => {
await db.transaction(async (tx) => {
// 1. Check if the clientMeks are valid
const userClients = await tx
.select()
.from(userClient)
.where(and(eq(userClient.userId, userId), eq(userClient.state, "active")));
if (
clientMeks.length !== userClients.length ||
!clientMeks.every((clientMek) =>
userClients.some((userClient) => userClient.clientId === clientMek.clientId),
)
) {
throw new Error("Invalid key list");
}
// 2. Retire the old active MEK and insert the new one
await tx
.update(mek)
.set({
state: "retired",
retiredAt: new Date(),
})
.where(and(eq(mek.userId, userId), lt(mek.version, version), eq(mek.state, "active")));
await tx.insert(mek).values({
userId,
version,
createdBy,
createdAt: new Date(),
state: "active",
});
// 3. Insert the new client MEKs
await tx.insert(clientMek).values(
clientMeks.map(({ clientId, encMek }) => ({
userId,
clientId,
mekVersion: version,
encMek,
})),
);
});
};
export const getAllValidMeks = async (userId: number) => {
return await db
.select()
.from(mek)
.where(and(eq(mek.userId, userId), or(eq(mek.state, "active"), eq(mek.state, "retired"))))
.execute();
};
export const getInitialMek = async (userId: number) => {
const meks = await db
.select()
@@ -89,68 +35,6 @@ export const getInitialMek = async (userId: number) => {
return meks[0] ?? null;
};
export const getNextActiveMekVersion = async (userId: number) => {
const meks = await db
.select({ version: mek.version })
.from(mek)
.where(eq(mek.userId, userId))
.orderBy(desc(mek.version))
.limit(1)
.execute();
if (!meks[0]) {
throw new Error("No MEK found");
}
return meks[0].version + 1;
};
export const registerClientMeks = async (
userId: number,
clientId: number,
clientMeks: {
version: number;
encMek: string;
}[],
) => {
await db.transaction(async (tx) => {
// 1. Check if the client is valid
const userClients = await tx
.select()
.from(userClient)
.where(
and(
eq(userClient.userId, userId),
eq(userClient.clientId, clientId),
eq(userClient.state, "active"),
),
);
if (userClients.length === 0) {
throw new Error("Invalid client");
}
// 2. Check if the clientMeks are valid
const meks = await tx
.select()
.from(mek)
.where(and(eq(mek.userId, userId), or(eq(mek.state, "active"), eq(mek.state, "retired"))));
if (
clientMeks.length !== meks.length ||
!clientMeks.every((clientMek) => meks.some((mek) => mek.version === clientMek.version))
) {
throw new Error("Invalid key list");
}
// 3. Insert the client MEKs
await tx.insert(clientMek).values(
clientMeks.map(({ version, encMek }) => ({
userId,
clientId,
mekVersion: version,
encMek,
})),
);
});
};
export const getAllValidClientMeks = async (userId: number, clientId: number) => {
return await db
.select()

View File

@@ -30,8 +30,9 @@ export const clientMek = sqliteTable(
clientId: integer("client_id")
.notNull()
.references(() => client.id),
mekVersion: integer("master_encryption_key_version").notNull(),
encMek: text("encrypted_master_encryption_key").notNull(),
mekVersion: integer("version").notNull(),
encMek: text("encrypted_key").notNull(), // Base64
encMekSig: text("encrypted_key_signature").notNull(), // Base64
},
(t) => ({
pk: primaryKey({ columns: [t.userId, t.clientId, t.mekVersion] }),