lsコマンドもどき。

概要

C言語の勉強と遊びを兼ねて。
フォーマット識別子はここを参考にした。
Linux上で動かすことを想定している。

ソースコード

/* 
 * lsコマンドに、ディレクトリか通常ファイルかを判別して 
 * 表示するプログラム。 
 * 引数は1個だけとる。その引数はlsコマンドで通常使われるような、 
 * 表示したいディレクトリのパス。 
 * ディレクトリか通常ファイルかjpgファイルか、を判別できるようにした。 
 * ディレクトリであれば、[DIR]が頭につく。 
 * jpgファイルであれば、[JPG]が頭につく。 
 */ 
  
#include <stdio.h> 
#include <stdlib.h> 
#include <dirent.h> 
#include <string.h> 
#include <sys/types.h> 
#include <sys/stat.h> 
  
/* 
 * jpgファイルかどうかを判別する。 
 * jpgファイルなら1を返す。 
 * jpgファイルでないなら0を返す。 
 * ファイルを開くことに失敗したら、-1を返す。 
 */ 
int chkjpg ( char *path ) { 
  FILE *file; 
  unsigned char buf[3]; 
  int i, size; 
  // マジックナンバーと比較する時に使うカウント用変数。 
  unsigned short int count = 0; 
  // jpgファイルのマジックナンバー(フォーマット識別子) 
  unsigned int jpg_magic[3] = { 255, 216, 255 }; 
  
  // ファイルを開く。 
  file = fopen( path, "rb" ); 
  if ( file == NULL ) { 
    perror( "fopen" ); 
    return -1; 
  } 
  
  size = fread( buf, 1, 3, file ); 
  
/* 
  printf( "** chkjpg " ); 
  for ( i = 0; i < size; i++ ) { 
    printf( "%02x", buf[i] ); 
  } 
  printf( "\n" ); 
  printf( "**chkjpg2 " ); 
  for (i = 0; i < size; i++ ) { 
    printf( "%d", buf[i] ); 
  } 
  printf( "\n" ); 
  printf( "**chkjpg_compare " ); 
  for ( i = 0; i < size; i++ ) { 
    if ( buf[i] == magic[i] ) { 
      printf( "OK!" ); 
    } 
  } 
  printf( "\n" ); 
*/ 
  // マジックナンバーを比較する。正しければcount変数の値が増える。 
  for ( i = 0; i < size; i++ ) { 
    if ( buf[i] == jpg_magic[i] ) { 
      count++; 
    } 
  } 
  
  fclose( file ); 
  if ( count == size ) { 
    return 1; 
  } else {  
    return 0; 
  } 
} 
  
int main ( int argc, char *argv[] ) 
{ 
  // DIR型はdirent.hをincludeすると使えるようになる。 
  // ディレクトリを指し示す。 
  DIR *dir; 
  // ディレクトリの中にあるファイルの情報を保持する。 
  struct dirent *dp; 
  // dirent型よりもより詳しいファイルの情報を保持する。 
  struct stat sb; 
  // Debianでのファイル名の文字数制限は255byteらしい。 
  // NAME_MAXはdirent.hをincludeすると使えるようになる。 
  char path[NAME_MAX * 2]; 
  // 一時的にパスを格納する。最終的に絶対パスが入る。 
  char tmp_path[NAME_MAX * 2]; 
  
  /* 
   * 引数があるかどうかを判断して、無いならカレントディレクトリをpathに保存。 
   * 引数があるなら、その引数をpathに保存。 
   */ 
  if ( argc <= 1 ) { 
    strcpy( path, "." ); 
  } else { 
    strcpy( path, argv[1] ); 
  } 
  
  // ディレクトリを開く。 
  dir = opendir( path ); 
  if ( dir == NULL ) { 
    perror( "opendir" ); 
    // exit関数はstdlib.hで定義されている。 
    exit( EXIT_FAILURE ); 
  } 
  
  // ディレクトリの中のファイルを見ていく。 
  for ( dp = readdir ( dir ); dp != NULL; dp = readdir ( dir ) ) { 
    // stat関数に絶対パスが渡るようにする。 
    strcpy( tmp_path, path ); 
    strcat( tmp_path, "/" ); 
    strcat( tmp_path, dp -> d_name ); 
    if ( stat( tmp_path, &sb ) == -1 ) { 
      // perror( "stat" ); 
      // closedir( dir ); 
      // exit( EXIT_FAILURE ); 
      continue; 
    } 
  
    // ディレクトリか通常ファイルかを判別。 
    if ( S_ISDIR( sb.st_mode ) ) { 
      printf("[DIR]"); 
    } else if ( chkjpg ( tmp_path ) ) { // jpgファイルか通常ファイルかを判別。 
      printf("[JPG]"); 
    } 
  
    // ファイル名を表示。 
    printf( "name : %s\n", dp -> d_name ); 
  } 
  
  closedir( dir ); 
  exit( EXIT_SUCCESS ); 
} 

コメント

タイトルとURLをコピーしました