library.DefaultDoipEntityUdpMessageHandler.kt Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of doip-sim-ecu-dsl Show documentation
This is a kotlin based domain specific language (dsl), to quickly and intuitively write custom DoIP ECU simulations.
There is a newer version: 0.15.1
package library

import io.ktor.network.sockets.*
import io.ktor.utils.io.core.*
import kotlinx.coroutines.channels.SendChannel
import org.slf4j.Logger
import org.slf4j.LoggerFactory
import kotlin.math.max

open class DefaultDoipEntityUdpMessageHandler(
    val doipEntity: DoipEntity,
    val config: DoipEntityConfig
) : DoipUdpMessageHandler {
    private val logger: Logger = LoggerFactory.getLogger(DefaultDoipEntityUdpMessageHandler::class.java)

    companion object {
        fun generateVamByEntityConfig(doipEntity: DoipEntity): List =
            with(doipEntity.config) {
                listOf(DoipUdpVehicleAnnouncementMessage(vin, logicalAddress, gid, eid)) +
                        doipEntity.ecus.filter { it.config.additionalVam != null }.map { it.config.additionalVam!!.toVam(it.config, doipEntity.config) }
            }
    }

    suspend fun sendVamResponse(
        sendChannel: SendChannel,
        sourceAddress: SocketAddress,
    ) {
        val vams = generateVamByEntityConfig(doipEntity)
        vams.forEach { vam ->
            logger.info("Sending VAM for ${vam.logicalAddress.toString(16)}")
            sendChannel.send(
                Datagram(
                    packet = ByteReadPacket(vam.asByteArray),
                    address = sourceAddress
                )
            )
        }
    }

    override suspend fun handleUdpVehicleInformationRequest(
        sendChannel: SendChannel,
        sourceAddress: SocketAddress,
        message: DoipUdpVehicleInformationRequest
    ) {
        sendVamResponse(sendChannel, sourceAddress)
    }

    override suspend fun handleUdpVehicleInformationRequestWithEid(
        sendChannel: SendChannel,
        sourceAddress: SocketAddress,
        message: DoipUdpVehicleInformationRequestWithEid
    ) {
        if (config.eid.contentEquals(message.eid)) {
            logger.info("Received VIR for VIN, responding with VAMs")
            sendVamResponse(sendChannel, sourceAddress)
        } else {
            logger.info("Received VIR for different VIN, not responding")
        }
    }

    override suspend fun handleUdpVehicleInformationRequestWithVIN(
        sendChannel: SendChannel,
        sourceAddress: SocketAddress,
        message: DoipUdpVehicleInformationRequestWithVIN
    ) {

        if (config.vin.contentEquals(message.vin)) {
            sendVamResponse(sendChannel, sourceAddress)
        }
    }

    override suspend fun handleUdpEntityStatusRequest(
        sendChannel: SendChannel,
        sourceAddress: SocketAddress,
        message: DoipUdpEntityStatusRequest
    ) {
        sendChannel.send(
            Datagram(
                packet = ByteReadPacket(
                    DoipUdpEntityStatusResponse(
                        config.nodeType.value,
                        255.toByte(),
                        max(doipEntity.connectionHandlers.size, 255).toByte(),
                        config.maxDataSize
                    )
                        .asByteArray
                ),
                address = sourceAddress
            )
        )
    }

    override suspend fun handleUdpDiagnosticPowerModeRequest(
        sendChannel: SendChannel,
        sourceAddress: SocketAddress,
        message: DoipUdpDiagnosticPowerModeRequest
    ) {
        sendChannel.send(
            Datagram(
                packet = ByteReadPacket(
                    DoipUdpDiagnosticPowerModeResponse(0)
                        .asByteArray
                ),
                address = sourceAddress
            )
        )
    }
}