@ruvector/edge-net 0.4.6 → 0.5.0

package/plugins/implementations/e2e-encryption.js

@@ -0,0 +1,160 @@
+/**
+ * End-to-End Encryption Plugin
+ *
+ * X25519 key exchange + ChaCha20-Poly1305 encryption.
+ * Provides forward secrecy with automatic key rotation.
+ *
+ * @module @ruvector/edge-net/plugins/e2e-encryption
+ */
+
+import { randomBytes, createCipheriv, createDecipheriv, createHash } from 'crypto';
+
+export class E2EEncryptionPlugin {
+    constructor(config = {}) {
+        this.config = {
+            keyRotationInterval: config.keyRotationInterval || 3600000, // 1 hour
+            forwardSecrecy: config.forwardSecrecy ?? true,
+        };
+
+        // Session keys (in production, use proper X25519)
+        this.sessionKeys = new Map(); // peerId -> { key, iv, createdAt }
+        this.rotationTimer = null;
+    }
+
+    async init() {
+        if (this.config.forwardSecrecy) {
+            this.rotationTimer = setInterval(
+                () => this._rotateKeys(),
+                this.config.keyRotationInterval
+            );
+        }
+    }
+
+    async destroy() {
+        if (this.rotationTimer) {
+            clearInterval(this.rotationTimer);
+        }
+    }
+
+    /**
+     * Establish encrypted session with peer
+     */
+    async establishSession(peerId, peerPublicKey) {
+        // In production: X25519 key exchange
+        // For demo: Generate shared secret from peer ID
+        const sharedSecret = createHash('sha256')
+            .update(peerId + '-' + Date.now())
+            .digest();
+
+        const sessionKey = {
+            key: sharedSecret,
+            iv: randomBytes(16),
+            createdAt: Date.now(),
+            messageCount: 0,
+        };
+
+        this.sessionKeys.set(peerId, sessionKey);
+
+        return {
+            sessionId: createHash('sha256').update(sharedSecret).digest('hex').slice(0, 16),
+            publicKey: randomBytes(32).toString('hex'), // Our ephemeral public key
+        };
+    }
+
+    /**
+     * Encrypt message for peer
+     */
+    encrypt(peerId, plaintext) {
+        const session = this.sessionKeys.get(peerId);
+        if (!session) {
+            throw new Error(`No session with peer: ${peerId}`);
+        }
+
+        // Use AES-256-GCM (ChaCha20-Poly1305 in production)
+        const iv = randomBytes(12);
+        const cipher = createCipheriv('aes-256-gcm', session.key, iv);
+
+        const data = typeof plaintext === 'string' ? plaintext : JSON.stringify(plaintext);
+        const encrypted = Buffer.concat([
+            cipher.update(data, 'utf8'),
+            cipher.final(),
+        ]);
+        const authTag = cipher.getAuthTag();
+
+        session.messageCount++;
+
+        return {
+            iv: iv.toString('base64'),
+            ciphertext: encrypted.toString('base64'),
+            authTag: authTag.toString('base64'),
+            messageNum: session.messageCount,
+        };
+    }
+
+    /**
+     * Decrypt message from peer
+     */
+    decrypt(peerId, encryptedMessage) {
+        const session = this.sessionKeys.get(peerId);
+        if (!session) {
+            throw new Error(`No session with peer: ${peerId}`);
+        }
+
+        const iv = Buffer.from(encryptedMessage.iv, 'base64');
+        const ciphertext = Buffer.from(encryptedMessage.ciphertext, 'base64');
+        const authTag = Buffer.from(encryptedMessage.authTag, 'base64');
+
+        const decipher = createDecipheriv('aes-256-gcm', session.key, iv);
+        decipher.setAuthTag(authTag);
+
+        const decrypted = Buffer.concat([
+            decipher.update(ciphertext),
+            decipher.final(),
+        ]);
+
+        return decrypted.toString('utf8');
+    }
+
+    /**
+     * Rotate session keys for forward secrecy
+     */
+    _rotateKeys() {
+        const now = Date.now();
+        for (const [peerId, session] of this.sessionKeys) {
+            if (now - session.createdAt > this.config.keyRotationInterval) {
+                // Generate new session key
+                const newKey = createHash('sha256')
+                    .update(session.key)
+                    .update(randomBytes(32))
+                    .digest();
+
+                session.key = newKey;
+                session.createdAt = now;
+                session.messageCount = 0;
+            }
+        }
+    }
+
+    /**
+     * Check if session exists
+     */
+    hasSession(peerId) {
+        return this.sessionKeys.has(peerId);
+    }
+
+    /**
+     * End session with peer
+     */
+    endSession(peerId) {
+        return this.sessionKeys.delete(peerId);
+    }
+
+    getStats() {
+        return {
+            activeSessions: this.sessionKeys.size,
+            rotationInterval: this.config.keyRotationInterval,
+        };
+    }
+}
+
+export default E2EEncryptionPlugin;

