generated from PlexSheep/rs-base
Compare commits
4 commits
f5ec931b33
...
c9782ef6dc
Author | SHA1 | Date | |
---|---|---|---|
c9782ef6dc | |||
1cb492d90a | |||
7bea6ede4c | |||
e829027eb5 |
2 changed files with 41 additions and 31 deletions
|
@ -1,13 +1,8 @@
|
|||
//! # Dump data
|
||||
//!
|
||||
//! This crate is part of [`pt`](../libpt/index.html), but can also be used as a standalone
|
||||
//! module.
|
||||
//!
|
||||
//! Hedu is made for hexdumping data. `libpt` offers a cli application using this module.
|
||||
//! # Underlying mechanisms of how hedu works with data
|
||||
|
||||
use std::io::{prelude::*, Read, SeekFrom};
|
||||
|
||||
use anyhow::{bail, Result};
|
||||
use anyhow::Result;
|
||||
use libpt::bintols::display::humanbytes;
|
||||
use libpt::log::{debug, error, trace, warn};
|
||||
|
||||
|
@ -16,6 +11,23 @@ pub const LINE_SEP_HORIZ: char = '─';
|
|||
pub const LINE_SEP_VERT: char = '│';
|
||||
pub const CHAR_BORDER: &'static str = "|";
|
||||
|
||||
pub trait DataSource: Read {
|
||||
fn skip(&mut self, length: usize) -> std::io::Result<()>;
|
||||
}
|
||||
impl DataSource for std::io::Stdin {
|
||||
fn skip(&mut self, _length: usize) -> std::io::Result<()> {
|
||||
warn!("can't skip bytes for the stdin!");
|
||||
Ok(())
|
||||
}
|
||||
}
|
||||
impl DataSource for std::fs::File {
|
||||
fn skip(&mut self, length: usize) -> std::io::Result<()> {
|
||||
self.seek(SeekFrom::Current(length as i64))?;
|
||||
// returns the new position from the start, we don't need it here.
|
||||
Ok(())
|
||||
}
|
||||
}
|
||||
|
||||
#[derive(Debug)]
|
||||
pub struct Hedu {
|
||||
pub chars: bool,
|
||||
|
@ -33,6 +45,7 @@ pub struct Hedu {
|
|||
}
|
||||
|
||||
impl Hedu {
|
||||
/// create a new hedu struct, initializing all kinds of configs and values needed while running
|
||||
pub fn new(chars: bool, skip: usize, show_identical: bool, limit: usize) -> Self {
|
||||
Hedu {
|
||||
chars,
|
||||
|
@ -49,11 +62,15 @@ impl Hedu {
|
|||
first_iter: true,
|
||||
}
|
||||
}
|
||||
|
||||
/// print the display buffer to the stdout
|
||||
#[inline]
|
||||
pub fn display(&mut self) {
|
||||
println!("{}", self.display_buf);
|
||||
self.display_buf = String::new();
|
||||
}
|
||||
|
||||
/// display a separator line
|
||||
#[inline]
|
||||
pub fn sep(&mut self) {
|
||||
if self.chars {
|
||||
|
@ -63,11 +80,15 @@ impl Hedu {
|
|||
}
|
||||
self.display();
|
||||
}
|
||||
|
||||
/// display a newline
|
||||
#[inline]
|
||||
pub fn newline(&mut self) {
|
||||
self.display_buf += "\n";
|
||||
self.display();
|
||||
}
|
||||
|
||||
/// display the dump of a single line of data
|
||||
fn dump_a_line(&mut self) {
|
||||
self.display_buf += &format!("{:08X} {LINE_SEP_VERT} ", self.data_idx);
|
||||
if self.len != 0 {
|
||||
|
@ -105,6 +126,7 @@ impl Hedu {
|
|||
self.display();
|
||||
}
|
||||
|
||||
/// skip duplicate lines and display skip info
|
||||
fn skip_lines(&mut self, data: &mut dyn DataSource) -> Result<()> {
|
||||
trace!(buf = format!("{:?}", self.buf), "found a duplicating line");
|
||||
let start_line = self.data_idx;
|
||||
|
@ -115,7 +137,7 @@ impl Hedu {
|
|||
"******** {LINE_SEP_VERT} {:<49}",
|
||||
format!(
|
||||
"(repeats {} lines)",
|
||||
self.data_idx - start_line / (BYTES_PER_LINE) + 1
|
||||
(self.data_idx - start_line / (BYTES_PER_LINE) + 1) / BYTES_PER_LINE
|
||||
)
|
||||
);
|
||||
if self.chars {
|
||||
|
@ -125,10 +147,11 @@ impl Hedu {
|
|||
buf = format!("{:X?}", self.buf),
|
||||
"dumping buf after line skip"
|
||||
);
|
||||
self.alt_buf ^= 1; // read into the other buf, so we can check for sameness
|
||||
self.display();
|
||||
Ok(())
|
||||
}
|
||||
|
||||
/// dump a data sources contents according to the set configs
|
||||
pub fn dump(&mut self, data: &mut dyn DataSource) -> Result<()> {
|
||||
// skip a given number of bytes
|
||||
if self.skip > 0 {
|
||||
|
@ -183,12 +206,15 @@ impl Hedu {
|
|||
self.display();
|
||||
Ok(())
|
||||
}
|
||||
|
||||
/// adjust the counters of the hedu struct, adding self.len
|
||||
#[inline]
|
||||
fn adjust_counters(&mut self) {
|
||||
self.rd_counter += self.len;
|
||||
self.data_idx += self.len;
|
||||
}
|
||||
|
||||
/// read data from the data source
|
||||
fn rd_data(&mut self, data: &mut dyn DataSource) -> Result<()> {
|
||||
match data.read(&mut self.buf[self.alt_buf]) {
|
||||
Ok(mut len) => {
|
||||
|
@ -198,33 +224,17 @@ impl Hedu {
|
|||
}
|
||||
self.len = len;
|
||||
self.adjust_counters();
|
||||
return Ok(());
|
||||
Ok(())
|
||||
}
|
||||
Err(err) => {
|
||||
error!("error while reading data: {err}");
|
||||
bail!(err)
|
||||
Err(err.into())
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
pub trait DataSource: Read {
|
||||
fn skip(&mut self, length: usize) -> std::io::Result<()>;
|
||||
}
|
||||
impl DataSource for std::io::Stdin {
|
||||
fn skip(&mut self, _length: usize) -> std::io::Result<()> {
|
||||
warn!("can't skip bytes for the stdin!");
|
||||
Ok(())
|
||||
}
|
||||
}
|
||||
impl DataSource for std::fs::File {
|
||||
fn skip(&mut self, length: usize) -> std::io::Result<()> {
|
||||
self.seek(SeekFrom::Current(length as i64))?;
|
||||
// returns the new position from the start, we don't need it here.
|
||||
Ok(())
|
||||
}
|
||||
}
|
||||
|
||||
/// interpret characters for the --chars option
|
||||
fn mask_chars(c: char) -> char {
|
||||
if c.is_ascii_graphic() {
|
||||
return c;
|
||||
|
|
|
@ -1,6 +1,6 @@
|
|||
//! # Executable for the hedu submodule
|
||||
//! # A simple hexdumper with a somewhat fancy format
|
||||
//!
|
||||
//! Dump data to a fancy format.
|
||||
//! Dump data from any readable source, such as the stdin or a file
|
||||
#![warn(clippy::pedantic)]
|
||||
|
||||
use std::{fs::File, io::IsTerminal, path::PathBuf};
|
||||
|
@ -23,7 +23,7 @@ use dumper::*;
|
|||
{usage-heading} {usage}
|
||||
{all-args}{tab}
|
||||
|
||||
autocrate: {version}
|
||||
{name}: {version}
|
||||
Author: {author-with-newline}
|
||||
"#
|
||||
)]
|
||||
|
|
Reference in a new issue