jmb/scanfile
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

sqlc regeneration

Browse source
This commit is contained in:
Johannes Bülow 2025-06-15 21:13:56 +02:00
parent 197c6041d9
commit e5f95b905b
Signed by: jmb
GPG key ID: B56971CF7B8F83A6
4 changed files with 214 additions and 48 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

18
server/internal/sqlc/models.go
View file

@ -34,6 +34,24 @@ type FileProperty struct {
LibmagicApple pgtype.Text
}
type Msoffice struct {
ID int64
FileID pgtype.UUID
Verdict pgtype.Text
ContainerFormat pgtype.Text
Encrypted pgtype.Bool
FileFormat pgtype.Text
VbaMacros pgtype.Text
XlmMacros pgtype.Text
VbaStomping pgtype.Bool
NbAutoexec pgtype.Int4
NbIocs pgtype.Int4
NbMacros pgtype.Int4
NbSuspicious pgtype.Int4
OlevbaResults []byte
Macros []byte
}
type MsofficeMraptor struct {
ID int64
FileID pgtype.UUID

48
server/internal/sqlc/queries-file_properties.sql.go
View file

@ -46,54 +46,6 @@ func (q *Queries) InsertFileDIEC(ctx context.Context, arg InsertFileDIECParams)
return err
}
const insertFileMsofficeMraptor = `-- name: InsertFileMsofficeMraptor :exec
INSERT INTO msoffice_mraptor (
file_id, data
) VALUES ($1, $2)
`
type InsertFileMsofficeMraptorParams struct {
FileID pgtype.UUID
Data []byte
}
func (q *Queries) InsertFileMsofficeMraptor(ctx context.Context, arg InsertFileMsofficeMraptorParams) error {
_, err := q.db.Exec(ctx, insertFileMsofficeMraptor, arg.FileID, arg.Data)
return err
}
const insertFileMsofficeOleid = `-- name: InsertFileMsofficeOleid :exec
INSERT INTO msoffice_oleid (
file_id, data
) VALUES ($1, $2)
`
type InsertFileMsofficeOleidParams struct {
FileID pgtype.UUID
Data []byte
}
func (q *Queries) InsertFileMsofficeOleid(ctx context.Context, arg InsertFileMsofficeOleidParams) error {
_, err := q.db.Exec(ctx, insertFileMsofficeOleid, arg.FileID, arg.Data)
return err
}
const insertFileMsofficeOlevba = `-- name: InsertFileMsofficeOlevba :exec
INSERT INTO msoffice_olevba (
file_id, data
) VALUES ($1, $2)
`
type InsertFileMsofficeOlevbaParams struct {
FileID pgtype.UUID
Data []byte
}
func (q *Queries) InsertFileMsofficeOlevba(ctx context.Context, arg InsertFileMsofficeOlevbaParams) error {
_, err := q.db.Exec(ctx, insertFileMsofficeOlevba, arg.FileID, arg.Data)
return err
}
const insertFileProperties = `-- name: InsertFileProperties :exec
INSERT INTO file_properties (
id, sha256, md5, libmagic_mime, libmagic_extension, libmagic_apple

162
server/internal/sqlc/queries-msoffice.sql.go Normal file
View file

@ -0,0 +1,162 @@
// Code generated by sqlc. DO NOT EDIT.
// versions:
// sqlc v1.29.0
// source: queries-msoffice.sql
package sqlc
import (
"context"
"github.com/jackc/pgx/v5/pgtype"
)
const getMSOfficeData = `-- name: GetMSOfficeData :one
SELECT t1.file_id, t1.data AS oleid, t2.data AS olevba, t3.data AS mraptor
FROM msoffice_oleid as t1
LEFT join msoffice_olevba AS t2 ON t2.file_id = t1.file_id
LEFT JOIN msoffice_mraptor AS t3 ON t3.file_id = t1.file_id
WHERE t1.file_id = $1
`
type GetMSOfficeDataRow struct {
FileID pgtype.UUID
Oleid []byte
Olevba []byte
Mraptor []byte
}
func (q *Queries) GetMSOfficeData(ctx context.Context, fileID pgtype.UUID) (GetMSOfficeDataRow, error) {
row := q.db.QueryRow(ctx, getMSOfficeData, fileID)
var i GetMSOfficeDataRow
err := row.Scan(
&i.FileID,
&i.Oleid,
&i.Olevba,
&i.Mraptor,
)
return i, err
}
const getMSOfficeResults = `-- name: GetMSOfficeResults :one
SELECT id, file_id, verdict, container_format, encrypted, file_format, vba_macros, xlm_macros, vba_stomping, nb_autoexec, nb_iocs, nb_macros, nb_suspicious, olevba_results, macros FROM msoffice
WHERE file_id = $1
LIMIT 1
`
func (q *Queries) GetMSOfficeResults(ctx context.Context, fileID pgtype.UUID) (Msoffice, error) {
row := q.db.QueryRow(ctx, getMSOfficeResults, fileID)
var i Msoffice
err := row.Scan(
&i.ID,
&i.FileID,
&i.Verdict,
&i.ContainerFormat,
&i.Encrypted,
&i.FileFormat,
&i.VbaMacros,
&i.XlmMacros,
&i.VbaStomping,
&i.NbAutoexec,
&i.NbIocs,
&i.NbMacros,
&i.NbSuspicious,
&i.OlevbaResults,
&i.Macros,
)
return i, err
}
const insertFileMsofficeMraptor = `-- name: InsertFileMsofficeMraptor :exec
INSERT INTO msoffice_mraptor (
file_id, data
) VALUES ($1, $2)
`
type InsertFileMsofficeMraptorParams struct {
FileID pgtype.UUID
Data []byte
}
func (q *Queries) InsertFileMsofficeMraptor(ctx context.Context, arg InsertFileMsofficeMraptorParams) error {
_, err := q.db.Exec(ctx, insertFileMsofficeMraptor, arg.FileID, arg.Data)
return err
}
const insertFileMsofficeOleid = `-- name: InsertFileMsofficeOleid :exec
INSERT INTO msoffice_oleid (
file_id, data
) VALUES ($1, $2)
`
type InsertFileMsofficeOleidParams struct {
FileID pgtype.UUID
Data []byte
}
func (q *Queries) InsertFileMsofficeOleid(ctx context.Context, arg InsertFileMsofficeOleidParams) error {
_, err := q.db.Exec(ctx, insertFileMsofficeOleid, arg.FileID, arg.Data)
return err
}
const insertFileMsofficeOlevba = `-- name: InsertFileMsofficeOlevba :exec
INSERT INTO msoffice_olevba (
file_id, data
) VALUES ($1, $2)
`
type InsertFileMsofficeOlevbaParams struct {
FileID pgtype.UUID
Data []byte
}
func (q *Queries) InsertFileMsofficeOlevba(ctx context.Context, arg InsertFileMsofficeOlevbaParams) error {
_, err := q.db.Exec(ctx, insertFileMsofficeOlevba, arg.FileID, arg.Data)
return err
}
const insertMSOfficeResults = `-- name: InsertMSOfficeResults :exec
INSERT INTO msoffice (
file_id, verdict, container_format, encrypted, file_format, vba_macros, xlm_macros,
vba_stomping, nb_autoexec, nb_iocs, nb_macros, nb_suspicious, olevba_results, macros
) VALUES (
$1, $2, $3, $4, $5, $6, $7, $8, $9, $10, $11, $12, $13, $14
)
`
type InsertMSOfficeResultsParams struct {
FileID pgtype.UUID
Verdict pgtype.Text
ContainerFormat pgtype.Text
Encrypted pgtype.Bool
FileFormat pgtype.Text
VbaMacros pgtype.Text
XlmMacros pgtype.Text
VbaStomping pgtype.Bool
NbAutoexec pgtype.Int4
NbIocs pgtype.Int4
NbMacros pgtype.Int4
NbSuspicious pgtype.Int4
OlevbaResults []byte
Macros []byte
}
func (q *Queries) InsertMSOfficeResults(ctx context.Context, arg InsertMSOfficeResultsParams) error {
_, err := q.db.Exec(ctx, insertMSOfficeResults,
arg.FileID,
arg.Verdict,
arg.ContainerFormat,
arg.Encrypted,
arg.FileFormat,
arg.VbaMacros,
arg.XlmMacros,
arg.VbaStomping,
arg.NbAutoexec,
arg.NbIocs,
arg.NbMacros,
arg.NbSuspicious,
arg.OlevbaResults,
arg.Macros,
)
return err
}

34
server/internal/sqlc/queries-processing_jobs.sql.go
View file

@ -65,6 +65,40 @@ func (q *Queries) FinishProcessingJob(ctx context.Context, id int64) error {
return err
}
const getAllJobs = `-- name: GetAllJobs :many
SELECT id, file_id, created, started, completed, status, job_type, error, messages FROM processing_jobs
`
func (q *Queries) GetAllJobs(ctx context.Context) ([]ProcessingJob, error) {
rows, err := q.db.Query(ctx, getAllJobs)
if err != nil {
return nil, err
}
defer rows.Close()
var items []ProcessingJob
for rows.Next() {
var i ProcessingJob
if err := rows.Scan(
&i.ID,
&i.FileID,
&i.Created,
&i.Started,
&i.Completed,
&i.Status,
&i.JobType,
&i.Error,
&i.Messages,
); err != nil {
return nil, err
}
items = append(items, i)
}
if err := rows.Err(); err != nil {
return nil, err
}
return items, nil
}
const getJob = `-- name: GetJob :one
SELECT id, file_id, created, started, completed, status, job_type, error, messages FROM processing_jobs
WHERE id = $1