Программа-калькулятор на C с использованием аргументов командной строки (argc, argv)

Я написал программу для получения входных аргументов из командной строки и создания базового калькулятора на C. Код ниже:

#include <stdio.h>
#include <stdlib.h>
#include <ctype.h>


void CheckArgumentCount(int argc);
int CheckOperands(char *argv[]);
long Calculate(long nr1, long nr2, char operation);

int main(int argc, char *argv[])
{
    long nr1 = CheckOperands(&argv[1]);
    long nr2 = CheckOperands(&argv[3]);
    int result;

    CheckArgumentCount(argc);
    result = Calculate(nr1, nr2, *argv[2]);
    printf("Result: %d", result);

    return 0;
}


void CheckArgumentCount(int argc)
{
    if (argc > 3 || argc < 3)
    {
        puts("Wrong number of arguments to perform the calculation.\n");
    }
}

int CheckOperands(char *argv[])
{
        int i = 1;
        
        while (*argv[i] != '\0')
        {
            if (*argv[i] < '0' || *argv[i] > '9')
            {
                return *argv[i];
            }
            argv++;
        }
        return atoi(argv[i]);
}


long Calculate(long nr1, long nr2, char operation)
{
    long result = 0;

    switch (operation)
    {
        case '+':
            result = nr1 + nr2;
            break;
        case '-':
            result = nr1 - nr2;
            break;
        case '*':
            result = nr1 * nr2;
            break;
        case '/':
            if (nr2 == 0)
            {
                puts("Error! Divion by zero.");
            }
            else
            {
                result = nr1 / nr2;
            }
            break;
        default:
            puts("Operator invalid.");
            break;
    }
    return result;
}

Который компилируется нормально, так как компилятор не показывает никаких ошибок. Однако я получаю эту ошибку на терминале zsh при запуске, когда я предоставляю 3 аргумента, например:

zsh: segmentation fault ./calc 1 + 3

Я предполагаю, что это может что-то сделать с моей функцией CheckOperands? Любые советы будут оценены.


c argv calculator argc
person soundsfierce    schedule 16.11.2020    source источник
comment
OT: argv != 3 проще, чем argc > 3 || argc < 3, не так ли?   -  person Jabberwocky    schedule 16.11.2020
comment
Вам необходимо проверить количество аргументов перед разыменованием любого argc[n], где n > 1. Также CheckArgumentCount просто выводит сообщение об ошибке, но в этом случае вы не прерываете работу.   -  person Jabberwocky    schedule 16.11.2020

Ответы (1)


arrow_upward
1
arrow_downward

Вам нужно проверить количество аргументов перед использованием CheckOperands или, скорее, перед разыменованием любого argc[n] с n › 1.

Вы хотите это:

int main(int argc, char* argv[])
{
  if (!CheckArgumentCount(argc))
    exit(1);   // if argument count differenmt from 3 exit program
  ...

и это:

int CheckArgumentCount(int argc)
{
  if (argc != 4)   // argc is one more because if the command 
                      invocation which is argv[0]
  {
    puts("Wrong number of arguments to perform the calculation.\n");
    return 0;
  }

  return 1;
}

И ошибок больше в CheckOperands.

Общий исправленный код:

#include <stdio.h>
#include <stdlib.h>
#include <ctype.h>

int CheckArgumentCount(int argc);
int CheckOperands(char* operand);
long Calculate(long nr1, long nr2, char operation);

int main(int argc, char* argv[])
{
  if (!CheckArgumentCount(argc))
    exit(1);

  long nr1 = CheckOperands(argv[1]);
  long nr2 = CheckOperands(argv[3]);
  int result;


  result = Calculate(nr1, nr2, argv[2][0]);
  printf("Result: %d", result);

  return 0;
}


int CheckArgumentCount(int argc)
{
  if (argc != 4)
  {
    puts("Wrong number of arguments to perform the calculation.\n");
    return 0;
  }

  return 1;
}

int CheckOperands(char* operand)
{
  int i = 0;

  while (operand[i] != '\0')
  {
    if (operand[i] < '0' || operand[i] > '9')
    {
      return operand[i];
    }

    i++;
  }
  return atoi(operand);
}


long Calculate(long nr1, long nr2, char operation)
{
  long result = 0;

  switch (operation)
  {
  case '+':
    result = nr1 + nr2;
    break;
  case '-':
    result = nr1 - nr2;
    break;
  case '*':
    result = nr1 * nr2;
    break;
  case '/':
    if (nr2 == 0)
    {
      puts("Error! Divion by zero.");
    }
    else
    {
      result = nr1 / nr2;
    }
    break;
  default:
    puts("Operator invalid.");
    break;
  }
  return result;
}
person Jabberwocky    schedule 16.11.2020
comment
Спасибо за советы, я исправил код со всеми предложениями, однако он все еще не выполняет расчет и просто зависает после ввода аргументов. Действительно ли функция CheckOperands ведет себя правильно? - person soundsfierce; 16.11.2020