ReplyTracker.java

package net.i2p.i2ptunnel.socks;

import java.util.Map;

import net.i2p.data.Destination;
import net.i2p.i2ptunnel.udp.*;
import net.i2p.util.Log;

/**
 * Track who the reply goes to
 * @author zzz
 */
public class ReplyTracker<S extends Sink> implements Source, Sink {

    public ReplyTracker(S reply, Map<Destination, S> cache) {
        this.reply = reply;
        this.cache = cache;
    }
    
    public void setSink(Sink sink) {
        this.sink = sink;
    }

    public void start() {}

    /**
     *  May throw RuntimeException from underlying sink
     *  @throws RuntimeException
     */
    public void send(Destination to, byte[] data) {
        this.cache.put(to, this.reply);
        this.sink.send(to, data);
    }
    
    private S reply;
    private Map<Destination, S> cache;
    private Sink sink;
}