Add a lock to guard manipulation of `StepStub`.

Change-Id: Icaadb1f5aea0cb7f266f24ed6756c4f6541768bd
Signed-off-by: Jay Guo <guojiannan1101@gmail.com>