Subversion Repositories openarty

////////////////////////////////////////////////////////////////////////////////
//
// Filename: 	manping.cpp
//
// Project:	OpenArty, an entirely open SoC based upon the Arty platform
//
// Purpose:	To command the network to ping a target.
//
//
// Creator:	Dan Gisselquist, Ph.D.
//		Gisselquist Technology, LLC
//
////////////////////////////////////////////////////////////////////////////////
//
// Copyright (C) 2015-2016, Gisselquist Technology, LLC
//
// This program is free software (firmware): 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.
//
// This program is distributed in the hope that it will be useful, but WITHOUT
// ANY WARRANTY; without even the implied warranty of MERCHANTIBILITY 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.  (It's in the $(ROOT)/doc directory, run make with no
// target there if the PDF file isn't present.)  If not, see
// <http://www.gnu.org/licenses/> for a copy.
//
// License:	GPL, v3, as defined and found on www.gnu.org,
//		http://www.gnu.org/licenses/gpl.html
//
//
////////////////////////////////////////////////////////////////////////////////
//
//
#include <stdio.h>
#include <stdlib.h>
#include <unistd.h>
#include <strings.h>
#include <ctype.h>
#include <string.h>
#include <signal.h>
#include <assert.h>
 
#include "port.h"
#include "regdefs.h"
 
#define	TXGO		0x04000
#define	NOHWCRC		0x08000
#define	NOHWMAC		0x10000
#define	NETRESET	0x20000
 
//
// Define DONT_INVERT for debugging only, as it will break the interface
// test
//
// #define	DONT_INVERT
 
 
FPGA	*m_fpga;
void	closeup(int v) {
	m_fpga->kill();
	exit(0);
}
 
void	usage(void) {
	printf("USAGE: manping EN:RX:xx:xx:xx:xx AR:TY:EN:TX:xx:xx de.st.ip.x ar.ty.ip.x\n");
}
 
bool	strtoenetaddr(char *s, unsigned char *addr) {
	char	*p, *c;
 
	p = s;
	addr[0] = (unsigned char)(strtoul(p, NULL, 16)&0x0ff);
	c = strchr(p,':');
	if((!c) || ((c-p)>=3))
		return false;
 
	p = c+1;
	addr[1] = (unsigned char)(strtoul(p, NULL, 16)&0x0ff);
	c = strchr(p,':');
	if((!c) || ((c-p)>=3))
		return false;
 
	p = c+1;
	addr[2] = (unsigned char)(strtoul(p, NULL, 16)&0x0ff);
	c = strchr(p,':');
	if((!c) || ((c-p)>=3))
		return false;
 
	p = c+1;
	addr[3] = (unsigned char)(strtoul(p, NULL, 16)&0x0ff);
	c = strchr(p,':');
	if((!c) || ((c-p)>=3))
		return false;
 
	p = c+1;
	addr[4] = (unsigned char)(strtoul(p, NULL, 16)&0x0ff);
	c = strchr(p,':');
	if((!c) || ((c-p)>=3))
		return false;
 
	p = c+1;
	addr[5] = (unsigned char)(strtoul(p, NULL, 16)&0x0ff);
 
	return true;
}
 
bool	strtoinetaddr(char *s, unsigned char *addr) {
	char	*p, *c;
 
	p = s;
	addr[0] = (unsigned char)(strtoul(p, NULL, 10)&0x0ff);
	c = strchr(p,'.');
	if((!c) || ((c-p)>3))
		return false;
 
	p = c+1;
	addr[1] = (unsigned char)(strtoul(p, NULL, 10)&0x0ff);
	c = strchr(p,'.');
	if((!c) || ((c-p)>3))
		return false;
 
	p = c+1;
	addr[2] = (unsigned char)(strtoul(p, NULL, 10)&0x0ff);
	c = strchr(p,'.');
	if((!c) || ((c-p)>3))
		return false;
 
	p = c+1;
	addr[3] = (unsigned char)(strtoul(p, NULL, 10)&0x0ff);
 
	return true;
}
 
