Many resources are needed to download a project. Please understand that we have to compensate our server costs. Thank you in advance. Project price only 1 $
You can buy this project and download/modify it how often you want.
/**
*
*/
package io.pkts.packet;
import io.pkts.buffer.Buffer;
import java.io.IOException;
import java.io.OutputStream;
/**
* @author [email protected]
*/
public final class IPPacketImpl implements IPPacket {
private final MACPacket parent;
private final Buffer headers;
private final int options;
// private final String sourceIp;
// private final String destinationIp;
/**
*
*/
public IPPacketImpl(final MACPacket parent, final Buffer headers, final int options) {
assert parent != null;
assert headers != null;
this.parent = parent;
this.headers = headers;
this.options = options;
// assert (sourceIp != null) && !sourceIp.isEmpty();
// assert (destinationIp != null) && !destinationIp.isEmpty();
// this.sourceIp = sourceIp;
// this.destinationIp = this.destinationIp;
}
@Override
public int getIpChecksum() {
return this.headers.getUnsignedShort(10);
}
/**
* Algorithm adopted from RFC 1071 - Computing the Internet Checksum
*
* @return
*/
private int calculateChecksum() {
long sum = 0;
for (int i = 0; i < this.headers.capacity() - 1; i += 2) {
if (i != 10) {
sum += this.headers.getUnsignedShort(i);
}
}
while (sum >> 16 != 0) {
sum = (sum & 0xffff) + (sum >> 16);
}
return (int) ~sum & 0xFFFF;
}
/**
* Get the raw source ip.
*
* Note, these are the raw bits and should be treated as such. If you really
* want to print it, then you should treat it as unsigned
*
* @return
*/
public int getRawSourceIp() {
return this.headers.getInt(12);
}
/**
*
* {@inheritDoc}
*/
@Override
public String getSourceIP() {
final short a = this.headers.getUnsignedByte(12);
final short b = this.headers.getUnsignedByte(13);
final short c = this.headers.getUnsignedByte(14);
final short d = this.headers.getUnsignedByte(15);
return a + "." + b + "." + c + "." + d;
}
/**
* Get the raw destination ip.
*
* Note, these are the raw bits and should be treated as such. If you really
* want to print it, then you should treat it as unsigned
*
* @return
*/
public int getRawDestinationIp() {
return this.headers.getInt(16);
}
/**
*
* {@inheritDoc}
*/
@Override
public String getDestinationIP() {
final short a = this.headers.getUnsignedByte(16);
final short b = this.headers.getUnsignedByte(17);
final short c = this.headers.getUnsignedByte(18);
final short d = this.headers.getUnsignedByte(19);
return a + "." + b + "." + c + "." + d;
}
/**
* {@inheritDoc}
*/
@Override
public void verify() {
// nothing to do for ip packets
}
@Override
public long getArrivalTime() {
return this.parent.getArrivalTime();
}
@Override
public String getSourceMacAddress() {
return this.parent.getSourceMacAddress();
}
@Override
public String getDestinationMacAddress() {
return this.parent.getDestinationMacAddress();
}
@Override
public void write(final OutputStream out) throws IOException {
throw new RuntimeException("Sorry, not implemented just yet.");
}
@Override
public int getTotalLength() {
// byte 3 - 4
return this.headers.getUnsignedShort(3);
}
/**
* {@inheritDoc}
*/
@Override
public void setSourceMacAddress(final String macAddress) {
this.parent.setSourceMacAddress(macAddress);
}
/**
* {@inheritDoc}
*/
@Override
public void setDestinationMacAddress(final String macAddress) {
this.parent.setDestinationMacAddress(macAddress);
}
@Override
public void setSourceIP(final int a, final int b, final int c, final int d) {
this.headers.setByte(12, (byte) a);
this.headers.setByte(13, (byte) b);
this.headers.setByte(14, (byte) c);
this.headers.setByte(15, (byte) d);
reCalculateChecksum();
}
@Override
public void setSourceIP(final byte a, final byte b, final byte c, final byte d) {
this.headers.setByte(12, a);
this.headers.setByte(13, b);
this.headers.setByte(14, c);
this.headers.setByte(15, d);
}
@Override
public void setDestinationIP(final int a, final int b, final int c, final int d) {
this.headers.setByte(16, (byte) a);
this.headers.setByte(17, (byte) b);
this.headers.setByte(18, (byte) c);
this.headers.setByte(19, (byte) d);
reCalculateChecksum();
}
@Override
public void setDestinationIP(final byte a, final byte b, final byte c, final byte d) {
this.headers.setByte(16, a);
this.headers.setByte(17, b);
this.headers.setByte(18, c);
this.headers.setByte(19, d);
}
@Override
public void setSourceIP(final String sourceIp) {
setIP(12, sourceIp);
}
@Override
public void setDestinationIP(final String destinationIP) {
setIP(16, destinationIP);
}
/**
* Very naive initial implementation. Should be changed to do a better job
* and its performance probably can go up a lot as well.
*
* @param startIndex
* @param address
*/
private void setIP(final int startIndex, final String address) {
final String[] parts = address.split("\\.");
this.headers.setByte(startIndex + 0, (byte) Integer.parseInt(parts[0]));
this.headers.setByte(startIndex + 1, (byte) Integer.parseInt(parts[1]));
this.headers.setByte(startIndex + 2, (byte) Integer.parseInt(parts[2]));
this.headers.setByte(startIndex + 3, (byte) Integer.parseInt(parts[3]));
reCalculateChecksum();
}
/**
* Whenever we change a value in the IP packet we need to update the
* checksum as well.
*/
@Override
public void reCalculateChecksum() {
final int checksum = calculateChecksum();
this.headers.setUnsignedShort(10, checksum);
}
@Override
public boolean verifyIpChecksum() {
return calculateChecksum() == getIpChecksum();
}
@Override
public IPPacket clone() {
final MACPacket mac = this.parent.clone();
final IPPacket pkt = new IPPacketImpl(mac, this.headers.clone(), this.options);
return pkt;
}
}
