コード例 #1
0
ファイル: filetype.cpp プロジェクト: tempbottle/fawkes
/** Get mime-type of file given by file descriptor.
 * This function gives a brief mime-type for the given file.
 * @param fd file descriptor of open file, make sure the file descriptor is rewinded.
 * Warning, the file descriptor is closed by the underlying libmagic. Use dup() to
 * duplicate it and pass this as file descriptor if you need the file afterwards.
 * @return descriptive string
 */
std::string
mimetype_file(int fd)
{
  std::string rv;

#ifdef HAVE_LIBMAGIC
#  ifdef MAGIC_MIME_TYPE
  magic_t m = magic_open( MAGIC_ERROR | MAGIC_MIME_TYPE );
#  else
  magic_t m = magic_open( MAGIC_ERROR | MAGIC_MIME );
#  endif
  magic_load( m, NULL );

  const char * res = magic_descriptor( m, fd );
  if ( res == NULL ) {
    fawkes::Exception e("Failed to determine mime type of descriptor: %s", magic_error(m));
    magic_close(m);
    throw e;
  }

  rv = res;
#  ifndef MAGIC_MIME_TYPE
  rv = rv.substr(0, rv.find(","));
#  endif
  magic_close(m);
#else
  throw fawkes::Exception("Failed to determine file type "
			  "(libmagic not available at compile time)");
#endif
  return rv;
}
コード例 #2
0
ファイル: filetype.cpp プロジェクト: tempbottle/fawkes
/** Get filetype of file given by file descriptor.
 * Returns a long decriptive string of the filetype, similar to the file
 * console utility.
 * @param fd file descriptor of open file, make sure the file descriptor is rewinded
 * Warning, the file descriptor is closed by the underlying libmagic. Use dup() to
 * duplicate it and pass this as file descriptor if you need the file afterwards.
 * @return descriptive string
 */
std::string
filetype_file(int fd)
{
  std::string rv;

#ifdef HAVE_LIBMAGIC
  magic_t m = magic_open( MAGIC_ERROR );
  magic_load( m, NULL );

  const char * res = magic_descriptor( m, fd );
  if ( res == NULL ) {
    fawkes::Exception e("Failed to determine file type of descriptor: %s", magic_error(m));
    magic_close(m);
    throw e;
  }

  rv = res;
  magic_close( m );
#else
  throw fawkes::Exception("Failed to determine file type "
			  "(libmagic not available at compile time)");
#endif

  return rv;
}
コード例 #3
0
ファイル: magic.c プロジェクト: montekki/radare2
R_API const char *r_magic_descriptor(RMagic* m, int fd) {
	return magic_descriptor (m, fd);
}
コード例 #4
0
ファイル: spider.cpp プロジェクト: danilfilippov93/u-search
const char *Spider::DetectMimeType(const std::string &path) {
  int smb_fd = smbc_open(path.c_str(), O_RDONLY, 0);
  if (UNLIKELY(smb_fd < 0)) {
    if (LIKELY(errno == EISDIR))
      return "inode/directory";

    DetectError();
    MSS_ERROR(("smbc_open " + path).c_str(), error_);
    return "unknown";
  }

  // Extract name of the file
  // Don't detele '/' symbol it need to form path.
  std::string name(path, path.rfind("/"));

  // Create a storage for file header in TMPDIR and open it
  int fd = open((TMPDIR + name).c_str(), O_CREAT | O_RDWR | O_EXCL,
                00744 /* rwxr--r-- */);
  if (UNLIKELY(fd == -1)) {
    if (LIKELY(errno = ENOTDIR)) {
      // TODO(yulyugin): Check if TMPDIR doesn't exists create it.
    }
    DetectError();
    MSS_ERROR("open", error_);
    if (UNLIKELY(smbc_close(smb_fd))) {
      DetectError();
      MSS_ERROR("smbc_close", error_);
    }
    return "unknown";
  }

  void *buf = malloc(HEADERSIZE);  // Buffer to store header.

  // Copy file header to TMPDIR
  if (UNLIKELY(smbc_read(smb_fd, buf, HEADERSIZE) < 0)) {
    DetectError();
    MSS_ERROR("smbc_read", error_);
    if (UNLIKELY(smbc_close(smb_fd))) {
      DetectError();
      MSS_ERROR("smbc_close", error_);
    }
    if (UNLIKELY(close(fd))) {
      DetectError();
      MSS_ERROR("close", error_);
    }
    free(buf);
    return "unknown";
  }

  if (UNLIKELY(write(fd, buf, HEADERSIZE) < 0)) {
    DetectError();
    MSS_ERROR("write", error_);
    if (UNLIKELY(smbc_close(smb_fd))) {
      DetectError();
      MSS_ERROR("smbc_close", error_);
    }
    if (UNLIKELY(close(fd))) {
      DetectError();
      MSS_ERROR("close", error_);
    }
    free(buf);
    return "unknown";
  }

  // Move to the begining of the file
  if (UNLIKELY(lseek(fd, 0, SEEK_SET) != 0)) {
    DetectError();
    MSS_ERROR("lseek", error_);
    if (UNLIKELY(smbc_close(smb_fd))) {
      DetectError();
      MSS_ERROR("smbc_close", error_);
    }
    if (UNLIKELY(close(fd))) {
      DetectError();
      MSS_ERROR("close", error_);
    }
    free(buf);
    return "unknown";
  }

  const char *mime_type = magic_descriptor(cookie_, fd);
  if (UNLIKELY(mime_type == NULL)) {
    error_ = magic_errno(cookie_);
    MSS_ERROR("magic_descriptor", error_);
    if (UNLIKELY(smbc_close(smb_fd))) {
      DetectError();
      MSS_ERROR("smbc_close", error_);
    }
    if (UNLIKELY(close(fd))) {
      DetectError();
      MSS_ERROR("close", error_);
    }
    free(buf);
    return "unknown";
  }

  if (UNLIKELY(smbc_close(smb_fd))) {
    DetectError();
    MSS_ERROR("smbc_close", error_);
  }
  free(buf);

  // Remove temporary file.
  if (UNLIKELY(unlink((TMPDIR + name).c_str()))) {
    DetectError();
    MSS_ERROR("unlink", error_);
  }

  return mime_type;
}