Line | Branch | Exec | Source |
---|---|---|---|
1 | #pragma once | ||
2 | |||
3 | #include "safety_declarations.h" | ||
4 | |||
5 | static bool rivian_longitudinal = false; | ||
6 | |||
7 | 21936 | static void rivian_rx_hook(const CANPacket_t *to_push) { | |
8 | 21936 | int bus = GET_BUS(to_push); | |
9 | 21936 | int addr = GET_ADDR(to_push); | |
10 | |||
11 |
2/2✓ Branch 0 taken 10658 times.
✓ Branch 1 taken 11278 times.
|
21936 | if (bus == 0) { |
12 | // Vehicle speed | ||
13 |
2/2✓ Branch 0 taken 16 times.
✓ Branch 1 taken 10642 times.
|
10658 | if (addr == 0x208) { |
14 | 16 | vehicle_moving = GET_BYTE(to_push, 6) | GET_BYTE(to_push, 7); | |
15 | } | ||
16 | |||
17 | // Driver torque | ||
18 |
2/2✓ Branch 0 taken 4970 times.
✓ Branch 1 taken 5688 times.
|
10658 | if (addr == 0x380) { |
19 | 4970 | int torque_driver_new = (((GET_BYTE(to_push, 2) << 4) | (GET_BYTE(to_push, 3) >> 4))) - 2050U; | |
20 | 4970 | update_sample(&torque_driver, torque_driver_new); | |
21 | } | ||
22 | |||
23 | // Gas pressed | ||
24 |
2/2✓ Branch 0 taken 22 times.
✓ Branch 1 taken 10636 times.
|
10658 | if (addr == 0x150) { |
25 | 22 | gas_pressed = GET_BYTE(to_push, 3) | (GET_BYTE(to_push, 4) & 0xC0U); | |
26 | } | ||
27 | |||
28 | // Brake pressed | ||
29 |
2/2✓ Branch 0 taken 26 times.
✓ Branch 1 taken 10632 times.
|
10658 | if (addr == 0x38f) { |
30 | 26 | brake_pressed = GET_BIT(to_push, 23U); | |
31 | } | ||
32 | |||
33 | 10658 | generic_rx_checks(addr == 0x120); // ACM_lkaHbaCmd | |
34 |
2/2✓ Branch 0 taken 5329 times.
✓ Branch 1 taken 5329 times.
|
10658 | if (rivian_longitudinal) { |
35 | 5329 | generic_rx_checks(addr == 0x160); // ACM_longitudinalRequest | |
36 | } | ||
37 | } | ||
38 | |||
39 |
2/2✓ Branch 0 taken 5646 times.
✓ Branch 1 taken 16290 times.
|
21936 | if (bus == 2) { |
40 | // Cruise state | ||
41 |
2/2✓ Branch 0 taken 16 times.
✓ Branch 1 taken 5630 times.
|
5646 | if (addr == 0x100) { |
42 | 16 | pcm_cruise_check(GET_BIT(to_push, 21U)); | |
43 | } | ||
44 | } | ||
45 | 21936 | } | |
46 | |||
47 | 5498 | static bool rivian_tx_hook(const CANPacket_t *to_send) { | |
48 | const SteeringLimits RIVIAN_STEERING_LIMITS = { | ||
49 | .max_steer = 250, | ||
50 | .max_rate_up = 3, | ||
51 | .max_rate_down = 5, | ||
52 | .max_rt_delta = 125, | ||
53 | .max_rt_interval = 250000, | ||
54 | .driver_torque_multiplier = 2, | ||
55 | .driver_torque_allowance = 100, | ||
56 | .type = TorqueDriverLimited, | ||
57 | }; | ||
58 | |||
59 | const LongitudinalLimits RIVIAN_LONG_LIMITS = { | ||
60 | .max_accel = 200, | ||
61 | .min_accel = -350, | ||
62 | .inactive_accel = 0, | ||
63 | }; | ||
64 | |||
65 | 5498 | bool tx = true; | |
66 | 5498 | int bus = GET_BUS(to_send); | |
67 | |||
68 |
2/2✓ Branch 0 taken 5494 times.
✓ Branch 1 taken 4 times.
|
5498 | if (bus == 0) { |
69 | 5494 | int addr = GET_ADDR(to_send); | |
70 | |||
71 | // Steering control | ||
72 |
2/2✓ Branch 0 taken 4886 times.
✓ Branch 1 taken 608 times.
|
5494 | if (addr == 0x120) { |
73 | 4886 | int desired_torque = ((GET_BYTE(to_send, 2) << 3U) | (GET_BYTE(to_send, 3) >> 5U)) - 1024U; | |
74 | 4886 | bool steer_req = GET_BIT(to_send, 28U); | |
75 | |||
76 |
2/2✓ Branch 1 taken 2434 times.
✓ Branch 2 taken 2452 times.
|
4886 | if (steer_torque_cmd_checks(desired_torque, steer_req, RIVIAN_STEERING_LIMITS)) { |
77 | 2434 | tx = false; | |
78 | } | ||
79 | } | ||
80 | |||
81 | // Longitudinal control | ||
82 |
2/2✓ Branch 0 taken 608 times.
✓ Branch 1 taken 4886 times.
|
5494 | if (addr == 0x160) { |
83 | 608 | int raw_accel = ((GET_BYTE(to_send, 2) << 3) | (GET_BYTE(to_send, 3) >> 5)) - 1024U; | |
84 |
2/2✓ Branch 1 taken 376 times.
✓ Branch 2 taken 232 times.
|
608 | if (longitudinal_accel_checks(raw_accel, RIVIAN_LONG_LIMITS)) { |
85 | 376 | tx = false; | |
86 | } | ||
87 | } | ||
88 | } | ||
89 | |||
90 | 5498 | return tx; | |
91 | } | ||
92 | |||
93 | 16896 | static int rivian_fwd_hook(int bus, int addr) { | |
94 | 16896 | int bus_fwd = -1; | |
95 | 16896 | bool block_msg = false; | |
96 | |||
97 |
2/2✓ Branch 0 taken 5632 times.
✓ Branch 1 taken 11264 times.
|
16896 | if (bus == 0) { |
98 | // SCCM_WheelTouch | ||
99 |
2/2✓ Branch 0 taken 2 times.
✓ Branch 1 taken 5630 times.
|
5632 | if (addr == 0x321) { |
100 | 2 | block_msg = true; | |
101 | } | ||
102 | |||
103 |
2/2✓ Branch 0 taken 5630 times.
✓ Branch 1 taken 2 times.
|
5632 | if (!block_msg) { |
104 | 5630 | bus_fwd = 2; | |
105 | } | ||
106 | } | ||
107 | |||
108 |
2/2✓ Branch 0 taken 5632 times.
✓ Branch 1 taken 11264 times.
|
16896 | if (bus == 2) { |
109 | // ACM_lkaHbaCmd | ||
110 |
2/2✓ Branch 0 taken 2 times.
✓ Branch 1 taken 5630 times.
|
5632 | if (addr == 0x120) { |
111 | 2 | block_msg = true; | |
112 | } | ||
113 | |||
114 | // ACM_longitudinalRequest | ||
115 |
4/4✓ Branch 0 taken 2816 times.
✓ Branch 1 taken 2816 times.
✓ Branch 2 taken 1 times.
✓ Branch 3 taken 2815 times.
|
5632 | if (rivian_longitudinal && (addr == 0x160)) { |
116 | 1 | block_msg = true; | |
117 | } | ||
118 | |||
119 |
2/2✓ Branch 0 taken 5629 times.
✓ Branch 1 taken 3 times.
|
5632 | if (!block_msg) { |
120 | 5629 | bus_fwd = 0; | |
121 | } | ||
122 | } | ||
123 | |||
124 | 16896 | return bus_fwd; | |
125 | } | ||
126 | |||
127 | 62 | static safety_config rivian_init(uint16_t param) { | |
128 | // 0x120 = ACM_lkaHbaCmd, 0x160 = ACM_longitudinalRequest, 0x321 = SCCM_WheelTouch | ||
129 | static const CanMsg RIVIAN_TX_MSGS[] = {{0x120, 0, 8}, {0x321, 2, 7}}; | ||
130 | static const CanMsg RIVIAN_LONG_TX_MSGS[] = {{0x120, 0, 8}, {0x321, 2, 7}, {0x160, 0, 5}}; | ||
131 | |||
132 | static RxCheck rivian_rx_checks[] = { | ||
133 | {.msg = {{0x208, 0, 8, .frequency = 50U}, { 0 }, { 0 }}}, // ESP_Status (speed) | ||
134 | {.msg = {{0x150, 0, 7, .frequency = 50U}, { 0 }, { 0 }}}, // VDM_PropStatus (gas pedal) | ||
135 | {.msg = {{0x38f, 0, 6, .frequency = 50U}, { 0 }, { 0 }}}, // iBESP2 (brakes) | ||
136 | {.msg = {{0x100, 2, 8, .frequency = 100U}, { 0 }, { 0 }}}, // ACM_Status (cruise state) | ||
137 | }; | ||
138 | |||
139 | UNUSED(param); | ||
140 | #ifdef ALLOW_DEBUG | ||
141 | 62 | const int FLAG_RIVIAN_LONG_CONTROL = 1; | |
142 | 62 | rivian_longitudinal = GET_FLAG(param, FLAG_RIVIAN_LONG_CONTROL); | |
143 | #endif | ||
144 | |||
145 |
2/2✓ Branch 0 taken 32 times.
✓ Branch 1 taken 30 times.
|
62 | return rivian_longitudinal ? BUILD_SAFETY_CFG(rivian_rx_checks, RIVIAN_LONG_TX_MSGS) : \ |
146 | BUILD_SAFETY_CFG(rivian_rx_checks, RIVIAN_TX_MSGS); | ||
147 | } | ||
148 | |||
149 | const safety_hooks rivian_hooks = { | ||
150 | .init = rivian_init, | ||
151 | .rx = rivian_rx_hook, | ||
152 | .tx = rivian_tx_hook, | ||
153 | .fwd = rivian_fwd_hook, | ||
154 | }; | ||
155 |