Linux命令的C mmap实现

Question

我有以下代码，它们基本上重现了Linux中wc命令的功能。我的问题是如何使用mmap重写代码？我知道我可以先使用struct stat sb;，然后再使用char *file_in_memory = mmap(NULL, sb.st_size, PROT_READ, MAP_PRIVATE, fd, 0);，但我无法使用它/我不知道如何在while循环while ((n = read(file, buffer, LUNG_BUF - 1)) > 0)中正确实现它。在运行代码后的尝试中，它将仅显示值0。

#include <stdio.h>
#include <string.h>
#include <unistd.h>
#include <fcntl.h>
#include <sys/types.h>

#define LUNG_BUF 4096

int main(int argc, char** argv)
{
  int bytes = 0;
  int words = 0;
  int newLine = 0;
  int max_value;  // the maximum of the above three
  int dim;        // string width of the max value

  char buffer[LUNG_BUF];
  enum states { WHITESPACE, WORD };
  int state = WHITESPACE;
  if ( argc !=2 )
  {
    printf( "No file name\n%s", argv[0]);
  }
  else
  {
    int file = open(argv[1], O_RDONLY);

    if(file < 0)
    {
      printf("can not open :%s\n",argv[1]);
    }
    else
    {
      char *thefile = argv[1];
      size_t n;

      while ((n = read(file, buffer, LUNG_BUF - 1)) > 0)
      {
        buffer[n] = '\0';
        char *ptr = buffer;

        while (*ptr)
        {
          bytes++;
          if (*ptr == ' ' || *ptr == '\t')
          {
            state = WHITESPACE;
          }
          else if (*ptr == '\n')
          {
            newLine++;
            state = WHITESPACE;
          }
          else
          {
            if (state == WHITESPACE)
            {
                words++;
            }
            state = WORD;
          }
          ptr++;
        }
      }

      // find out the largest value of all and determine the printed width of it
      max_value = newLine;
      if (words > max_value)
        max_value = words;
      if (bytes > max_value)
        max_value = bytes;
      dim = snprintf(NULL, 0, "%d", max_value);

      // print lines, words, bytes and filename aligned to the longest number
      printf("%*d %*d %*d %s\n", dim, newLine, dim, words, dim, bytes, thefile);
    }
  }
}

我正在尝试的脚本：

#include <stdio.h>
#include <string.h>
#include <unistd.h>
#include <fcntl.h>
#include <sys/types.h>

#define LUNG_BUF 4096

int main(int argc, char** argv)
{
  int bytes = 0;
  int words = 0;
  int newLine = 0;
  int max_value;  // the maximum of the above three
  int dim;        // string width of the max value

  char buffer[LUNG_BUF];
  enum states { WHITESPACE, WORD };
  int state = WHITESPACE;
  if ( argc !=2 )
  {
    printf( "No file name\n%s", argv[0]);
  }
  else
  {
    int file = open(argv[1], O_RDONLY);

    if(file < 0)
    {
      printf("can not open :%s\n",argv[1]);
    }
    else
    {
      char *thefile = argv[1];
      size_t n;
      struct stat sb;
      char *file_in_memory = mmap(NULL, sb.st_size, PROT_READ, MAP_PRIVATE, fd, 0);
      for(int i=0;i<=sb.st_size;i++)
      {
        buffer[i] = '\0';
        char *ptr = buffer;
        while (*ptr)
        {
          bytes++;
          if (*ptr == ' ' || *ptr == '\t')
          {
            state = WHITESPACE;
          }
          else if (*ptr == '\n')
          {
            newLine++;
            state = WHITESPACE;
          }
          else
          {
            if (state == WHITESPACE)
            {
                words++;
            }
            state = WORD;
          }
          ptr++;
        }
      }

      // find out the largest value of all and determine the printed width of it
      max_value = newLine;
      if (words > max_value)
        max_value = words;
      if (bytes > max_value)
        max_value = bytes;
      dim = snprintf(NULL, 0, "%d", max_value);

      // print lines, words, bytes and filename aligned to the longest number
      printf("%*d %*d %*d %s\n", dim, newLine, dim, words, dim, bytes, thefile);
      munmap(file_in_memory, sb.st_size);
      close(file);
    }
  }
}

Answer 1

您上面发布的代码没有编译，并且有很多问题。我在下面整理了一下，希望对您有所帮助。我尝试不对其进行太多更改，以便您可以看到我的工作。

您实际上没有调用stat，并且传递给mmap的fd变量不是您用来打开文件的变量。如果可以的话，我将始终使用“ -Wall -Werror”编译您的代码。

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>
#include <fcntl.h>
#include <sys/types.h>
#include <sys/stat.h>
#include <sys/mman.h>

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

  if ( argc !=2 )
  {
    printf( "No file name\n%s", argv[0]);
    exit(-1);
  }

  char *fileName = argv[1];

  int file = open(fileName, O_RDONLY);

  if(file < 0)
  {
    perror("Error: ");
    exit(-1);
  }

  struct stat sb = {0};
  stat(fileName, &sb);
  char *filePtr = mmap(NULL, sb.st_size, PROT_READ, MAP_PRIVATE, file, 0);
  if (filePtr == MAP_FAILED)
  {
      perror("Error:");
      exit(-1);
  }

  int bytes = sb.st_size;
  int words = 0;
  int newLine = 0;

  enum states { WHITESPACE, WORD };
  int state = WHITESPACE;

  for(size_t pos=0;pos<=sb.st_size;pos++)
  {
      if (state == WHITESPACE)
      {
          if (filePtr[pos] == '\n')
          {
              newLine++;
          }
          else if ((filePtr[pos] != ' ') && (filePtr[pos] != '\t'))
          {
              state = WORD;
          }
      }
      else // (state == WORD)
      {
        if (filePtr[pos] == ' ' || filePtr[pos] == '\t')
        {
            state = WHITESPACE;
            words++;
        }
        else if (filePtr[pos] == '\n')
        {
            state = WHITESPACE;
            words++;
            newLine++;
        }
      }
  }

  // Max value is always bytes

  int dim = snprintf(NULL, 0, "%d", bytes);

  // print lines, words, bytes and filename aligned to the longest number
  printf("%*d %*d %*d %s\n", dim, newLine, dim, words, dim, bytes, fileName);

  munmap(filePtr, sb.st_size);
  close(file);
}