nuclei/v2/pkg/js/libs/bytes/buffer.go
Tarun Koyalwar 4f93520e47
javascript protocol for scripting (includes 15+ proto libs) (#4109)
* rebase js-layer PR from @ice3man543

* package restructuring

* working

* fix duplicated event & matcher status

* fix lint error

* fix response field

* add new functions

* multiple minor improvements

* fix incorrect stats in js protocol

* sort output metadata in cli

* remove temp files

* remove dead code

* add unit and integration test

* fix lint error

* add jsdoclint using llm

* fix error in test

* add js lint using llm

* generate docs of libs

* llm lint

* remove duplicated docs

* update generated docs

* update prompt in doclint

* update docs

* temp disable version check test

* fix unit test and add retry

* fix panic in it

* update and move jsdocs

* updated jsdocs

* update docs

* update container platform in test

* dir restructure and adding docs

* add api_reference and remove markdown docs

* fix imports

* add javascript design and contribution docs

* add js protocol documentation

* update integration test and docs

* update doc ext mdx->md

* minor update to docs

* new integration test and more

* move go libs and add docs

* gen new net docs and more

* final docs update

* add new devtool

* use fastdialer

* fix build fail

* use fastdialer + network sandbox support

* add reserved keyword 'Port'

* update Port to new syntax

* misc update

* always enable templatectx in js protocol

* move docs to 'js-proto-docs' repo

* remove scrapefuncs binary

---------

Co-authored-by: sandeep <8293321+ehsandeep@users.noreply.github.com>
2023-09-16 16:02:17 +05:30

79 lines
1.8 KiB
Go

package bytes
import (
"encoding/hex"
"github.com/dop251/goja"
"github.com/projectdiscovery/nuclei/v2/pkg/js/libs/structs"
)
// Buffer is a minimal buffer implementation over a byte slice
// that is used to pack/unpack binary data in nuclei js integration.
type Buffer struct {
buf []byte
}
// NewBuffer creates a new buffer from a byte slice.
func NewBuffer(call goja.ConstructorCall) interface{} {
obj := &Buffer{}
obj.buf = make([]byte, 0)
return map[string]interface{}{
"Write": obj.Write,
"Pack": obj.Pack,
"Bytes": obj.Bytes,
"String": obj.String,
"Len": obj.Len,
"Hex": obj.Hex,
"Hexdump": obj.Hexdump,
}
}
// Write appends a byte slice to the buffer.
func (b *Buffer) Write(data []byte) *Buffer {
b.buf = append(b.buf, data...)
return b
}
// WriteString appends a string to the buffer.
func (b *Buffer) WriteString(data string) *Buffer {
b.buf = append(b.buf, []byte(data)...)
return b
}
// Bytes returns the byte slice of the buffer.
func (b *Buffer) Bytes() []byte {
return b.buf
}
// String returns the string representation of the buffer.
func (b *Buffer) String() string {
return string(b.buf)
}
// Len returns the length of the buffer.
func (b *Buffer) Len() int {
return len(b.buf)
}
// Hex returns the hex representation of the buffer.
func (b *Buffer) Hex() string {
return hex.EncodeToString(b.buf)
}
// Hexdump returns the hexdump representation of the buffer.
func (b *Buffer) Hexdump() string {
return hex.Dump(b.buf)
}
// Pack uses structs.Pack and packs given data and appends it to the buffer.
// it packs the data according to the given format.
func (b *Buffer) Pack(formatStr string, msg []interface{}) error {
bin, err := structs.Pack(formatStr, msg)
if err != nil {
return err
}
b.Write(bin)
return nil
}