add safety checks

This commit is contained in:
2026-07-29 20:38:24 +02:00
parent 4eae9c7f23
commit 6b7b220b87
+18 -2
View File
@@ -1,8 +1,12 @@
#include "includes/include.h" #include "includes/include.h"
#include "src/driver.h" #include "src/driver.h"
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
int main(int argc, char *argv[]) { int main(int argc, char *argv[]) {
int sens; int sens;
char input[100];
printf("K1NG PRO (4k) driver | Cli\n"); printf("K1NG PRO (4k) driver | Cli\n");
printf("set any sens from 50-26000\n\n"); printf("set any sens from 50-26000\n\n");
@@ -10,8 +14,15 @@ int main(int argc, char *argv[]) {
if (argc < 2) { if (argc < 2) {
printf("set sensitivity to: "); printf("set sensitivity to: ");
if (scanf("%d", &sens) != 1) { if (fgets(input, sizeof(input), stdin) == NULL || input[0] == '\n') {
printf("u forgot to input a number?\n"); printf("incorrect input press enter to try again\n");
return 1;
}
char *endptr;
sens = (int)strtol(input, &endptr, 10);
if (*endptr != '\n' && *endptr != '\0') {
printf("incorrect input press enter to try again\n");
return 1; return 1;
} }
} else { } else {
@@ -19,6 +30,11 @@ int main(int argc, char *argv[]) {
printf("set sensitivity to: %d\n", sens); printf("set sensitivity to: %d\n", sens);
} }
if (sens < 50 || sens > 26000 || sens % 50 != 0) {
printf("incorrect input press enter to try again\n");
return 1;
}
run_driver(sens); run_driver(sens);
return 0; return 0;
} }