Browse Source

Fix: Cidr Change (#1712)

* only calculate ip if cidr changed

if the cidr did not change, the ip will not change to prevent ip shifts

* fix lint
pull/1713/head
Bernd Storath 1 month ago
committed by GitHub
parent
commit
842475f799
No known key found for this signature in database GPG Key ID: B5690EEEBB952194
  1. 3
      src/i18n/locales/en.json
  2. 28
      src/server/database/repositories/interface/service.ts
  3. 2
      src/server/database/repositories/interface/types.ts

3
src/i18n/locales/en.json

@ -193,7 +193,8 @@
}, },
"interface": { "interface": {
"cidr": "CIDR", "cidr": "CIDR",
"device": "Device" "device": "Device",
"cidrValid": "CIDR must be valid"
}, },
"otl": "One Time link", "otl": "One Time link",
"stringMalformed": "String is malformed", "stringMalformed": "String is malformed",

28
src/server/database/repositories/interface/service.ts

@ -1,4 +1,3 @@
import isCidr from 'is-cidr';
import { eq, sql } from 'drizzle-orm'; import { eq, sql } from 'drizzle-orm';
import { parseCidr } from 'cidr-tools'; import { parseCidr } from 'cidr-tools';
import { wgInterface } from './schema'; import { wgInterface } from './schema';
@ -58,10 +57,18 @@ export class InterfaceService {
} }
updateCidr(data: InterfaceCidrUpdateType) { updateCidr(data: InterfaceCidrUpdateType) {
if (!isCidr(data.ipv4Cidr) || !isCidr(data.ipv6Cidr)) {
throw new Error('Invalid CIDR');
}
return this.#db.transaction(async (tx) => { return this.#db.transaction(async (tx) => {
const oldCidr = await tx.query.wgInterface
.findFirst({
where: eq(wgInterface.name, 'wg0'),
columns: { ipv4Cidr: true, ipv6Cidr: true },
})
.execute();
if (!oldCidr) {
throw new Error('Interface not found');
}
await tx await tx
.update(wgInterface) .update(wgInterface)
.set(data) .set(data)
@ -74,8 +81,17 @@ export class InterfaceService {
// TODO: optimize // TODO: optimize
const clients = await tx.query.client.findMany().execute(); const clients = await tx.query.client.findMany().execute();
const nextIpv4 = nextIP(4, parseCidr(data.ipv4Cidr), clients); // only calculate ip if cidr has changed
const nextIpv6 = nextIP(6, parseCidr(data.ipv6Cidr), clients);
let nextIpv4 = client.ipv4Address;
if (data.ipv4Cidr !== oldCidr.ipv4Cidr) {
nextIpv4 = nextIP(4, parseCidr(data.ipv4Cidr), clients);
}
let nextIpv6 = client.ipv6Address;
if (data.ipv6Cidr !== oldCidr.ipv6Cidr) {
nextIpv6 = nextIP(6, parseCidr(data.ipv6Cidr), clients);
}
await tx await tx
.update(clientSchema) .update(clientSchema)

2
src/server/database/repositories/interface/types.ts

@ -1,5 +1,6 @@
import type { InferSelectModel } from 'drizzle-orm'; import type { InferSelectModel } from 'drizzle-orm';
import z from 'zod'; import z from 'zod';
import isCidr from 'is-cidr';
import type { wgInterface } from './schema'; import type { wgInterface } from './schema';
export type InterfaceType = InferSelectModel<typeof wgInterface>; export type InterfaceType = InferSelectModel<typeof wgInterface>;
@ -22,6 +23,7 @@ const device = z
const cidr = z const cidr = z
.string({ message: t('zod.interface.cidr') }) .string({ message: t('zod.interface.cidr') })
.min(1, { message: t('zod.interface.cidr') }) .min(1, { message: t('zod.interface.cidr') })
.refine((value) => isCidr(value), { message: t('zod.interface.cidrValid') })
.pipe(safeStringRefine); .pipe(safeStringRefine);
export const InterfaceUpdateSchema = schemaForType<InterfaceUpdateType>()( export const InterfaceUpdateSchema = schemaForType<InterfaceUpdateType>()(

Loading…
Cancel
Save