Add taint-tracking for text/tabwriter package.

ql/src/semmle/go/frameworks/Stdlib.qll

@@ -15,6 +15,7 @@ import semmle.go.frameworks.stdlib.CompressZlib
 import semmle.go.frameworks.stdlib.Path
 import semmle.go.frameworks.stdlib.PathFilepath
 import semmle.go.frameworks.stdlib.TextScanner
+import semmle.go.frameworks.stdlib.TextTabwriter
 
 /** A `String()` method. */
 class StringMethod extends TaintTracking::FunctionModel, Method {

ql/src/semmle/go/frameworks/stdlib/TextTabwriter.qll

@@ -0,0 +1,45 @@
+/**
+ * Provides classes modeling security-relevant aspects of the `text/tabwriter` package.
+ */
+
+import go
+
+/** Provides models of commonly used functions in the `text/tabwriter` package. */
+module TextTabwriter {
+  private class FunctionModels extends TaintTracking::FunctionModel {
+    FunctionInput inp;
+    FunctionOutput outp;
+
+    FunctionModels() {
+      // signature: func NewWriter(output io.Writer, minwidth int, tabwidth int, padding int, padchar byte, flags uint) *Writer
+      hasQualifiedName("text/tabwriter", "NewWriter") and
+      (inp.isResult() and outp.isParameter(0))
+    }
+
+    override predicate hasTaintFlow(FunctionInput input, FunctionOutput output) {
+      input = inp and output = outp
+    }
+  }
+
+  private class MethodModels extends TaintTracking::FunctionModel, Method {
+    FunctionInput inp;
+    FunctionOutput outp;
+
+    MethodModels() {
+      // signature: func (*Writer).Init(output io.Writer, minwidth int, tabwidth int, padding int, padchar byte, flags uint) *Writer
+      this.hasQualifiedName("text/tabwriter", "Writer", "Init") and
+      (
+        (inp.isReceiver() or inp.isResult()) and
+        outp.isParameter(0)
+      )
+      or
+      // signature: func (*Writer).Write(buf []byte) (n int, err error)
+      this.hasQualifiedName("text/tabwriter", "Writer", "Write") and
+      (inp.isParameter(0) and outp.isReceiver())
+    }
+
+    override predicate hasTaintFlow(FunctionInput input, FunctionOutput output) {
+      input = inp and output = outp
+    }
+  }
+}

ql/test/library-tests/semmle/go/frameworks/StdlibTaintFlow/TextTabwriter.go

@@ -0,0 +1,62 @@
+// Code generated by https://github.com/gagliardetto/codebox. DO NOT EDIT.
+
+package main
+
+import (
+	"io"
+	"text/tabwriter"
+)
+
+func TaintStepTest_TextTabwriterNewWriter_B0I0O0(sourceCQL interface{}) interface{} {
+	fromWriter656 := sourceCQL.(*tabwriter.Writer)
+	var intoWriter414 io.Writer
+	intermediateCQL := tabwriter.NewWriter(intoWriter414, 0, 0, 0, 0, 0)
+	link(fromWriter656, intermediateCQL)
+	return intoWriter414
+}
+
+func TaintStepTest_TextTabwriterWriterInit_B0I0O0(sourceCQL interface{}) interface{} {
+	fromWriter518 := sourceCQL.(tabwriter.Writer)
+	var intoWriter650 io.Writer
+	fromWriter518.Init(intoWriter650, 0, 0, 0, 0, 0)
+	return intoWriter650
+}
+
+func TaintStepTest_TextTabwriterWriterInit_B0I1O0(sourceCQL interface{}) interface{} {
+	fromWriter784 := sourceCQL.(*tabwriter.Writer)
+	var intoWriter957 io.Writer
+	var mediumObjCQL tabwriter.Writer
+	intermediateCQL := mediumObjCQL.Init(intoWriter957, 0, 0, 0, 0, 0)
+	link(fromWriter784, intermediateCQL)
+	return intoWriter957
+}
+
+func TaintStepTest_TextTabwriterWriterWrite_B0I0O0(sourceCQL interface{}) interface{} {
+	fromByte520 := sourceCQL.([]byte)
+	var intoWriter443 tabwriter.Writer
+	intoWriter443.Write(fromByte520)
+	return intoWriter443
+}
+
+func RunAllTaints_TextTabwriter() {
+	{
+		source := newSource(0)
+		out := TaintStepTest_TextTabwriterNewWriter_B0I0O0(source)
+		sink(0, out)
+	}
+	{
+		source := newSource(1)
+		out := TaintStepTest_TextTabwriterWriterInit_B0I0O0(source)
+		sink(1, out)
+	}
+	{
+		source := newSource(2)
+		out := TaintStepTest_TextTabwriterWriterInit_B0I1O0(source)
+		sink(2, out)
+	}
+	{
+		source := newSource(3)
+		out := TaintStepTest_TextTabwriterWriterWrite_B0I0O0(source)
+		sink(3, out)
+	}
+}