unsigned	calccrc(const int bytelen, const unsigned *buf) {
	const unsigned int	taps = 0xedb88320u;
#ifdef	DONT_INVERT
	unsigned int	crc = 0;
#else
	unsigned int	crc = 0xffffffff; // initial value
#endif
	int	bidx;
	int	bp = 0;
 
	for(bidx = 0; bidx<bytelen; bidx++) {
		if (bidx == 14)
			bidx+=2;
		unsigned char	byte = buf[(bidx>>2)]>>(24-((bidx&3)<<3));
 
		// printf("CRC[%2d]: %02x ([%2d]0x%08x)\n", bidx, byte, (bidx>>2), buf[(bidx>>2)]);
 
		for(int bit=8; --bit>= 0; byte >>= 1) {
			if ((crc ^ byte) & 1) {
				crc >>= 1;
				crc ^= taps;
			} else
				crc >>= 1;
		} bp++;
	}
#ifndef	DONT_INVERT
	crc ^= 0xffffffff;
#endif
	// Now, we need to reverse these bytes
	// ABCD
	unsigned a,b,c,d;
	a = (crc>>24); // &0x0ff;
	b = (crc>>16)&0x0ff;
	c = (crc>> 8)&0x0ff;
	d = crc; // (crc    )&0x0ff;
	crc = (d<<24)|(c<<16)|(b<<8)|a;
 
	// printf("%d bytes processed\n", bp);
	return crc;
}
 
void	ipchecksum(unsigned *packet) {
	int npkt = (packet[0]>>24)&0x0f;
	unsigned checksum = 0;
 
	packet[2] &= 0xffff0000;
	printf("PKT[2] set to %08x\n", packet[2]);
	printf("checksum = %08x\n", checksum);
	for(int i=0; i<npkt; i++)
		checksum += packet[i] & 0x0ffff;
	printf("checksum = %08x\n", checksum);
	for(int i=0; i<npkt; i++)
		checksum += (packet[i]>>16)&0x0ffff;
	printf("checksum = %08x\n", checksum);
	checksum = (checksum & 0x0ffff) + (checksum >> 16);
	checksum = (checksum & 0x0ffff) + (checksum >> 16);
	packet[2] |= (checksum & 0x0ffff)^0x0ffff;
 
	printf("PKT[2] set to 0x%08x\n", packet[2]);
	checksum = 0;
	for(int i=0; i<npkt; i++)
		checksum += packet[i] & 0x0ffff;
	for(int i=0; i<npkt; i++)
		checksum += (packet[i]>>16)&0x0ffff;
	checksum = (checksum & 0x0ffff) + (checksum >> 16);
	checksum = (checksum & 0x0ffff) + (checksum >> 16);
	checksum ^= 0x0ffff;
 
	assert(checksum == 0);
}
 
void	clear_scope(FPGA *fpga) {
	unsigned	scopev;
 
	scopev = m_fpga->readio(R_NETSCOPE);
	int delay = (scopev>>20)&0x0f;
	delay = (1<<(delay))-32;
	m_fpga->writeio(R_NETSCOPE, (delay));
}
 
