sashakoshka/go-gemini
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

tofu: Protect HostWriter with a mutex

Browse Source
This commit is contained in:
Adnan Maolood 2021-01-14 16:35:54 -05:00
parent 9fe837ffac
commit da3e9ac0fe
1 changed files with 8 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
tofu/tofu.go
View File

@ -121,6 +121,7 @@ func (k *KnownHosts) TOFU(hostname string, cert *x509.Certificate) error {
// HostWriter writes host entries to an io.Writer. // HostWriter writes host entries to an io.Writer.
type HostWriter struct { type HostWriter struct {
bw *bufio.Writer bw *bufio.Writer
mu sync.Mutex
} }
// NewHostsWriter returns a new host writer that writes to the provided writer. // NewHostsWriter returns a new host writer that writes to the provided writer.
@ -131,11 +132,14 @@ func NewHostsWriter(w io.Writer) *HostWriter {
} }
// WriteHost writes the host to the underlying io.Writer. // WriteHost writes the host to the underlying io.Writer.
func (f *HostWriter) WriteHost(host Host) error { func (h *HostWriter) WriteHost(host Host) error {
f.bw.WriteString(host.String()) h.mu.Lock()
f.bw.WriteByte('\n') defer h.mu.Unlock()
if err := f.bw.Flush(); err != nil { h.bw.WriteString(host.String())
h.bw.WriteByte('\n')
if err := h.bw.Flush(); err != nil {
return fmt.Errorf("failed to write to hosts file: %w", err) return fmt.Errorf("failed to write to hosts file: %w", err)
} }
return nil return nil