Skip to content

Commit

Permalink
Allow candidate exchange via REST+SSE
Browse files Browse the repository at this point in the history
Closes #19
  • Loading branch information
Brutus5000 committed Jan 19, 2024
1 parent 73d1bc5 commit 3195c2b
Show file tree
Hide file tree
Showing 5 changed files with 125 additions and 2 deletions.
Original file line number Diff line number Diff line change
@@ -0,0 +1,17 @@
package com.faforever.icebreaker.security

import io.quarkus.oidc.runtime.OidcJwtCallerPrincipal
import io.quarkus.security.identity.SecurityIdentity
import jakarta.enterprise.context.ApplicationScoped

@ApplicationScoped
class CurrentUserService(
private val securityIdentity: SecurityIdentity,
) {
fun getCurrentUserId(): Long? {
val principal = (securityIdentity.principal as? OidcJwtCallerPrincipal)
val subject = principal?.claims?.claimsMap?.get("sub") as? String

return subject?.toLong()
}
}
Original file line number Diff line number Diff line change
@@ -0,0 +1,59 @@
package com.faforever.icebreaker.service

import com.fasterxml.jackson.annotation.JsonInclude
import com.fasterxml.jackson.annotation.JsonSubTypes
import com.fasterxml.jackson.annotation.JsonTypeInfo
import com.fasterxml.jackson.annotation.JsonValue

@JsonTypeInfo(use = JsonTypeInfo.Id.NAME, include = JsonTypeInfo.As.PROPERTY, property = "eventType")
@JsonSubTypes(
JsonSubTypes.Type(value = CandidatesMessage::class, name = "candidates"),
JsonSubTypes.Type(value = ConnectedMessage::class, name = "connected"),
)
@JsonInclude(JsonInclude.Include.ALWAYS)
interface EventMessage {
val gameId: Long
val senderId: Long
val recipientId: Long?
}

@JvmRecord
data class ConnectedMessage(
override val gameId: Long,
override val senderId: Long,
override val recipientId: Long? = null,
) : EventMessage