package/plugins/implementations/federated-learning.js

@@ -0,0 +1,249 @@
+/**
+ * Federated Learning Plugin
+ *
+ * Train ML models across nodes without sharing raw data.
+ * Implements FedAvg with differential privacy.
+ *
+ * @module @ruvector/edge-net/plugins/federated-learning
+ */
+
+import { EventEmitter } from 'events';
+import { randomBytes } from 'crypto';
+
+export class FederatedLearningPlugin extends EventEmitter {
+    constructor(config = {}) {
+        super();
+
+        this.config = {
+            aggregationStrategy: config.aggregationStrategy || 'fedavg',
+            localEpochs: config.localEpochs || 5,
+            differentialPrivacy: config.differentialPrivacy ?? true,
+            noiseMultiplier: config.noiseMultiplier || 1.0,
+            minParticipants: config.minParticipants || 3,
+            roundTimeout: config.roundTimeout || 60000,
+        };
+
+        this.rounds = new Map();         // roundId -> RoundState
+        this.localModels = new Map();    // modelId -> weights
+        this.globalModels = new Map();   // modelId -> aggregated weights
+    }
+
+    /**
+     * Start a new training round
+     */
+    startRound(modelId, globalWeights) {
+        const roundId = `round-${Date.now()}-${randomBytes(4).toString('hex')}`;
+
+        const round = {
+            id: roundId,
+            modelId,
+            globalWeights,
+            participants: new Map(),
+            status: 'collecting',
+            startedAt: Date.now(),
+        };
+
+        this.rounds.set(roundId, round);
+        this.emit('round:started', { roundId, modelId });
+
+        // Set timeout
+        setTimeout(() => {
+            if (round.status === 'collecting') {
+                this._aggregateRound(roundId);
+            }
+        }, this.config.roundTimeout);
+
+        return roundId;
+    }
+
+    /**
+     * Train locally and submit update
+     */
+    async trainLocal(roundId, localData, options = {}) {
+        const round = this.rounds.get(roundId);
+        if (!round) {
+            throw new Error(`Round not found: ${roundId}`);
+        }
+
+        // Simulate local training
+        const localUpdate = await this._performLocalTraining(
+            round.globalWeights,
+            localData,
+            options.epochs || this.config.localEpochs
+        );
+
+        // Add differential privacy noise if enabled
+        if (this.config.differentialPrivacy) {
+            this._addDifferentialPrivacy(localUpdate);
+        }
+
+        // Submit update
+        const participantId = options.participantId || randomBytes(8).toString('hex');
+        round.participants.set(participantId, {
+            update: localUpdate,
+            dataSize: localData.length,
+            submittedAt: Date.now(),
+        });
+
+        this.emit('update:submitted', { roundId, participantId });
+
+        // Check if we have enough participants
+        if (round.participants.size >= this.config.minParticipants) {
+            this._aggregateRound(roundId);
+        }
+
+        return { participantId, updateSize: localUpdate.length };
+    }
+
+    /**
+     * Perform local training (simulated)
+     */
+    async _performLocalTraining(globalWeights, localData, epochs) {
+        // In production: Use ONNX Runtime or TensorFlow.js
+        // For demo: Simulate gradient descent
+        const weights = globalWeights ? [...globalWeights] : Array(10).fill(0);
+
+        for (let epoch = 0; epoch < epochs; epoch++) {
+            for (const sample of localData) {
+                // Simplified SGD update
+                for (let i = 0; i < weights.length; i++) {
+                    const gradient = (sample.features?.[i] || Math.random()) * 0.01;
+                    weights[i] -= gradient;
+                }
+            }
+        }
+
+        return weights;
+    }
+
+    /**
+     * Add differential privacy noise
+     */
+    _addDifferentialPrivacy(weights) {
+        const sigma = this.config.noiseMultiplier;
+        for (let i = 0; i < weights.length; i++) {
+            // Gaussian noise
+            const u1 = Math.random();
+            const u2 = Math.random();
+            const noise = Math.sqrt(-2 * Math.log(u1)) * Math.cos(2 * Math.PI * u2);
+            weights[i] += noise * sigma;
+        }
+    }
+
+    /**
+     * Aggregate updates using FedAvg
+     */
+    _aggregateRound(roundId) {
+        const round = this.rounds.get(roundId);
+        if (!round || round.status !== 'collecting') {
+            return;
+        }
+
+        round.status = 'aggregating';
+
+        const updates = Array.from(round.participants.values());
+        if (updates.length === 0) {
+            round.status = 'failed';
+            this.emit('round:failed', { roundId, reason: 'No updates' });
+            return;
+        }
+
+        let aggregatedWeights;
+
+        switch (this.config.aggregationStrategy) {
+            case 'fedavg':
+                aggregatedWeights = this._fedAvg(updates);
+                break;
+            case 'fedprox':
+                aggregatedWeights = this._fedProx(updates, round.globalWeights);
+                break;
+            default:
+                aggregatedWeights = this._fedAvg(updates);
+        }
+
+        // Store aggregated model
+        this.globalModels.set(round.modelId, aggregatedWeights);
+        round.status = 'completed';
+        round.aggregatedWeights = aggregatedWeights;
+        round.completedAt = Date.now();
+
+        this.emit('round:completed', {
+            roundId,
+            modelId: round.modelId,
+            participants: round.participants.size,
+            duration: round.completedAt - round.startedAt,
+        });
+
+        return aggregatedWeights;
+    }
+
+    /**
+     * FedAvg aggregation
+     */
+    _fedAvg(updates) {
+        if (updates.length === 0) return null;
+
+        const totalSamples = updates.reduce((sum, u) => sum + u.dataSize, 0);
+        const numWeights = updates[0].update.length;
+        const aggregated = Array(numWeights).fill(0);
+
+        for (const { update, dataSize } of updates) {
+            const weight = dataSize / totalSamples;
+            for (let i = 0; i < numWeights; i++) {
+                aggregated[i] += update[i] * weight;
+            }
+        }
+
+        return aggregated;
+    }
+
+    /**
+     * FedProx aggregation (with proximal term)
+     */
+    _fedProx(updates, globalWeights) {
+        const fedAvgResult = this._fedAvg(updates);
+        if (!globalWeights) return fedAvgResult;
+
+        // Add proximal regularization
+        const mu = 0.01; // Proximal strength
+        for (let i = 0; i < fedAvgResult.length; i++) {
+            fedAvgResult[i] = (1 - mu) * fedAvgResult[i] + mu * globalWeights[i];
+        }
+
+        return fedAvgResult;
+    }
+
+    /**
+     * Get current global model
+     */
+    getGlobalModel(modelId) {
+        return this.globalModels.get(modelId);
+    }
+
+    /**
+     * Get round status
+     */
+    getRoundStatus(roundId) {
+        const round = this.rounds.get(roundId);
+        if (!round) return null;
+
+        return {
+            id: round.id,
+            modelId: round.modelId,
+            status: round.status,
+            participants: round.participants.size,
+            startedAt: round.startedAt,
+            completedAt: round.completedAt,
+        };
+    }
+
+    getStats() {
+        return {
+            totalRounds: this.rounds.size,
+            globalModels: this.globalModels.size,
+            config: this.config,
+        };
+    }
+}
+
+export default FederatedLearningPlugin;

