fmu:Add Range checking for PWM5 modes

PX4 · Oct 23, 2018 · 99294b2 · 99294b2
1 parent d832b4f
commit 99294b2
Showing 1 changed file with 11 additions and 2 deletions.
diff --git a/src/drivers/px4fmu/fmu.cpp b/src/drivers/px4fmu/fmu.cpp
@@ -1835,10 +1835,14 @@ PX4FMU::pwm_ioctl(file *filp, int cmd, unsigned long arg)
 
 	/* FALLTHROUGH */
 	case PWM_SERVO_SET(5):
+		if (_mode < MODE_6PWM) {
+			ret = -EINVAL;
+			break;
+		}
 
 	/* FALLTHROUGH */
 	case PWM_SERVO_SET(4):
-		if (_mode < MODE_6PWM) {
+		if (_mode < MODE_5PWM1CAP) {
 			ret = -EINVAL;
 			break;
 		}
@@ -1901,12 +1905,17 @@ PX4FMU::pwm_ioctl(file *filp, int cmd, unsigned long arg)
 
 	/* FALLTHROUGH */
 	case PWM_SERVO_GET(5):
-	case PWM_SERVO_GET(4):
 		if (_mode < MODE_6PWM) {
 			ret = -EINVAL;
 			break;
 		}
 
+	case PWM_SERVO_GET(4):
+		if (_mode < MODE_5PWM1CAP) {
+			ret = -EINVAL;
+			break;
+		}
+
 #endif
 
 	/* FALLTHROUGH */