@JvmRecord
data class CandidatesMessage(
override val gameId: Long,
override val senderId: Long,
override val recipientId: Long,
val candidates: List<CandidateDescriptor>,
) : EventMessage {

enum class CandidateType(@JsonValue val jsonValue: String) {
PEER_REFLEXIVE_CANDIDATE("prflx"),
SERVER_REFLEXIVE_CANDIDATE("srflx"),
RELAYED_CANDIDATE("relay"),
HOST_CANDIDATE("host"),
LOCAL_CANDIDATE("local"),
STUN_CANDIDATE("stun"),
}

@JvmRecord
data class CandidateDescriptor(
val foundation: String,
val protocol: String,
val priority: Long,
val ip: String?,
val port: Int,
val type: CandidateType,
val generation: Int,
val id: String,
val relAddr: String?,
val relPort: Int,
) : Comparable<CandidateDescriptor> {
override operator fun compareTo(other: CandidateDescriptor) = (other.priority - priority).toInt()
}
}
29 changes: 29 additions & 0 deletions src/main/kotlin/com/faforever/icebreaker/service/SessionService.kt
Original file line number Diff line number Diff line change
Expand Up @@ -3,12 +3,15 @@ package com.faforever.icebreaker.service
import com.faforever.icebreaker.config.FafProperties
import com.faforever.icebreaker.persistence.IceSessionEntity
import com.faforever.icebreaker.persistence.IceSessionRepository
import com.faforever.icebreaker.security.CurrentUserService
import com.faforever.icebreaker.util.AsyncRunner
import io.quarkus.scheduler.Scheduled
import io.quarkus.security.ForbiddenException
import io.quarkus.security.UnauthorizedException
import io.quarkus.security.identity.SecurityIdentity
import io.smallrye.jwt.build.Jwt
import io.smallrye.mutiny.Multi
import io.smallrye.mutiny.helpers.MultiEmitterProcessor
import jakarta.enterprise.inject.Instance
import jakarta.inject.Singleton
import jakarta.transaction.Transactional
Expand All @@ -26,8 +29,11 @@ class SessionService(
private val fafProperties: FafProperties,
private val iceSessionRepository: IceSessionRepository,
private val securityIdentity: SecurityIdentity,
private val currentUserService: CurrentUserService,
) {
private val activeSessionHandlers = sessionHandlers.filter { it.active }
private val eventEmitter = MultiEmitterProcessor.create<EventMessage>()
private val eventBroadcast: Multi<EventMessage> = eventEmitter.toMulti().broadcast().toAllSubscribers()

fun buildToken(gameId: Long): String {
val userId = when (val principal = securityIdentity?.principal) {
Expand Down Expand Up @@ -111,4 +117,27 @@ class SessionService(
iceSessionRepository.delete(iceSession)
}
}

fun listenForEventMessages(gameId: Long): Multi<EventMessage> {
val userId = currentUserService.getCurrentUserId()
eventEmitter.emit(ConnectedMessage(gameId = gameId, senderId = currentUserService.getCurrentUserId()!!))

return eventBroadcast.filter {
it.gameId == gameId && (it.recipientId == userId || (it.recipientId == null && it.senderId != userId))
}
}

fun onCandidatesReceived(gameId: Long, candidatesMessage: CandidatesMessage) {
// Check messages for manipulation. We need to prevent cross-channel vulnerabilities.
check(candidatesMessage.gameId == gameId) {
"gameId $gameId from endpoint does not match gameId ${candidatesMessage.gameId} in candidateMessage"
}

val currentUserId = currentUserService.getCurrentUserId()
check(candidatesMessage.senderId == currentUserService.getCurrentUserId()) {
"current user id $currentUserId from endpoint does not match sourceId ${candidatesMessage.senderId} in candidateMessage"
}

eventEmitter.emit(candidatesMessage)
}
}
18 changes: 18 additions & 0 deletions src/main/kotlin/com/faforever/icebreaker/web/SessionController.kt
Original file line number Diff line number Diff line change
@@ -1,9 +1,12 @@
package com.faforever.icebreaker.web

import com.faforever.icebreaker.service.CandidatesMessage
import com.faforever.icebreaker.service.EventMessage
import com.faforever.icebreaker.service.Session
import com.faforever.icebreaker.service.SessionService
import io.quarkus.runtime.annotations.RegisterForReflection
import io.quarkus.security.PermissionsAllowed
import io.smallrye.mutiny.Multi
import jakarta.inject.Singleton
import jakarta.ws.rs.Consumes
import jakarta.ws.rs.GET
Expand All @@ -12,6 +15,7 @@ import jakarta.ws.rs.Path
import jakarta.ws.rs.Produces
import jakarta.ws.rs.core.MediaType
import org.jboss.resteasy.reactive.RestPath
import org.jboss.resteasy.reactive.RestStreamElementType

@Path("/session")
@Singleton
Expand Down Expand Up @@ -54,4 +58,18 @@ class SessionController(
),
)
}

@POST
@Path("/game/{gameId}/events")
@PermissionsAllowed("USER:lobby")
@Consumes(MediaType.APPLICATION_JSON)
fun postEvent(@RestPath gameId: Long, candidatesMessage: CandidatesMessage) {
sessionService.onCandidatesReceived(gameId, candidatesMessage)
}

@GET
@Path("/game/{gameId}/events")
@PermissionsAllowed("USER:lobby")
@RestStreamElementType(MediaType.APPLICATION_JSON)
fun getSessionEvents(@RestPath gameId: Long): Multi<EventMessage> = sessionService.listenForEventMessages(gameId)
}
4 changes: 2 additions & 2 deletions src/main/resources/application.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -100,11 +100,11 @@ smallrye:
tQIDAQAB
-----END PUBLIC KEY-----
log:
level: DEBUG
level: INFO
category:
"org.hibernate.SQL":
level: DEBUG
"com.faforever":
level: DEBUG
"io.quarkus":
level: DEBUG
level: INFO

0 comments on commit 3195c2b

Please sign in to comment.