diff options
| -rw-r--r-- | tools/checkwps/checkwps.c | 5 |
1 files changed, 2 insertions, 3 deletions
diff --git a/tools/checkwps/checkwps.c b/tools/checkwps/checkwps.c index bfcb99f..273db78 100644 --- a/tools/checkwps/checkwps.c +++ b/tools/checkwps/checkwps.c @@ -294,9 +294,8 @@ int main(int argc, char **argv) while (argv[filearg]) { printf("Checking %s...\n", argv[filearg]); #ifdef HAVE_REMOTE_LCD - if((strcmp(&argv[filearg][strlen(argv[filearg])-4], "rwps") == 0) || - (strcmp(&argv[filearg][strlen(argv[filearg])-4], "rsbs") == 0) || - (strcmp(&argv[filearg][strlen(argv[filearg])-4], "rfms") == 0)) + char *ext = strrchr(argv[filearg], '.'); + if (strcmp(ext, "rwps") == 0 || strcmp(ext, "rsbs") == 0 || strcmp(ext, "rfms") == 0) screen = SCREEN_REMOTE; else screen = SCREEN_MAIN; |