C++: streambuf

streambuf定义了对流缓冲区操作的接口,主要分为输出缓冲流和输入缓冲流。streambuf有两种用法,一是直接使用各个接口,二是继承并实现新的I/O channels。

1. 输出缓冲流:
  • pbase(): put base,输出缓冲流的首指针。
  • pptr(): put pointer,当前可写位置
  • epptr(): end put pointer,输出缓冲流的尾指针+1,即one past the last character。

函数:
setp(char new_pbase, char new_epptr)**
注:设置pbase,pptr,epptr三个指针的位置:

  • pbase()=new_pbase;
  • pptr()=bew_pbase;
  • epptr()=new_epptr;

pbump(int offset)
注: 移动pptr offset个位置。当我们写入num个数据,pbump(num)。

overflow
当缓冲区满时的操作;
pptr() == epptr()

2. 输入缓冲流:
  • eback(): 指向输入缓冲流的首指针
  • gptr(): 当前输入缓冲流的位置
  • egptr(): 输入缓冲流的末尾
  • gbump(): 移动gptr
  • setg(): 设置eback/gptr/egptr的位置

underflow()
当无缓冲区数据可读时。
cond: gptr()=egptr()

setg
setg (eback_base,new_gptr, new_egptr)
分别设置:

  • eback
  • gptr
  • egptr
3. 基于streambuf自定义一个输入输出缓冲流

prebufferstream.hpp:

/*  Author: zhangwen1@unionpay.com */
/*  Date: 2020-11-24               */

/*
 * Provide Custom Streambufeer Example 
*/ 

#ifndef PREBUFFERSTREAM_HPP
#define PREBUFFERSTREAM_HPP

#include <streambuf>
#include <cstdio>
#include <exception>

class prebuffer: public std::streambuf{

public:
  typedef std::char_traits<char> traits_ty;
  typedef traits_ty::int_type int_type;
  typedef traits_ty::pos_type pos_type;
  typedef traits_ty::off_type off_type;

private:
  int m_mode;
  int m_filedes;

  static const int s_buf_size = 4092;
  static const int s_pback_size = 4;
  char buffer[s_buf_size];

  void init(int fd, int om, bool throw_exception);

  int flushoutput();

public:

  prebuffer(int fd, int om, bool throw_exception=false);
  prebuffer(const prebuffer&)=delete;
  prebuffer& operator=(const prebuffer&)=delete;
  prebuffer(prebuffer&&)=delete;
  prebuffer& operator=(prebuffer&&)=delete;

  ~prebuffer(){close();}

  virtual int_type overflow(int_type c) override;
  virtual int_type underflow() override;

  bool is_open() const { return m_filedes >= 0; }

  void close();
  virtual int sync() override;

  std::string get_str(){
    return std::string(buffer,this->pptr()-buffer);
  }

};


class FileError: public std::exception{
  const char* what(void) const throw(){
    return "Invaild fd";
  }
};

#endif

prebufferstream.cc:

#include "prebufferstream.hpp"

#include <iostream>
#include <unistd.h>
#include <cstring>

void prebuffer::init(int fd, int om, bool throw_exception){
  m_filedes = fd;
  if (m_filedes > 0){
    m_mode = om;
  }
  if (throw_exception && !is_open()){
      throw FileError();
  }
  setg(buffer+s_pback_size,
       buffer+s_pback_size,
       buffer+s_pback_size);
  setp(buffer,
       buffer+s_buf_size-1);
}

prebuffer::prebuffer(int fd, int om, bool throw_exception){
  init(fd,om,throw_exception);
}

void prebuffer::close(){
  if (is_open()){
    this->sync();
    m_mode = 0;
    m_filedes = -1;
  }
}

prebuffer::int_type prebuffer::underflow(){

  if (gptr() < egptr()){
    return static_cast<unsigned char>(*gptr());
  }
  int num_putback=0;
  if (s_pback_size > 0){
    num_putback = gptr()-eback();
    if (num_putback > s_pback_size){
      num_putback = s_pback_size;
    }
    std::memcpy(buffer+s_pback_size-num_putback,gptr()-num_putback,num_putback);
  }

  const int num = ::read(m_filedes,buffer+s_pback_size,s_buf_size-s_pback_size);

  if (num <= 0){return EOF;}
  setg(buffer+s_pback_size-num_putback,buffer+s_pback_size,buffer+s_pback_size+num);

  return static_cast<unsigned char>(*gptr());

}

prebuffer::int_type prebuffer::overflow(int_type c){
  std::cout<<"Overflow Happend"<<std::endl;
  if (!(m_mode & std::ios::out) || !is_open()) return EOF;
  if (c != EOF){
    *pptr() = c;
    pbump(1);
  }
  if (flushoutput() == EOF)
    {
      return -1; 
    }

  return c;
}

int prebuffer::flushoutput(){
  if (!(m_mode & std::ios::out) || !is_open()) return EOF;
  size_t num = pptr()-pbase();
  if (::write(m_filedes,buffer,num) == -1){
    return EOF;
  }
  pbump(-num);
  return num;
}

int prebuffer::sync(){
  if (flushoutput() == EOF){
    return -1;
  }
  return 0;
}

测试, main.cc:

#include "prebufferstream.hpp"
#include <fcntl.h> 
#include <ostream>
#include <istream>
#include <unistd.h>
#include <iostream>

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

  std::string file_name1 = "test";
  std::string file_name2 = "intest";
  int fdout;
  int fdin;
  if ((fdout=open(file_name1.c_str(),O_RDWR | O_CREAT | O_APPEND)) < 0){
    std::cout<<"ERROR Open File"<<std::endl;
  }
  if ((fdin=open(file_name2.c_str(),O_RDONLY | O_CREAT | O_APPEND)) < 0){
    std::cout<<"ERROR Open File"<<std::endl;
  }


  prebuffer ot{fdout,std::ios::out};
  prebuffer it{fdin,std::ios::in};
  std::ostream dout(&ot);
  std::istream dint(&it);
  dout<<1<<2<<3<<4<<5<<"\n";

  std::cout<<ot.get_str()<<std::endl;

  ot.sync();
  int line;
  while (dint>>line){
    std::cout<<line<<" ";
  }

  close(fdout);
  close(fdin);

  return 0;
}
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 199,519评论 5 468
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 83,842评论 2 376
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 146,544评论 0 330
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 53,742评论 1 271
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 62,646评论 5 359
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 48,027评论 1 275
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 37,513评论 3 390
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 36,169评论 0 254
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 40,324评论 1 294
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 35,268评论 2 317
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 37,299评论 1 329
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 32,996评论 3 315
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 38,591评论 3 303
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 29,667评论 0 19
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 30,911评论 1 255
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 42,288评论 2 345
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 41,871评论 2 341

推荐阅读更多精彩内容