[fix] invalid choice

divshekhar · Jan 26, 2022 · 4bfe0ac · 4bfe0ac
1 parent 7414bcd
commit 4bfe0ac
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/src/osps.cpp b/src/osps.cpp
@@ -62,6 +62,12 @@ int main()
     std::cin >> choice;
     std::cout << "You have chosen " << scheduling_algo(choice) << std::endl;
 
+    if (!(choice >= 1 && choice <= 5))
+    {
+        std::cout << "Invalid choice. Exiting..." << std::endl;
+        return 0;
+    }
+
     int n;
     osps::Process p[100];