WIP
This commit is contained in:
David Biancolin
79
tests/big-blkdev.c
Normal file
79
tests/big-blkdev.c
Normal file
@@ -0,0 +1,79 @@
|
||||
#include <stdio.h>
|
||||
#include <stdlib.h>
|
||||
|
||||
#include "mmio.h"
|
||||
#include "blkdev.h"
|
||||
|
||||
#define SECTOR_WORDS (BLKDEV_SECTOR_SIZE / sizeof(uint64_t))
|
||||
#define TEST_SECTORS 128
|
||||
|
||||
unsigned long sector_buf[SECTOR_WORDS];
|
||||
|
||||
void write_sector(unsigned int secnum)
|
||||
{
|
||||
int req_tag, resp_tag;
|
||||
|
||||
for (int i = 0; i < SECTOR_WORDS; i++)
|
||||
sector_buf[i] = (secnum << 6) | i;
|
||||
|
||||
while (reg_read8(BLKDEV_NREQUEST) == 0);
|
||||
req_tag = blkdev_send_request((unsigned long) sector_buf, secnum, 1, 1);
|
||||
while (reg_read8(BLKDEV_NCOMPLETE) == 0);
|
||||
resp_tag = reg_read8(BLKDEV_COMPLETE);
|
||||
|
||||
if (req_tag != resp_tag) {
|
||||
printf("Response tag %d does not match request tag %d\n",
|
||||
req_tag, resp_tag);
|
||||
exit(EXIT_FAILURE);
|
||||
}
|
||||
}
|
||||
|
||||
void check_sector(unsigned int secnum)
|
||||
{
|
||||
int req_tag, resp_tag;
|
||||
|
||||
while (reg_read8(BLKDEV_NREQUEST) == 0);
|
||||
req_tag = blkdev_send_request((unsigned long) sector_buf, secnum, 1, 0);
|
||||
while (reg_read8(BLKDEV_NCOMPLETE) == 0);
|
||||
resp_tag = reg_read8(BLKDEV_COMPLETE);
|
||||
|
||||
if (req_tag != resp_tag) {
|
||||
printf("Response tag %d does not match request tag %d\n",
|
||||
req_tag, resp_tag);
|
||||
exit(EXIT_FAILURE);
|
||||
}
|
||||
|
||||
for (int i = 0; i < SECTOR_WORDS; i++) {
|
||||
unsigned long expected = (secnum << 6) | i;
|
||||
unsigned long actual = sector_buf[i];
|
||||
if (actual != expected) {
|
||||
printf("Word %d in sector %x does not match expected\n",
|
||||
i, secnum);
|
||||
printf("Expected %lx, got %lx\n",
|
||||
expected, actual);
|
||||
exit(EXIT_FAILURE);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
int main(void)
|
||||
{
|
||||
unsigned int nsectors = blkdev_nsectors();
|
||||
unsigned int stride = nsectors / TEST_SECTORS;
|
||||
|
||||
printf("Writing %u of %u sectors\n", TEST_SECTORS, nsectors);
|
||||
|
||||
for (int i = 0; i < TEST_SECTORS; i++) {
|
||||
int sector = i * stride;
|
||||
write_sector(sector);
|
||||
}
|
||||
|
||||
printf("Checking sectors\n", nsectors);
|
||||
|
||||
for (int i = 0; i < TEST_SECTORS; i++) {
|
||||
int sector = i * stride;
|
||||
check_sector(sector);
|
||||
}
|
||||
|
||||
return 0;
|
||||
}
|
||||
98
tests/nic-loopback.c
Normal file
98
tests/nic-loopback.c
Normal file
@@ -0,0 +1,98 @@
|
||||
#include "mmio.h"
|
||||
#include <stdlib.h>
|
||||
#include <stdio.h>
|
||||
#include <string.h>
|
||||
|
||||
#include "nic.h"
|
||||
#include "encoding.h"
|
||||
|
||||
#define NPACKETS 10
|
||||
#define TEST_OFFSET 3
|
||||
#define TEST_LEN 356
|
||||
#define ARRAY_LEN 360
|
||||
#define NTRIALS 3
|
||||
|
||||
uint32_t src[NPACKETS][ARRAY_LEN];
|
||||
uint32_t dst[NPACKETS][ARRAY_LEN];
|
||||
uint64_t lengths[NPACKETS];
|
||||
|
||||
static inline void send_recv()
|
||||
{
|
||||
uint64_t send_packet, recv_addr;
|
||||
int ncomps, send_comps_left = NPACKETS, recv_comps_left = NPACKETS;
|
||||
int recv_idx = 0;
|
||||
|
||||
for (int i = 0; i < NPACKETS; i++) {
|
||||
uint64_t pkt_size = TEST_LEN * sizeof(uint32_t);
|
||||
uint64_t src_addr = (uint64_t) &src[i][TEST_OFFSET];
|
||||
send_packet = (pkt_size << 48) | src_addr;
|
||||
recv_addr = (uint64_t) dst[i];
|
||||
reg_write64(SIMPLENIC_SEND_REQ, send_packet);
|
||||
reg_write64(SIMPLENIC_RECV_REQ, recv_addr);
|
||||
}
|
||||
|
||||
while (send_comps_left > 0 || recv_comps_left > 0) {
|
||||
ncomps = nic_send_comp_avail();
|
||||
asm volatile ("fence");
|
||||
for (int i = 0; i < ncomps; i++)
|
||||
reg_read16(SIMPLENIC_SEND_COMP);
|
||||
send_comps_left -= ncomps;
|
||||
|
||||
ncomps = nic_recv_comp_avail();
|
||||
asm volatile ("fence");
|
||||
for (int i = 0; i < ncomps; i++) {
|
||||
lengths[recv_idx] = reg_read16(SIMPLENIC_RECV_COMP);
|
||||
recv_idx++;
|
||||
}
|
||||
recv_comps_left -= ncomps;
|
||||
}
|
||||
}
|
||||
|
||||
void run_test(void)
|
||||
{
|
||||
unsigned long start, end;
|
||||
int i, j;
|
||||
|
||||
memset(dst, 0, sizeof(dst));
|
||||
asm volatile ("fence");
|
||||
|
||||
start = rdcycle();
|
||||
send_recv();
|
||||
end = rdcycle();
|
||||
|
||||
printf("send/recv %lu cycles\n", end - start);
|
||||
|
||||
for (i = 0; i < NPACKETS; i++) {
|
||||
if (lengths[i] != TEST_LEN * sizeof(uint32_t)) {
|
||||
printf("recv got wrong # bytes\n");
|
||||
exit(EXIT_FAILURE);
|
||||
}
|
||||
|
||||
for (j = 0; j < TEST_LEN; j++) {
|
||||
if (dst[i][j] != src[i][j + TEST_OFFSET]) {
|
||||
printf("Data mismatch @ %d, %d: %x != %x\n",
|
||||
i, j, dst[i][j], src[i][j + TEST_OFFSET]);
|
||||
exit(EXIT_FAILURE);
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
int main(void)
|
||||
{
|
||||
int i, j;
|
||||
|
||||
for (i = 0; i < NPACKETS; i++) {
|
||||
for (j = 0; j < ARRAY_LEN; j++)
|
||||
src[i][j] = i * ARRAY_LEN + j;
|
||||
}
|
||||
|
||||
for (i = 0; i < NTRIALS; i++) {
|
||||
printf("Trial %d\n", i);
|
||||
run_test();
|
||||
}
|
||||
|
||||
printf("All correct\n");
|
||||
|
||||
return 0;
|
||||
}
|
||||
@@ -8,22 +8,22 @@
|
||||
|
||||
static inline int nic_send_req_avail(void)
|
||||
{
|
||||
return reg_read16(SIMPLENIC_COUNTS) & 0xf;
|
||||
return reg_read32(SIMPLENIC_COUNTS) & 0xff;
|
||||
}
|
||||
|
||||
static inline int nic_recv_req_avail(void)
|
||||
{
|
||||
return (reg_read16(SIMPLENIC_COUNTS) >> 4) & 0xf;
|
||||
return (reg_read32(SIMPLENIC_COUNTS) >> 8) & 0xff;
|
||||
}
|
||||
|
||||
static inline int nic_send_comp_avail(void)
|
||||
{
|
||||
return (reg_read16(SIMPLENIC_COUNTS) >> 8) & 0xf;
|
||||
return (reg_read32(SIMPLENIC_COUNTS) >> 16) & 0xff;
|
||||
}
|
||||
|
||||
static inline int nic_recv_comp_avail(void)
|
||||
{
|
||||
return (reg_read16(SIMPLENIC_COUNTS) >> 12) & 0xf;
|
||||
return (reg_read32(SIMPLENIC_COUNTS) >> 24) & 0xff;
|
||||
}
|
||||
|
||||
static void nic_send(void *data, unsigned long len)
|
||||
|
||||
254
tests/pingd.c
Normal file
254
tests/pingd.c
Normal file
@@ -0,0 +1,254 @@
|
||||
#include "mmio.h"
|
||||
#include "nic.h"
|
||||
|
||||
#include <stdint.h>
|
||||
#include <stdlib.h>
|
||||
#include <stdio.h>
|
||||
#include <string.h>
|
||||
|
||||
#define ETH_MAX_WORDS 190
|
||||
#define NET_IP_ALIGN 2
|
||||
#define ETH_HEADER_SIZE 14
|
||||
#define MAC_ADDR_SIZE 6
|
||||
#define IP_ADDR_SIZE 4
|
||||
|
||||
#define IPV4_ETHTYPE 0x0800
|
||||
#define ARP_ETHTYPE 0x0806
|
||||
#define ICMP_PROT 1
|
||||
#define ECHO_REPLY 0
|
||||
#define ECHO_REQUEST 8
|
||||
#define ARP_REQUEST 1
|
||||
#define ARP_REPLY 2
|
||||
#define HTYPE_ETH 1
|
||||
|
||||
static inline uint16_t ntohs(uint16_t nint)
|
||||
{
|
||||
return ((nint & 0xff) << 8) | ((nint >> 8) & 0xff);
|
||||
}
|
||||
|
||||
static inline uint16_t htons(uint16_t nint)
|
||||
{
|
||||
return ntohs(nint);
|
||||
}
|
||||
|
||||
struct eth_header {
|
||||
uint8_t padding[NET_IP_ALIGN];
|
||||
uint8_t dst_mac[MAC_ADDR_SIZE];
|
||||
uint8_t src_mac[MAC_ADDR_SIZE];
|
||||
uint16_t ethtype;
|
||||
};
|
||||
|
||||
struct arp_header {
|
||||
uint16_t htype;
|
||||
uint16_t ptype;
|
||||
uint8_t hlen;
|
||||
uint8_t plen;
|
||||
uint16_t oper;
|
||||
uint8_t sha[MAC_ADDR_SIZE];
|
||||
uint8_t spa[IP_ADDR_SIZE];
|
||||
uint8_t tha[MAC_ADDR_SIZE];
|
||||
uint8_t tpa[IP_ADDR_SIZE];
|
||||
};
|
||||
|
||||
struct ipv4_header {
|
||||
uint8_t ver_ihl;
|
||||
uint8_t dscp_ecn;
|
||||
uint16_t length;
|
||||
uint16_t ident;
|
||||
uint16_t flags_frag_off;
|
||||
uint8_t ttl;
|
||||
uint8_t prot;
|
||||
uint16_t cksum;
|
||||
uint32_t src_addr;
|
||||
uint32_t dst_addr;
|
||||
};
|
||||
|
||||
struct icmp_header {
|
||||
uint8_t type;
|
||||
uint8_t code;
|
||||
uint16_t cksum;
|
||||
uint32_t rest;
|
||||
};
|
||||
|
||||
static int checksum(uint16_t *data, int len)
|
||||
{
|
||||
int i;
|
||||
uint32_t sum = 0;
|
||||
|
||||
for (i = 0; i < len; i++)
|
||||
sum += ntohs(data[i]);
|
||||
|
||||
while ((sum >> 16) != 0)
|
||||
sum = (sum & 0xffff) + (sum >> 16);
|
||||
|
||||
sum = ~sum & 0xffff;
|
||||
|
||||
return sum;
|
||||
}
|
||||
|
||||
#define ceil_div(n, d) (((n) - 1) / (d) + 1)
|
||||
|
||||
static int process_arp(void *buf, uint8_t *mac)
|
||||
{
|
||||
struct eth_header *eth = buf;
|
||||
struct arp_header *arp;
|
||||
size_t size = ETH_HEADER_SIZE + sizeof(*arp);
|
||||
uint8_t tmp_addr[IP_ADDR_SIZE];
|
||||
|
||||
// Verify arp packet
|
||||
arp = buf + sizeof(*eth);
|
||||
if (ntohs(arp->oper) != ARP_REQUEST) {
|
||||
printf("Wrong arp operation: %d\n", ntohs(arp->oper));
|
||||
return -1;
|
||||
}
|
||||
|
||||
if (ntohs(arp->htype) != HTYPE_ETH) {
|
||||
printf("Wrong ARP HTYPE\n");
|
||||
return -1;
|
||||
}
|
||||
|
||||
if (ntohs(arp->ptype) != IPV4_ETHTYPE) {
|
||||
printf("Wrong ARP PTYPE\n");
|
||||
return -1;
|
||||
}
|
||||
|
||||
if (arp->hlen != 6) {
|
||||
printf("Wrong ARP HLEN: %d\n", arp->hlen);
|
||||
return -1;
|
||||
}
|
||||
|
||||
if (arp->plen != 4) {
|
||||
printf("Wrong ARP PLEN: %d\n", arp->plen);
|
||||
return -1;
|
||||
}
|
||||
|
||||
// Make the source the destination, and add our mac address
|
||||
memcpy(eth->dst_mac, eth->src_mac, MAC_ADDR_SIZE);
|
||||
memcpy(eth->src_mac, mac, MAC_ADDR_SIZE);
|
||||
|
||||
// create ARP reply
|
||||
arp->oper = htons(ARP_REPLY);
|
||||
|
||||
// Make tha the sha, and fill in sha with actual mac address
|
||||
memcpy(arp->tha, arp->sha, MAC_ADDR_SIZE);
|
||||
memcpy(arp->sha, mac, MAC_ADDR_SIZE);
|
||||
|
||||
// Swap spa and tpa in arp packet
|
||||
memcpy(tmp_addr, arp->tpa, IP_ADDR_SIZE);
|
||||
memcpy(arp->tpa, arp->spa, IP_ADDR_SIZE);
|
||||
memcpy(arp->spa, tmp_addr, IP_ADDR_SIZE);
|
||||
|
||||
size = ceil_div(size + NET_IP_ALIGN, 8) * 8;
|
||||
nic_send(buf, size);
|
||||
|
||||
return 0;
|
||||
}
|
||||
static int process_icmp(void *buf, uint8_t *mac)
|
||||
{
|
||||
struct eth_header *eth = buf;
|
||||
struct ipv4_header *ipv4;
|
||||
struct icmp_header *icmp;
|
||||
int ihl, icmp_size;
|
||||
ssize_t size;
|
||||
uint32_t tmp_addr;
|
||||
|
||||
// verify IPv4
|
||||
ipv4 = buf + sizeof(*eth);
|
||||
ihl = ipv4->ver_ihl & 0xf;
|
||||
|
||||
if (checksum((uint16_t *) ipv4, ihl << 1) != 0) {
|
||||
printf("Bad IP header checksum %04x\n", ipv4->cksum);
|
||||
return -1;
|
||||
}
|
||||
|
||||
if (ipv4->prot != ICMP_PROT) {
|
||||
printf("Wrong IP protocol %d\n", ipv4->prot);
|
||||
return -1;
|
||||
}
|
||||
|
||||
// verify ICMP
|
||||
icmp = (buf + sizeof(*eth) + (ihl << 2));
|
||||
|
||||
if (icmp->type != ECHO_REQUEST) {
|
||||
printf("Wrong ICMP type %d\n", icmp->type);
|
||||
return -1;
|
||||
}
|
||||
|
||||
if (icmp->code != 0) {
|
||||
printf("Wrong ICMP code %d\n", icmp->code);
|
||||
return -1;
|
||||
}
|
||||
|
||||
icmp_size = ntohs(ipv4->length) - (ihl << 2);
|
||||
if (checksum((uint16_t *) icmp, icmp_size >> 1) != 0) {
|
||||
printf("Bad ICMP checksum %04x\n", icmp->cksum);
|
||||
return -1;
|
||||
}
|
||||
|
||||
// Set the destination and source MACs
|
||||
memcpy(eth->dst_mac, eth->src_mac, MAC_ADDR_SIZE);
|
||||
memcpy(eth->src_mac, mac, MAC_ADDR_SIZE);
|
||||
|
||||
// Swap the source and destination IP addresses
|
||||
tmp_addr = ipv4->dst_addr;
|
||||
ipv4->dst_addr = ipv4->src_addr;
|
||||
ipv4->src_addr = tmp_addr;
|
||||
|
||||
// compute the IPv4 header checksum
|
||||
ipv4->cksum = 0;
|
||||
ipv4->cksum = htons(checksum((uint16_t *) ipv4, ihl << 1));
|
||||
|
||||
// set the ICMP type to reply and compute checksum
|
||||
icmp->cksum = 0;
|
||||
icmp->type = ECHO_REPLY;
|
||||
icmp->cksum = htons(checksum((uint16_t *) icmp, icmp_size >> 1));
|
||||
size = ntohs(ipv4->length) + ETH_HEADER_SIZE;
|
||||
|
||||
size = ceil_div(size + NET_IP_ALIGN, 8) * 8;
|
||||
nic_send(buf, size);
|
||||
|
||||
return 0;
|
||||
}
|
||||
|
||||
static int process_packet(void *buf, uint8_t *mac)
|
||||
{
|
||||
struct eth_header *eth;
|
||||
|
||||
// read the ICMP request
|
||||
nic_recv(buf);
|
||||
eth = buf;
|
||||
printf("Got packet: [ethtype=%04x]\n", ntohs(eth->ethtype));
|
||||
// Check ethernet type
|
||||
switch (ntohs(eth->ethtype)) {
|
||||
case IPV4_ETHTYPE:
|
||||
return process_icmp(buf, mac);
|
||||
case ARP_ETHTYPE:
|
||||
return process_arp(buf, mac);
|
||||
default:
|
||||
printf("Wrong ethtype %x\n", ntohs(eth->ethtype));
|
||||
return -1;
|
||||
}
|
||||
}
|
||||
|
||||
uint64_t buffer[ETH_MAX_WORDS];
|
||||
|
||||
int main(void)
|
||||
{
|
||||
uint64_t macaddr_long;
|
||||
uint8_t *macaddr;
|
||||
|
||||
macaddr_long = nic_macaddr();
|
||||
macaddr = (uint8_t *) &macaddr_long;
|
||||
|
||||
printf("macaddr - %02x", macaddr[0]);
|
||||
for (int i = 1; i < MAC_ADDR_SIZE; i++)
|
||||
printf(":%02x", macaddr[i]);
|
||||
printf("\n");
|
||||
|
||||
for (;;) {
|
||||
if (process_packet(buffer, macaddr))
|
||||
return -1;
|
||||
}
|
||||
|
||||
return 0;
|
||||
}
|
||||
Reference in New Issue
Block a user