EscapingPrint.h

Go to the documentation of this file.

#pragma once
#include "Print.h"
 
namespace tiny_dlna {
 
struct EscapingPrint : public Print {
  EscapingPrint(Print& d) : dest(d) {}
  
  size_t write(uint8_t c) override {
    // Always return the logical expanded length, regardless of what dest.write() returns
    switch (c) {
      case '&':
        assert(dest.print("&")==5);
        return 5;
      case '<':
        assert(dest.print("&lt;")==4);
        return 4;
      case '>':
        assert(dest.print("&gt;")==4);
        return 4;
      case '"':
        assert(dest.print("&quot;")==6);
        return 6;
      case '\'':
        assert(dest.print("&apos;")==6);
        return 6;
      default:
        assert(dest.write(&c, 1)==1);
        return 1; 
    }
  }
  
  size_t write(const uint8_t* buffer, size_t size) override {
    size_t r = 0;
    for (size_t i = 0; i < size; ++i) {
      r += write(buffer[i]);
    }
    return r;
  }
  
 protected:
  Print& dest;
};
 
}  // namespace tiny_dlna