IP.java

/*
 * Copyright (c) 2013 LDBC
 * Linked Data Benchmark Council (http://ldbc.eu)
 *
 * This file is part of ldbc_socialnet_dbgen.
 *
 * ldbc_socialnet_dbgen is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 * 
 * ldbc_socialnet_dbgen is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 * 
 * You should have received a copy of the GNU General Public License
 * along with ldbc_socialnet_dbgen.  If not, see <http://www.gnu.org/licenses/>.
 *
 * Copyright (C) 2011 OpenLink Software <bdsmt@openlinksw.com>
 * All Rights Reserved.
 *
 * This program is free software; you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation;  only Version 2 of the License dated
 * June 1991.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program; if not, write to the Free Software
 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301, USA.
 */
package ldbc.snb.datagen.objects;

import org.apache.hadoop.io.Writable;

import java.io.DataInput;
import java.io.DataOutput;
import java.io.IOException;

public class IP implements Writable {

    public static final int BYTE_MASK = 0xFF;
    public static final int BYTE_SIZE = 8;
    public static final int IP4_SIZE_BITS = 32;
    public static final int IP4_SIZE_BYTES = IP4_SIZE_BITS / 8;
    ;
    public static final int BYTE1_SHIFT_POSITION = 24;
    public static final int BYTE2_SHIFT_POSITION = 16;
    public static final int BYTE3_SHIFT_POSITION = 8;

    int ip;
    int mask;

    public IP () {

    }

    public IP(int byte1, int byte2, int byte3, int byte4, int networkMask) {
        ip = ((byte1 & BYTE_MASK) << BYTE1_SHIFT_POSITION) |
                ((byte2 & BYTE_MASK) << BYTE2_SHIFT_POSITION) |
                ((byte3 & BYTE_MASK) << BYTE3_SHIFT_POSITION) |
                (byte4 & BYTE_MASK);

        mask = (networkMask == IP4_SIZE_BITS) ? 0 : 1;
        for (int k = networkMask + 1; k < IP4_SIZE_BITS; k++) {
            mask = mask | mask << 1;
        }
    }

    public IP(IP i) {
        this.ip = i.ip;
        this.mask = i.mask;
    }

    public IP(int ip, int mask) {
        this.ip = ip;
        this.mask = mask;
    }

    public int getIp() {
        return ip;
    }

    public int getMask() {
        return mask;
    }

    public boolean belongsToMyNetwork(IP ip) {
        return (mask == ip.mask) && ((this.ip & ~this.mask) == (ip.ip & ~ip.mask));
    }

    public String toString() {
        return ((ip >>> BYTE1_SHIFT_POSITION) & BYTE_MASK) + "." +
                ((ip >>> BYTE2_SHIFT_POSITION) & BYTE_MASK) + "." +
                ((ip >>> BYTE3_SHIFT_POSITION) & BYTE_MASK) + "." +
                (ip & BYTE_MASK);
    }

    @Override
    public boolean equals(Object obj) {
        IP a = (IP) obj;
        return this.ip == a.ip && this.mask == a.mask;
    }

    public void copy( IP ip ) {
	    this.ip = ip.ip;
	    this.mask = ip.mask;
    }

    public void readFields(DataInput arg0) throws IOException {
       	ip = arg0.readInt(); 
        mask = arg0.readInt();
    }

    public void write(DataOutput arg0) throws IOException {
        arg0.writeInt(ip);
        arg0.writeInt(mask);
    }

}