int main(int argc, char **argv) {
	int	skp=0, port = FPGAPORT;
	bool	config_hw_mac = true, config_hw_crc = true;
	FPGA::BUSW	txstat;
	int	argn;
	unsigned	checksum;
	unsigned	urand[16], nu = 0;
 
	{
		FILE *fp;
		for(int i=0; i<16; i++)
			urand[i] = rand();
 
		// Now, see if we can do better than the library random
		// number generator--but don't fail if we can't.
		fp = fopen("/dev/urandom", "r");
		if (fp != NULL) {
			int nr = fread(urand, sizeof(short), 16, fp);
			fclose(fp);
		}
	}
 
 
	FPGAOPEN(m_fpga);
 
	signal(SIGSTOP, closeup);
	signal(SIGHUP, closeup);
 
	txstat = m_fpga->readio(R_NET_TXCMD);
 
	// Take the ethernet out of reset
	if ((txstat & NETRESET) != 0)
		m_fpga->writeio(R_NET_TXCMD, (txstat &=(~NETRESET)));
 
	unsigned	packet[14];
 
	unsigned char	smac[6], dmac[6];
	unsigned char	sip[4],  dip[4];
 
	// I know the ethernet MAC of the computer I wish to test with
	dmac[0] = 0xc8; dmac[1] = 0x3a; dmac[2] = 0x35;
	dmac[3] = 0xd2; dmac[4] = 0x07; dmac[5] = 0xb1;
	// And just something from /dev/urandom to create our source address
	smac[0] = 0xd2; smac[1] = 0xd8; smac[2] = 0x28;
	smac[3] = 0xe8; smac[4] = 0xb0; smac[5] = 0x96;
 
	// Similarly with the destination IP of the computer I wish to test with
	dip[0] = 192; dip[1] = 168; dip[2] = 10; dip[3] = 1;
	// and let's pick a source IP just ... somewhere on that network
	sip[0] = 192; sip[1] = 168; sip[2] = 10; sip[3] = 22;
 
	clear_scope(m_fpga);
 
	argn = 1;
 
	{
		bool	bad_address = false;
		char	*badp = NULL;
		if ((argn<argc)&&(strchr(argv[argn], ':'))) {
			if (!strtoenetaddr(argv[argn++], dmac)) {
				badp = argv[argn-1];
				bad_address = true;
			} else if ((argn<argc)&&(strchr(argv[argn], ':'))) {
				if (!strtoenetaddr(argv[argn++], smac)) {
					badp = argv[argn-1];
					bad_address = true;
				}
			}
		} if ((argn<argc)&&(!bad_address)&&(strchr(argv[argn], '.'))) {
			if (!strtoinetaddr(argv[argn++], dip)) {
				badp = argv[argn-1];
				bad_address = true;
			} else if ((argn<argc)&&(strchr(argv[argn], '.'))) {
				if (!strtoinetaddr(argv[argn++], sip)) {
					badp = argv[argn-1];
					bad_address = true;
				}
			}
		}
 
		if (bad_address) {
			usage();
			fprintf(stderr, "ERR: could not comprehend address, %s\n", badp);
			exit(EXIT_FAILURE);
		}
	}
 
	printf("Building packet\n");
	printf("From %3d.%3d.%3d.%3d [%02x:%02x:%02x:%02x:%02x:%02x]\n",
		sip[0], sip[1], sip[2], sip[3],
		smac[0], smac[1], smac[2], smac[3], smac[4], smac[5]);
	printf("To   %3d.%3d.%3d.%3d [%02x:%02x:%02x:%02x:%02x:%02x]\n",
		dip[0], dip[1], dip[2], dip[3],
		dmac[0], dmac[1], dmac[2], dmac[3], dmac[4], dmac[5]);
 
 
	// Let's build ourselves a ping packet
	packet[ 0] = (dmac[0]<<24)|(dmac[1]<<16)|(dmac[2]<<8)|(dmac[3]);
	packet[ 1] = (dmac[4]<<24)|(dmac[5]<<16)|(smac[0]<<8)|(smac[1]);
	packet[ 2] = (smac[2]<<24)|(smac[3]<<16)|(smac[4]<<8)|(smac[5]);
	packet[ 3] = 0x08000800;
	packet[ 4] = 0x4500001c; // IPv4, 20byte header, type of service = 0
	packet[ 5] = (urand[nu++]&0xffff0000); // Packet ID
	packet[ 6] = 0x80010000; // no flags, fragment offset=0, ttl=0, proto=1
	packet[ 7] = (sip[0]<<24)|(sip[1]<<16)|(sip[2]<<8)|(sip[3]);
	packet[ 8] = (dip[0]<<24)|(dip[1]<<16)|(dip[2]<<8)|(dip[3]);
	// Ping payload: type = 0x08 (PING, the response will be zero)
	//	CODE = 0
	//	Checksum will be filled in later
	packet[ 9] = 0x08000000;
	// This is the PING identifier and sequence number.  For now, we'll
	// just feed it random information--doesn't really matter what
	packet[10] = urand[nu++];
	// Now, the minimum ethernet packet is 16 words.  So, let's flush
	// ourselves out to that minimum length.
	packet[11] = 0;
	packet[12] = 0;
	packet[13] = 0;
	packet[14] = 0;
 
	// Calculate the IP header checksum
	ipchecksum(&packet[4]);
 
	// Calculate the PING payload checksum
	checksum  =  packet[ 9] & 0x0ffff;
	checksum += (packet[ 9]>>16)&0x0ffff;
	checksum +=  packet[10] & 0x0ffff;
	checksum += (packet[10]>>16)&0x0ffff;
	checksum  = ((checksum >> 16)&0x0ffff) + (checksum & 0x0ffff);
	checksum  = ((checksum >> 16)&0x0ffff) + (checksum & 0x0ffff);
	packet[ 9] = ((packet[9] & 0xffff0000)|(checksum))^0x0ffff;
 
	// Calculate the CRC--assuming we'll use it.
	packet[15] = calccrc(15*4, packet);
 
	// Clear any/all pending receiving errors or packets
	m_fpga->writeio(R_NET_RXCMD, 0x0fffff);
	if (config_hw_mac) {
		int ln;
 
		m_fpga->writeio(R_NET_MACHI, (smac[0]<<8)|(smac[1]));
		m_fpga->writeio(R_NET_MACLO, (smac[2]<<24)|(smac[3]<<16)|(smac[4]<<8)|(smac[5]));
 
		// Now, let's rebuild our packet for the non-hw-mac option,
		// now that we know the CRC.  In general, we're just going
		// to copy the packet we created earlier, but we need to
		// shift things as we do so.
		packet[ 0] = (dmac[0]<<24)|(dmac[1]<<16)|(dmac[2]<<8)|(dmac[3]);
		packet[ 1] = (dmac[4]<<24)|(dmac[5]<<16)|0x0800;
		packet[ 2] = packet[ 4];
		packet[ 3] = packet[ 5];
		packet[ 4] = packet[ 6];
		packet[ 5] = packet[ 7];
		packet[ 6] = packet[ 8];
		packet[ 7] = packet[ 9];
		packet[ 8] = packet[10];
		packet[ 9] = packet[11];
		packet[10] = packet[12];
		packet[11] = packet[13];
		packet[12] = packet[14];
		packet[13] = packet[15];
 
		ln = (config_hw_crc)?9:14;
		printf("Packet:\n");
		for(int i=0; i<14; i++)
			printf("\t%2d: 0x%08x\n", i, packet[i]);
 
		// Load the packet into the hardware buffer
		m_fpga->writei(R_NET_TXBUF, ln, packet);
 
		// And give it the transmit command.
		m_fpga->writeio(R_NET_TXCMD, TXGO|(ln<<2)|((config_hw_crc)?0:NOHWCRC));
 
	} else {
		int	ln;
 
		ln = (config_hw_crc)?11:12;
		printf("Packet:\n");
		for(int i=0; i<15; i++)
			printf("\t%3d: 0x%08x\n", i, packet[i]);
		printf("\tCRC: 0x%08x\n", packet[15]);
 
		// Load the packet into the hardware buffer
		m_fpga->writei(R_NET_TXBUF, ln, packet);
 
		// And give it the transmit command
		m_fpga->writeio(R_NET_TXCMD, TXGO|NOHWMAC|(ln<<2)|((config_hw_crc)?0:NOHWCRC));
	}
 
	// First, we need to look for any ARP requests, and we'll need to
	// respond to them.  If during this time we get a ping response
	// packet, we're done.
 
	printf("\nLooking for a response ...\n");
	unsigned rxstat;
	int	errcount = 0;
	do {
		rxstat = m_fpga->readio(R_NET_RXCMD);
		if (rxstat & 0x04000) {
			int	rxlen;
			unsigned *buf;
			printf("RX Status = %08x\n", rxstat);
			rxlen = ((rxstat & 0x03fff)+3)>>2;
			buf = new unsigned[rxlen];
			m_fpga->readi(R_NET_RXBUF, rxlen, buf);
			for(int i=0; i<rxlen; i++)
				printf("\tRX[%2d]: 0x%08x\n", i, buf[i]);
			delete[] buf;
			m_fpga->writeio(R_NET_RXCMD, 0xffffff);
			break;
		}
	} while(((rxstat & 0x04000)==0)&&(errcount++ < 50));
 
	rxstat = m_fpga->readio(R_NET_RXCMD);
	printf("Final Rx Status = %08x\n", rxstat);
 
 
	delete	m_fpga;
}