irep_serialization.cpp

Go to the documentation of this file.

/*******************************************************************\
 
Module: binary irep conversions with hashing
 
Author: CM Wintersteiger
 
Date: May 2007
 
\*******************************************************************/

 
#include "irep_serialization.h"
 
#include "exception_utils.h"
#include "string_container.h"
 
#include <climits>
#include <iostream>
 
void irep_serializationt::write_irep(
  std::ostream &out,
  const irept &irep)
{
  write_string_ref(out, irep.id());
 
  for(const auto &sub_irep : irep.get_sub())
  {
    out.put('S');
    reference_convert(sub_irep, out);
  }
 
  for(const auto &sub_irep_entry : irep.get_named_sub())
  {
    out.put('N');
    write_string_ref(out, sub_irep_entry.first);
    reference_convert(sub_irep_entry.second, out);
  }
 
  out.put(0); // terminator
}
 
const irept &irep_serializationt::reference_convert(std::istream &in)
{
  std::size_t id=read_gb_word(in);
 
  if(
    id >= ireps_container.ireps_on_read.size() ||
    !ireps_container.ireps_on_read[id].first)
  {
    irept irep = read_irep(in);
 
    if(id >= ireps_container.ireps_on_read.size())
      ireps_container.ireps_on_read.resize(1 + id * 2, {false, get_nil_irep()});
 
    // guard against self-referencing ireps
    if(ireps_container.ireps_on_read[id].first)
      throw deserialization_exceptiont("irep id read twice.");
 
    ireps_container.ireps_on_read[id] = {true, std::move(irep)};
  }
 
  return ireps_container.ireps_on_read[id].second;
}
 
irept irep_serializationt::read_irep(std::istream &in)
{
  irep_idt id = read_string_ref(in);
  irept::subt sub;
  irept::named_subt named_sub;
 
  while(in.peek()=='S')
  {
    in.get();
    sub.push_back(reference_convert(in));
  }
 
  while(in.peek()=='N')
  {
    in.get();
    irep_idt id = read_string_ref(in);
    named_sub.emplace(id, reference_convert(in));
  }
 
  while(in.peek()=='C')
  {
    in.get();
    irep_idt id = read_string_ref(in);
    named_sub.emplace(id, reference_convert(in));
  }
 
  if(in.get()!=0)
  {
    throw deserialization_exceptiont("irep not terminated");
  }
 
  return irept(std::move(id), std::move(named_sub), std::move(sub));
}

void irep_serializationt::reference_convert(
  const irept &irep,
  std::ostream &out)
{
  std::size_t h=ireps_container.irep_full_hash_container.number(irep);
 
  const auto res = ireps_container.ireps_on_write.insert(
    {h, ireps_container.ireps_on_write.size()});
 
  write_gb_word(out, res.first->second);
  if(res.second)
    write_irep(out, irep);
}

void write_gb_word(std::ostream &out, std::size_t u)
{
 
  while(true)
  {
    unsigned char value=u&0x7f;
    u>>=7;
 
    if(u==0)
    {
      out.put(value);
      break;
    }
 
    out.put(value | 0x80);
  }
}

std::size_t irep_serializationt::read_gb_word(std::istream &in)
{
  std::size_t res=0;
 
  unsigned shift_distance=0;
 
  while(in.good())
  {
    if(shift_distance >= sizeof(res) * CHAR_BIT)
      throw deserialization_exceptiont("input number too large");
 
    unsigned char ch=static_cast<unsigned char>(in.get());
    res|=(size_t(ch&0x7f))<<shift_distance;
    shift_distance+=7;
    if((ch&0x80)==0)
      break;
  }
 
  if(!in.good())
    throw deserialization_exceptiont("unexpected end of input stream");
 
  return res;
}

void write_gb_string(std::ostream &out, const std::string &s)
{
  for(std::string::const_iterator it=s.begin();
      it!=s.end();
      ++it)
  {
    if(*it==0 || *it=='\\')
      out.put('\\'); // escape specials
    out << *it;
  }
 
  out.put(0);
}

irep_idt irep_serializationt::read_gb_string(std::istream &in)
{
  char c;
  size_t length=0;
 
  while((c=static_cast<char>(in.get()))!=0)
  {
    if(length>=read_buffer.size())
      read_buffer.resize(read_buffer.size()*2, 0);
 
    if(c=='\\') // escaped chars
      read_buffer[length]=static_cast<char>(in.get());
    else
      read_buffer[length]=c;
 
    length++;
  }
 
  return irep_idt(std::string(read_buffer.data(), length));
}

void irep_serializationt::write_string_ref(
  std::ostream &out,
  const irep_idt &s)
{
  size_t id = s.get_no();
  if(id>=ireps_container.string_map.size())
    ireps_container.string_map.resize(id+1, false);
 
  if(ireps_container.string_map[id])
    write_gb_word(out, id);
  else
  {
    ireps_container.string_map[id]=true;
    write_gb_word(out, id);
    write_gb_string(out, id2string(s));
  }
}

irep_idt irep_serializationt::read_string_ref(std::istream &in)
{
  std::size_t id=read_gb_word(in);
 
  if(id>=ireps_container.string_rev_map.size())
    ireps_container.string_rev_map.resize(1+id*2,
      std::pair<bool, irep_idt>(false, irep_idt()));
 
  if(!ireps_container.string_rev_map[id].first)
  {
    irep_idt s=read_gb_string(in);
    ireps_container.string_rev_map[id]=
      std::pair<bool, irep_idt>(true, s);
  }
 
  return ireps_container.string_rev_map[id].second;
}