1 files changed, 9 insertions, 0 deletions
diff --git a/viewport.c b/viewport.c
index 0e9b9c5..8a18b44 100644
--- a/viewport.c
+++ b/viewport.c
@@ -20,6 +20,12 @@ viewport_memdraw(Viewport *v, Memimage *dst)
 	v->fbctl->memdraw(v->fbctl, dst);
 }
 
+static Framebuf *
+viewport_getfb(Viewport *v)
+{
+	return v->fbctl->getfb(v->fbctl);
+}
+
 Viewport *
 mkviewport(Rectangle r)
 {
@@ -32,12 +38,15 @@ mkviewport(Rectangle r)
 	v->fbctl = mkfbctl(r);
 	v->draw = viewport_draw;
 	v->memdraw = viewport_memdraw;
+	v->getfb = viewport_getfb;
 	return v;
 }
 
 void
 rmviewport(Viewport *v)
 {
+	if(v == nil)
+		return;
 	rmfbctl(v->fbctl);
 	free(v);
 }