add allowedEncryptedRooms and allowedNonPrivateRooms support

Author: ricardogarim

packages/federation-sdk/src/index.ts

@@ -90,6 +90,7 @@ export {
 	type BaseEventType,
 } from './utils/event-schemas';
 export { errCodes } from './utils/response-codes';
+export { NotAllowedError } from './services/invite.service';
 
 export { EventRepository } from './repositories/event.repository';
 export { RoomRepository } from './repositories/room.repository';

packages/federation-sdk/src/services/config.service.ts

@@ -31,6 +31,10 @@ export interface AppConfig {
 			downloadPerMinute: number;
 		};
 	};
+	invite: {
+		allowedEncryptedRooms: boolean;
+		allowedNonPrivateRooms: boolean;
+	};
 }
 
 export const AppConfigSchema = z.object({
@@ -69,6 +73,10 @@ export const AppConfigSchema = z.object({
 				.min(1, 'Download rate limit must be at least 1'),
 		}),
 	}),
+	invite: z.object({
+		allowedEncryptedRooms: z.boolean(),
+		allowedNonPrivateRooms: z.boolean(),
+	}),
 });
 
 export class ConfigService {
@@ -113,6 +121,10 @@ export class ConfigService {
 		return this.config.media;
 	}
 
+	getInviteConfig(): AppConfig['invite'] {
+		return this.config.invite;
+	}
+
 	async getSigningKey() {
 		// If config contains a signing key, use it
 		if (!this.config.signingKey) {

packages/federation-sdk/src/services/invite.service.ts

@@ -25,6 +25,13 @@ export type ProcessInviteEvent = {
 	room_version: string;
 };
 
+export class NotAllowedError extends Error {
+	constructor(message: string) {
+		super(message);
+		this.name = 'NotAllowedError';
+	}
+}
+
 @singleton()
 export class InviteService {
 	private readonly logger = createLogger('InviteService');
@@ -142,6 +149,42 @@ export class InviteService {
 		};
 	}
 
+	private async shouldProcessInvite(
+		event: PduForType<'m.room.member'>,
+	): Promise<void> {
+		const isRoomNonPrivate = event.unsigned.invite_room_state.some(
+			(
+				stateEvent: PersistentEventBase<
+					RoomVersion,
+					'm.room.join_rules'
+				>['event'],
+			) =>
+				stateEvent.type === 'm.room.join_rules' &&
+				stateEvent.content.join_rule === 'public',
+		);
+
+		const isRoomEncrypted = event.unsigned.invite_room_state.some(
+			(
+				stateEvent: PersistentEventBase<
+					RoomVersion,
+					'm.room.encryption'
+				>['event'],
+			) => stateEvent.type === 'm.room.encryption',
+		);
+
+		const { allowedEncryptedRooms, allowedNonPrivateRooms } =
+			this.configService.getInviteConfig();
+
+		const shouldRejectInvite =
+			(!allowedEncryptedRooms && isRoomEncrypted) ||
+			(!allowedNonPrivateRooms && isRoomNonPrivate);
+		if (shouldRejectInvite) {
+			throw new NotAllowedError(
+				`Could not process invite due to room being ${isRoomEncrypted ? 'encrypted' : 'public'}`,
+			);
+		}
+	}
+
 	async processInvite(
 		event: PduForType<'m.room.member'>,
 		roomId: RoomID,
@@ -150,6 +193,7 @@ export class InviteService {
 		authenticatedServer: string,
 	) {
 		// SPEC: when a user invites another user on a different homeserver, a request to that homeserver to have the event signed and verified must be made
+		await this.shouldProcessInvite(event);
 
 		const residentServer = roomId.split(':').pop();
 		if (!residentServer) {

packages/federation-sdk/src/services/room.service.ts

@@ -845,7 +845,7 @@ export class RoomService {
 
 		// trying to join room from another server
 		const makeJoinResponse = await federationService.makeJoin(
-			residentServer as string,
+			residentServer,
 			roomId,
 			userId,
 			roomVersion, // NOTE: check the comment in the called method

packages/homeserver/src/controllers/federation/invite.controller.ts

@@ -2,30 +2,55 @@ import { EventID, RoomID } from '@rocket.chat/federation-room';
 import {
 	EventAuthorizationService,
 	InviteService,
+	NotAllowedError,
 } from '@rocket.chat/federation-sdk';
 import { isAuthenticatedMiddleware } from '@rocket.chat/homeserver/middlewares/isAuthenticated';
 import { Elysia, t } from 'elysia';
 import { container } from 'tsyringe';
-import { ProcessInviteParamsDto, RoomVersionDto } from '../../dtos';
+import {
+	FederationErrorResponseDto,
+	ProcessInviteParamsDto,
+	ProcessInviteResponseDto,
+	RoomVersionDto,
+} from '../../dtos';
 
 export const invitePlugin = (app: Elysia) => {
 	const inviteService = container.resolve(InviteService);
 	const eventAuthService = container.resolve(EventAuthorizationService);
 
 	return app.use(isAuthenticatedMiddleware(eventAuthService)).put(
 		'/_matrix/federation/v2/invite/:roomId/:eventId',
-		async ({ body, params: { roomId, eventId }, authenticatedServer }) => {
+		async ({ body, set, params: { roomId, eventId }, authenticatedServer }) => {
 			if (!authenticatedServer) {
 				throw new Error('Missing authenticated server from request');
 			}
 
-			return inviteService.processInvite(
-				body.event,
-				roomId as RoomID,
-				eventId as EventID,
-				body.room_version,
-				authenticatedServer,
-			);
+			try {
+				return await inviteService.processInvite(
+					body.event,
+					roomId as RoomID,
+					eventId as EventID,
+					body.room_version,
+					authenticatedServer,
+				);
+			} catch (error) {
+				if (error instanceof NotAllowedError) {
+					set.status = 403;
+					return {
+						errcode: 'M_FORBIDDEN',
+						error: 'This server does not allow joining this type of room based on federation settings.',
+					};
+				}
+
+				set.status = 500;
+				return {
+					errcode: 'M_UNKNOWN',
+					error:
+						error instanceof Error
+							? error.message
+							: 'Internal server error while processing request',
+				};
+			}
 		},
 		{
 			params: ProcessInviteParamsDto,
@@ -34,6 +59,11 @@ export const invitePlugin = (app: Elysia) => {
 				room_version: RoomVersionDto,
 				invite_room_state: t.Any(),
 			}),
+			response: {
+				200: ProcessInviteResponseDto,
+				403: FederationErrorResponseDto,
+				500: FederationErrorResponseDto,
+			},
 			detail: {
 				tags: ['Federation'],
 				summary: 'Process room invite',

packages/homeserver/src/dtos/federation/error.dto.ts

@@ -0,0 +1,15 @@
+import { type Static, t } from 'elysia';
+
+export const FederationErrorResponseDto = t.Object({
+	errcode: t.Enum({
+		M_UNRECOGNIZED: 'M_UNRECOGNIZED',
+		M_UNAUTHORIZED: 'M_UNAUTHORIZED',
+		M_FORBIDDEN: 'M_FORBIDDEN',
+		M_UNKNOWN: 'M_UNKNOWN',
+	}),
+	error: t.String(),
+});
+
+export type FederationErrorResponseDto = Static<
+	typeof FederationErrorResponseDto
+>;

packages/homeserver/src/dtos/index.ts

@@ -12,6 +12,7 @@ export * from './federation/state-ids.dto';
 export * from './federation/state.dto';
 export * from './federation/transactions.dto';
 export * from './federation/versions.dto';
+export * from './federation/error.dto';
 
 // Internal DTOs
 export * from './internal/invite.dto';

packages/homeserver/src/homeserver.module.ts

@@ -88,6 +88,12 @@ export async function setup(options?: HomeserverSetupOptions) {
 				),
 			},
 		},
+		invite: {
+			allowedEncryptedRooms:
+				process.env.INVITE_ALLOWED_ENCRYPTED_ROOMS === 'true',
+			allowedNonPrivateRooms:
+				process.env.INVITE_ALLOWED_NON_PRIVATE_ROOMS === 'true',
+		},
 	});
 
 	const containerOptions: FederationContainerOptions = {

packages/room/src/types/v3-11.ts

@@ -714,7 +714,7 @@ const EventPduTypeRoomEncrypted = z.object({
 });
 
 const EventPduTypeRoomEncryption = z.object({
-	...PduNoContentTimelineEventSchema,
+	...PduNoContentEmptyStateKeyStateEventSchema,
 	type: z.literal('m.room.encryption'),
 	content: PduEncryptionEventContentSchema,
 });