all: propagate context throughout the codebase.

Change-Id: Icdb6f5911e49e121014077067ac8c9a83b0ea8cd
Reviewed-on: https://go-review.googlesource.com/66112
Reviewed-by: Ross Light <light@google.com>