diff options
Diffstat (limited to 'pkg/csource/csource.go')
| -rw-r--r-- | pkg/csource/csource.go | 9 |
1 files changed, 7 insertions, 2 deletions
diff --git a/pkg/csource/csource.go b/pkg/csource/csource.go index 24717a31a..c30db6b9d 100644 --- a/pkg/csource/csource.go +++ b/pkg/csource/csource.go @@ -366,8 +366,13 @@ func (ctx *context) copyin(w *bytes.Buffer, csumSeq *int, copyin prog.ExecCopyin case prog.ExecArgResult: ctx.copyinVal(w, copyin.Addr, arg.Size, ctx.resultArgToStr(arg), arg.Format) case prog.ExecArgData: - fmt.Fprintf(w, "\tNONFAILING(memcpy((void*)0x%x, \"%s\", %v));\n", - copyin.Addr, toCString(arg.Data, arg.Readable), len(arg.Data)) + if bytes.Equal(arg.Data, bytes.Repeat(arg.Data[:1], len(arg.Data))) { + fmt.Fprintf(w, "\tNONFAILING(memset((void*)0x%x, %v, %v));\n", + copyin.Addr, arg.Data[0], len(arg.Data)) + } else { + fmt.Fprintf(w, "\tNONFAILING(memcpy((void*)0x%x, \"%s\", %v));\n", + copyin.Addr, toCString(arg.Data, arg.Readable), len(arg.Data)) + } case prog.ExecArgCsum: switch arg.Kind { case prog.ExecArgCsumInet: |