package/plugins/implementations/reputation-staking.js

@@ -0,0 +1,243 @@
+/**
+ * Reputation Staking Plugin
+ *
+ * Stake credits as collateral for good behavior.
+ * Slashing mechanism for misbehavior detection.
+ *
+ * @module @ruvector/edge-net/plugins/reputation-staking
+ */
+
+import { EventEmitter } from 'events';
+
+export class ReputationStakingPlugin extends EventEmitter {
+    constructor(config = {}) {
+        super();
+
+        this.config = {
+            minStake: config.minStake || 10,
+            slashRate: config.slashRate || 0.1, // 10% slash
+            unbondingPeriod: config.unbondingPeriod || 604800000, // 7 days
+            maxSlashPerPeriod: config.maxSlashPerPeriod || 0.5, // Max 50% slash per period
+        };
+
+        // Staking state
+        this.stakes = new Map();          // nodeId -> { staked, reputation, unbonding }
+        this.slashHistory = new Map();    // nodeId -> [{ reason, amount, timestamp }]
+        this.unbondingQueue = [];         // [{ nodeId, amount, availableAt }]
+    }
+
+    /**
+     * Stake credits
+     */
+    stake(nodeId, amount, creditSystem) {
+        if (amount < this.config.minStake) {
+            throw new Error(`Minimum stake is ${this.config.minStake}`);
+        }
+
+        // Check balance
+        const balance = creditSystem.getBalance(nodeId);
+        if (balance < amount) {
+            throw new Error(`Insufficient balance: ${balance} < ${amount}`);
+        }
+
+        // Lock credits
+        creditSystem.spendCredits(nodeId, amount, `stake-${Date.now()}`);
+
+        // Create or update stake
+        let stake = this.stakes.get(nodeId);
+        if (!stake) {
+            stake = {
+                staked: 0,
+                reputation: 100, // Start at 100
+                unbonding: 0,
+                lastActivity: Date.now(),
+                successfulTasks: 0,
+                failedTasks: 0,
+            };
+            this.stakes.set(nodeId, stake);
+        }
+
+        stake.staked += amount;
+        stake.lastActivity = Date.now();
+
+        this.emit('staked', { nodeId, amount, totalStaked: stake.staked });
+
+        return stake;
+    }
+
+    /**
+     * Request unstaking (enters unbonding period)
+     */
+    unstake(nodeId, amount) {
+        const stake = this.stakes.get(nodeId);
+        if (!stake) {
+            throw new Error(`No stake found for: ${nodeId}`);
+        }
+
+        if (amount > stake.staked) {
+            throw new Error(`Cannot unstake more than staked: ${stake.staked}`);
+        }
+
+        stake.staked -= amount;
+        stake.unbonding += amount;
+
+        const availableAt = Date.now() + this.config.unbondingPeriod;
+        this.unbondingQueue.push({ nodeId, amount, availableAt });
+
+        this.emit('unstaking', { nodeId, amount, availableAt });
+
+        return { unbonding: stake.unbonding, availableAt };
+    }
+
+    /**
+     * Claim unbonded stake
+     */
+    claim(nodeId, creditSystem) {
+        const now = Date.now();
+        const stake = this.stakes.get(nodeId);
+        if (!stake) {
+            throw new Error(`No stake found for: ${nodeId}`);
+        }
+
+        let claimed = 0;
+        this.unbondingQueue = this.unbondingQueue.filter(item => {
+            if (item.nodeId === nodeId && item.availableAt <= now) {
+                claimed += item.amount;
+                stake.unbonding -= item.amount;
+                return false;
+            }
+            return true;
+        });
+
+        if (claimed > 0) {
+            creditSystem.earnCredits(nodeId, claimed, `unstake-claim-${Date.now()}`);
+            this.emit('claimed', { nodeId, amount: claimed });
+        }
+
+        return { claimed };
+    }
+
+    /**
+     * Slash stake for misbehavior
+     */
+    slash(nodeId, reason, severity = 1.0) {
+        const stake = this.stakes.get(nodeId);
+        if (!stake || stake.staked === 0) {
+            return { slashed: 0, reason: 'No stake to slash' };
+        }
+
+        // Calculate slash amount
+        const slashAmount = Math.min(
+            stake.staked * this.config.slashRate * severity,
+            stake.staked * this.config.maxSlashPerPeriod
+        );
+
+        stake.staked -= slashAmount;
+        stake.reputation = Math.max(0, stake.reputation - 10 * severity);
+        stake.failedTasks++;
+
+        // Record slash history
+        if (!this.slashHistory.has(nodeId)) {
+            this.slashHistory.set(nodeId, []);
+        }
+        this.slashHistory.get(nodeId).push({
+            reason,
+            amount: slashAmount,
+            severity,
+            timestamp: Date.now(),
+        });
+
+        this.emit('slashed', { nodeId, amount: slashAmount, reason, newReputation: stake.reputation });
+
+        return { slashed: slashAmount, newStake: stake.staked, newReputation: stake.reputation };
+    }
+
+    /**
+     * Record successful task (increases reputation)
+     */
+    recordSuccess(nodeId) {
+        const stake = this.stakes.get(nodeId);
+        if (!stake) return;
+
+        stake.successfulTasks++;
+        stake.reputation = Math.min(100, stake.reputation + 1);
+        stake.lastActivity = Date.now();
+
+        this.emit('success', { nodeId, reputation: stake.reputation });
+    }
+
+    /**
+     * Record failed task (may trigger slash)
+     */
+    recordFailure(nodeId, reason) {
+        const stake = this.stakes.get(nodeId);
+        if (!stake) return;
+
+        stake.failedTasks++;
+        stake.lastActivity = Date.now();
+
+        // Calculate failure rate
+        const totalTasks = stake.successfulTasks + stake.failedTasks;
+        const failureRate = stake.failedTasks / totalTasks;
+
+        // Slash if failure rate too high
+        if (failureRate > 0.3 && totalTasks >= 10) {
+            this.slash(nodeId, reason, failureRate);
+        } else {
+            // Just reduce reputation
+            stake.reputation = Math.max(0, stake.reputation - 5);
+        }
+    }
+
+    /**
+     * Get stake info
+     */
+    getStake(nodeId) {
+        return this.stakes.get(nodeId);
+    }
+
+    /**
+     * Get reputation score
+     */
+    getReputation(nodeId) {
+        const stake = this.stakes.get(nodeId);
+        return stake?.reputation ?? 0;
+    }
+
+    /**
+     * Get leaderboard
+     */
+    getLeaderboard(limit = 10) {
+        return Array.from(this.stakes.entries())
+            .map(([nodeId, stake]) => ({
+                nodeId,
+                staked: stake.staked,
+                reputation: stake.reputation,
+                successRate: stake.successfulTasks / (stake.successfulTasks + stake.failedTasks || 1),
+            }))
+            .sort((a, b) => b.reputation - a.reputation || b.staked - a.staked)
+            .slice(0, limit);
+    }
+
+    /**
+     * Check if node is eligible for tasks
+     */
+    isEligible(nodeId, minReputation = 50, minStake = 0) {
+        const stake = this.stakes.get(nodeId);
+        if (!stake) return false;
+
+        return stake.reputation >= minReputation && stake.staked >= minStake;
+    }
+
+    getStats() {
+        const stakes = Array.from(this.stakes.values());
+        return {
+            totalStaked: stakes.reduce((sum, s) => sum + s.staked, 0),
+            totalUnbonding: stakes.reduce((sum, s) => sum + s.unbonding, 0),
+            stakerCount: this.stakes.size,
+            averageReputation: stakes.reduce((sum, s) => sum + s.reputation, 0) / (stakes.length || 1),
+        };
+    }
+}
+
+export default ReputationStakingPlugin;