C基本头部命令

时间:2012-09-11 01:28:53

标签: c linux tail head

我正在尝试从linux中为我的编程类重新创建head和tail命令。 我们刚开始使用C,所以我对分配内存和指针的想法不熟悉。 我想知道为什么这不起作用。

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

int main(int argc,char **argv){

    /* Checks if correct amount of arguements */

    if(argc != 2 || argc != 4){
        printf("Usage: %s head <file> \n Or: head <file> -n <number of characters>", argv[0]);
        exit(-1);
    }

    if(strcmp(argv[1], "-n" != 0)){
        char fileName[strlen(argv[1])] = argv[1];
    }
}

//Compile error on char fileName[strlen(argv[1])] = argv[1];

任何其他见解也会有所帮助。

2 个答案:

答案 0 :(得分:3)

我认为写作更好:

char fileName[strlen(argv[1])+1];
strcpy(fileName, argv[1]);

或(如果你不打算复制字符串):

char* fileName = argv[1];

答案 1 :(得分:2)

首先,您的使用与您的参数检查不匹配。根据用法,您必须使用以下之一:

head <filename>
head <filename> -n <count>

换句话说,argv[1] 总是文件名,如果有两个以上的参数,argv[2]是需要设置为-n的文件名

其次,除非您想使用VLA(可变长度数组),否则您应该只需设置一个指向filename参数的指针:

char *fileName = argv[1];

您根本不需要更改它(您可能只是将其传递给fopen,因此尝试制作另一份副本是浪费。

此外,您的if声明错误为or,它应该是and。保证argc不会是2或不是4,因为它不能同时为两个。

我会从以下内容开始:

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

static int usage (void) {
    printf ("Usage: head <file>\n");
    printf ("   or: head <file> -n <number of characters>\n");
    return -1;
}

int main (int argc,char *argv[]) {
    char *fileName;
    int lineCount;

    // Checks if correct arguments

    if ((argc != 2) && (argc != 4)) return usage();

    if ((argc == 4) && (strcmp(argv[2], "-n" != 0)) return usage();

    // Get file spec and line count

    fileName = argv[1];

    lineCount = (argc == 2) ? 10 : atoi (argv[3]); // or strtol for purists
    if (linecount < 0) lineCount = 0;

    // Now go ahead and implement the logic for head.

}