Updated gerbil key handeling

This commit is contained in:
Owen Schwartz 2024-10-26 17:02:11 -04:00
parent 6cee5703b5
commit 7feb21e727
No known key found for this signature in database
GPG key ID: 8271FDFFD9E0CCBD
5 changed files with 25 additions and 17 deletions

View file

@ -20,9 +20,10 @@ export async function ensureActions() {
// Add new actions // Add new actions
for (const actionId of actionsToAdd) { for (const actionId of actionsToAdd) {
logger.debug(`Adding action: ${actionId}`);
await db.insert(actions).values({ actionId }).execute(); await db.insert(actions).values({ actionId }).execute();
// Add new actions to the Default role // Add new actions to the Default role
if (defaultRoles.length === 0) { if (defaultRoles.length != 0) {
await db.insert(roleActions) await db.insert(roleActions)
.values(defaultRoles.map(role => ({ roleId: role.roleId!, actionId, orgId: role.orgId! }))) .values(defaultRoles.map(role => ({ roleId: role.roleId!, actionId, orgId: role.orgId! })))
.execute(); .execute();
@ -31,6 +32,7 @@ export async function ensureActions() {
// Remove deprecated actions // Remove deprecated actions
if (actionsToRemove.length > 0) { if (actionsToRemove.length > 0) {
logger.debug(`Removing actions: ${actionsToRemove.join(', ')}`);
await db.delete(actions).where(inArray(actions.actionId, actionsToRemove)).execute(); await db.delete(actions).where(inArray(actions.actionId, actionsToRemove)).execute();
await db.delete(roleActions).where(inArray(roleActions.actionId, actionsToRemove)).execute(); await db.delete(roleActions).where(inArray(roleActions.actionId, actionsToRemove)).execute();
} }

View file

@ -6,7 +6,13 @@ import { eq, and } from "drizzle-orm";
import { __DIRNAME } from "@server/config"; import { __DIRNAME } from "@server/config";
// Load the names from the names.json file // Load the names from the names.json file
const file = join(__DIRNAME, "names.json"); const dev = process.env.ENVIRONMENT !== "prod";
let file;
if (!dev) {
file = join(__DIRNAME, "names.json");
} else {
file = join(__DIRNAME, "/db/names.json");
}
export const names = JSON.parse(readFileSync(file, "utf-8")); export const names = JSON.parse(readFileSync(file, "utf-8"));
export async function getUniqueSiteName(orgId: string): Promise<string> { export async function getUniqueSiteName(orgId: string): Promise<string> {

View file

@ -27,7 +27,7 @@ export type GetConfigResponse = {
export async function getConfig(req: Request, res: Response, next: NextFunction): Promise<any> { export async function getConfig(req: Request, res: Response, next: NextFunction): Promise<any> {
try { try {
// Validate request parameters // Validate request parameters
const parsedParams = getConfigSchema.safeParse(req.query); const parsedParams = getConfigSchema.safeParse(req.body);
if (!parsedParams.success) { if (!parsedParams.success) {
return next( return next(
createHttpError( createHttpError(
@ -44,9 +44,9 @@ export async function getConfig(req: Request, res: Response, next: NextFunction)
} }
// Fetch exit node // Fetch exit node
let exitNode = await db.select().from(exitNodes).where(eq(exitNodes.publicKey, publicKey)); let exitNodeQuery = await db.select().from(exitNodes).where(eq(exitNodes.publicKey, publicKey));
let exitNode;
if (!exitNode) { if (exitNodeQuery.length === 0) {
const address = await getNextAvailableSubnet(); const address = await getNextAvailableSubnet();
const listenPort = await getNextAvailablePort(); const listenPort = await getNextAvailablePort();
const subEndpoint = await getUniqueExitNodeEndpointName(); const subEndpoint = await getUniqueExitNodeEndpointName();
@ -61,6 +61,8 @@ export async function getConfig(req: Request, res: Response, next: NextFunction)
}).returning().execute(); }).returning().execute();
logger.info(`Created new exit node ${exitNode[0].name} with address ${exitNode[0].address} and port ${exitNode[0].listenPort}`); logger.info(`Created new exit node ${exitNode[0].name} with address ${exitNode[0].address} and port ${exitNode[0].listenPort}`);
} else {
exitNode = exitNodeQuery;
} }
if (!exitNode) { if (!exitNode) {
@ -98,16 +100,11 @@ export async function getConfig(req: Request, res: Response, next: NextFunction)
peers, peers,
}; };
return response(res, { logger.debug("Sending config: ", configResponse);
data: configResponse,
success: true,
error: false,
message: "Configuration retrieved successfully",
status: HttpCode.OK,
});
return res.status(HttpCode.OK).send(configResponse);
} catch (error) { } catch (error) {
logger.error('Error from getConfig:', error); logger.error(error);
return next(createHttpError(HttpCode.INTERNAL_SERVER_ERROR, "An error occurred...")); return next(createHttpError(HttpCode.INTERNAL_SERVER_ERROR, "An error occurred..."));
} }
} }
@ -119,10 +116,13 @@ async function getNextAvailableSubnet(): Promise<string> {
}).from(exitNodes); }).from(exitNodes);
const addresses = existingAddresses.map(a => a.address); const addresses = existingAddresses.map(a => a.address);
const subnet = findNextAvailableCidr(addresses, config.gerbil.block_size, config.gerbil.subnet_group); let subnet = findNextAvailableCidr(addresses, config.gerbil.block_size, config.gerbil.subnet_group);
if (!subnet) { if (!subnet) {
throw new Error('No available subnets remaining in space'); throw new Error('No available subnets remaining in space');
} }
// replace the last octet with 1
subnet = subnet.split('.').slice(0, 3).join('.') + '.1';
return subnet; return subnet;
} }

View file

@ -17,7 +17,7 @@ internalRouter.get("/traefik-config", traefik.traefikConfigProvider);
const gerbilRouter = Router(); const gerbilRouter = Router();
internalRouter.use("/gerbil", gerbilRouter); internalRouter.use("/gerbil", gerbilRouter);
gerbilRouter.get("/get-config", gerbil.getConfig); gerbilRouter.post("/get-config", gerbil.getConfig);
gerbilRouter.post("/receive-bandwidth", gerbil.receiveBandwidth); gerbilRouter.post("/receive-bandwidth", gerbil.receiveBandwidth);
// Badger routes // Badger routes

View file

@ -75,7 +75,7 @@ export async function createResource(req: Request, res: Response, next: NextFunc
} }
// Generate a unique resourceId // Generate a unique resourceId
const fullDomain = `${subdomain}.${org[0].orgId}.${org[0].domain}`; const fullDomain = `${subdomain}.${org[0].domain}`;
// Create new resource in the database // Create new resource in the database
const newResource = await db.insert(resources).values({ const newResource = await db.insert(resources).values({