ps2ip.c

Go to the documentation of this file.

/*
# _____     ___ ____     ___ ____
#  ____|   |    ____|   |        | |____|
# |     ___|   |____ ___|    ____| |    \    PS2DEV Open Source Project.
#-----------------------------------------------------------------------
# Copyright 2001-2004, ps2dev - http://www.ps2dev.org
# Licenced under Academic Free License version 2.0
# Review ps2sdk README & LICENSE files for further details.
*/
 
#include <types.h>
#include <stdio.h>
#include <intrman.h>
#include <netman.h>
#include <loadcore.h>
#include <thbase.h>
#include <sysclib.h>
#include <thevent.h>
#include <sysmem.h>
#include <lwip/memp.h>
 
#include "lwip/sys.h"
#include "lwip/tcp.h"
#include "lwip/tcpip.h"
#include "lwip/prot/dhcp.h"
#include "lwip/netif.h"
#include "lwip/dhcp.h"
#include "lwip/inet.h"
#include "netif/etharp.h"
 
#include "ps2ip_internal.h"
 
typedef struct pbuf PBuf;
typedef struct netif    NetIF;
typedef struct ip4_addr IPAddr;
 
static struct pbuf *TxHead, *TxTail;
 
#define MODNAME "TCP/IP Stack"
IRX_ID(MODNAME, 2, 3);
 
extern struct irx_export_table  _exp_ps2ip;
 
#if NOSYS
static int      iTimerARP=0;
 
#if     defined(PS2IP_DHCP)
static int      iTimerDHCP=0;
#endif  //defined(PS2IP_DHCP)
#endif
 
int
ps2ip_getconfig(char* pszName,t_ip_info* pInfo)
{
    NetIF*  pNetIF=netif_find(pszName);
 
    if (pNetIF==NULL)
    {
 
        //Net interface not found.
 
        memset(pInfo,0,sizeof(*pInfo));
        return  0;
    }
    strcpy(pInfo->netif_name,pszName);
    pInfo->ipaddr.s_addr=pNetIF->ip_addr.addr;
    pInfo->netmask.s_addr=pNetIF->netmask.addr;
    pInfo->gw.s_addr=pNetIF->gw.addr;
 
    memcpy(pInfo->hw_addr,pNetIF->hwaddr,sizeof(pInfo->hw_addr));
 
#if     LWIP_DHCP
    struct dhcp *dhcp = netif_dhcp_data(pNetIF);
 
    if ((dhcp != NULL) && (dhcp->state != DHCP_STATE_OFF))
    {
        pInfo->dhcp_enabled=1;
        pInfo->dhcp_status=dhcp->state;
    }
    else
    {
        pInfo->dhcp_enabled=0;
        pInfo->dhcp_status=DHCP_STATE_OFF;
    }
 
#else
 
    pInfo->dhcp_enabled=0;
 
#endif
 
    return  1;
}
 
 
int
ps2ip_setconfig(const t_ip_info* pInfo)
{
    NetIF*  pNetIF=netif_find(pInfo->netif_name);
 
    if  (pNetIF==NULL)
    {
        return  0;
    }
 
#if LWIP_DHCP
    struct dhcp *dhcp = netif_dhcp_data(pNetIF);
 
    //Enable dhcp here
 
    if (pInfo->dhcp_enabled)
    {
        if ((dhcp == NULL) || (dhcp->state == DHCP_STATE_OFF))
        {
            //Set initial IP address
            netif_set_ipaddr(pNetIF,(const IPAddr*)&pInfo->ipaddr);
            netif_set_netmask(pNetIF,(const IPAddr*)&pInfo->netmask);
            netif_set_gw(pNetIF,(const IPAddr*)&pInfo->gw);
 
            //Start DHCP client
            dhcp_start(pNetIF);
        }
    }
    else
    {
        if ((dhcp != NULL) && (dhcp->state != DHCP_STATE_OFF))
        {
            //Release DHCP lease
            dhcp_release(pNetIF);
 
            //Stop DHCP client
            dhcp_stop(pNetIF);
        }
 
        netif_set_ipaddr(pNetIF,(const IPAddr*)&pInfo->ipaddr);
        netif_set_netmask(pNetIF,(const IPAddr*)&pInfo->netmask);
        netif_set_gw(pNetIF,(const IPAddr*)&pInfo->gw);
    }
#else
    netif_set_ipaddr(pNetIF,(const IPAddr*)&pInfo->ipaddr);
    netif_set_netmask(pNetIF,(const IPAddr*)&pInfo->netmask);
    netif_set_gw(pNetIF,(const IPAddr*)&pInfo->gw);
#endif
 
    return  1;
}
 
static void InitDone(void* pvArg)
{
    dbgprintf("InitDone: TCPIP initialized\n");
    sys_sem_signal((sys_sem_t*)pvArg);
}
 
#if NOSYS
static void TimerThread(void* pvArg)
{
    while (1)
    {
        //TCP timer.
        tcp_tmr();
 
        //ARP timer.
        iTimerARP+=TCP_TMR_INTERVAL;
        if  (iTimerARP>=ARP_TMR_INTERVAL)
        {
            iTimerARP-=ARP_TMR_INTERVAL;
            etharp_tmr();
        }
 
#if     defined(PS2IP_DHCP)
 
        //DHCP timer.
 
        iTimerDHCP+=TCP_TMR_INTERVAL;
        if ((iTimerDHCP-TCP_TMR_INTERVAL)/DHCP_FINE_TIMER_MSECS!=iTimerDHCP/DHCP_FINE_TIMER_MSECS)
        {
            dhcp_fine_tmr();
        }
 
        if (iTimerDHCP>=DHCP_COARSE_TIMER_SECS*1000)
        {
            iTimerDHCP-=DHCP_COARSE_TIMER_SECS*1000;
            dhcp_coarse_tmr();
        }
#endif
 
        DelayThread(TCP_TMR_INTERVAL*250);
    }
}
 
static inline void InitTimer(void)
{
    iop_thread_t    Thread={TH_C, (u32)"PS2IP-timer", TimerThread, 0x300, 0x16};
    int     iTimerThreadID=CreateThread(&Thread);
 
    if (iTimerThreadID<0)
    {
        printf("InitTimer: Fatal error - Failed to create tcpip timer-thread!\n");
    }
 
    //Start timer-thread
    StartThread(iTimerThreadID, NULL);
}
#endif
 
err_t
ps2ip_input(PBuf* pInput,NetIF* pNetIF)
{
    err_t result;
 
    if((result = pNetIF->input(pInput, pNetIF)) != ERR_OK)
        pbuf_free(pInput);
 
    return result;
}
 
int _exit(int argc, char *argv[])
{
    (void)argc;
    (void)argv;
 
    return MODULE_NO_RESIDENT_END; // return "not resident"!
}
 
static struct netif NIF;
 
static void LinkStateUp(void){
    tcpip_callback((void*)&netif_set_link_up, &NIF);
}
 
static void LinkStateDown(void){
    tcpip_callback((void*)&netif_set_link_down, &NIF);
}
 
static void *AllocRxPacket(unsigned int size, void **payload)
{
    struct pbuf* pbuf;
 
    if((pbuf = pbuf_alloc(PBUF_RAW, size, PBUF_POOL)) != NULL)
    {
        *payload = pbuf->payload;
    }
 
    return pbuf;
}
 
static void FreeRxPacket(void *packet)
{
    pbuf_free(packet);
}
 
static void EnQRxPacket(void *packet)
{
    ps2ip_input(packet, &NIF);
}
 
static int NextTxPacket(void **payload)
{
    int len;
 
    if(TxTail != NULL)
    {
        *payload = TxTail->payload;
        len = TxTail->len;
    } else
        len = 0;
 
    return len;
}
 
static void DeQTxPacket(void)
{
    struct pbuf *toFree;
    int OldState;
 
    toFree = NULL;
 
    CpuSuspendIntr(&OldState);
    if(TxTail != NULL)
    {
        toFree = TxTail;
 
        if(TxTail == TxHead) {
            //Last in queue.
            TxTail = NULL;
            TxHead = NULL;
        } else {
            TxTail = TxTail->next;
        }
    }
    CpuResumeIntr(OldState);
 
    if(toFree != NULL)
    {
        toFree->next = NULL;
        pbuf_free(toFree);
    }
}
 
static void EnQTxPacket(struct pbuf *tx)
{
    int OldState;
 
    CpuSuspendIntr(&OldState);
 
    if(TxHead != NULL)
        TxHead->next = tx;
 
    TxHead = tx;
    tx->next = NULL;
 
    if(TxTail == NULL)  //Queue empty
        TxTail = TxHead;
 
    CpuResumeIntr(OldState);
}
 
static err_t
SMapLowLevelOutput(struct netif *pNetIF, struct pbuf* pOutput)
{
    err_t result;
 
    (void)pNetIF;
 
    result = ERR_OK;
    if(pOutput->tot_len > pOutput->len)
    {
        struct pbuf* pbuf;
 
        pbuf_ref(pOutput);  //Increment reference count because LWIP must free the PBUF, not the driver!
        if((pbuf = pbuf_coalesce(pOutput, PBUF_RAW)) != pOutput)
        {   //No need to increase reference count because pbuf_coalesce() does it.
            EnQTxPacket(pbuf);
            NetManNetIFXmit();
        } else
            result = ERR_MEM;
    } else {
        pbuf_ref(pOutput);  //This will be freed later.
        EnQTxPacket(pOutput);
        NetManNetIFXmit();
    }
 
    return result;
}
 
//SMapOutput():
 
//This function is called by the TCP/IP stack when an IP packet should be sent. It'll be invoked in the context of the
//tcpip-thread, hence no synchronization is required.
// For LWIP versions before v1.3.0.
#ifdef PRE_LWIP_130_COMPAT
static err_t
SMapOutput(struct netif *pNetIF, struct pbuf *pOutput, IPAddr* pIPAddr)
{
    struct pbuf *pBuf=etharp_output(pNetIF,pIPAddr,pOutput);
 
    return  pBuf!=NULL ? SMapLowLevelOutput(pNetIF, pBuf):ERR_OK;
}
#endif
 
//Should be called at the beginning of the program to set up the network interface.
static err_t SMapIFInit(struct netif* pNetIF)
{
    TxHead = NULL;
    TxTail = NULL;
 
    pNetIF->name[0]='s';
    pNetIF->name[1]='m';
#ifdef PRE_LWIP_130_COMPAT
    pNetIF->output=&SMapOutput; // For LWIP versions before v1.3.0.
#else
    pNetIF->output=&etharp_output;  // For LWIP 1.3.0 and later.
#endif
    pNetIF->linkoutput=&SMapLowLevelOutput;
    pNetIF->hwaddr_len=NETIF_MAX_HWADDR_LEN;
#ifdef PRE_LWIP_130_COMPAT
    pNetIF->flags|=(NETIF_FLAG_LINK_UP|NETIF_FLAG_BROADCAST);   // For LWIP versions before v1.3.0.
#else
    pNetIF->flags|=(NETIF_FLAG_ETHARP|NETIF_FLAG_BROADCAST);    // For LWIP v1.3.0 and later.
#endif
    pNetIF->mtu=1500;
 
    //Get MAC address.
    NetManIoctl(NETMAN_NETIF_IOCTL_ETH_GET_MAC, NULL, 0, pNetIF->hwaddr, sizeof(pNetIF->hwaddr));
//  DEBUG_PRINTF("MAC address : %02x:%02x:%02x:%02x:%02x:%02x\n",pNetIF->hwaddr[0],pNetIF->hwaddr[1],pNetIF->hwaddr[2],
//               pNetIF->hwaddr[3],pNetIF->hwaddr[4],pNetIF->hwaddr[5]);
 
    return  ERR_OK;
}
 
static inline int InitializeLWIP(void){
    sys_sem_t   Sema;
    int     result;
 
    dbgprintf("PS2IP: Module Loaded.\n");
 
    if ((result = RegisterLibraryEntries(&_exp_ps2ip))!=0)
    {
        printf("PS2IP: RegisterLibraryEntries returned: %d\n", result);
    } else {
        sys_sem_new(&Sema, 0);
        dbgprintf("PS2IP: Calling tcpip_init\n");
        tcpip_init(InitDone,&Sema);
 
        sys_arch_sem_wait(&Sema, 0);
        sys_sem_free(&Sema);
 
        dbgprintf("PS2IP: tcpip_init called\n");
#if NOSYS
        InitTimer();
#endif
 
        dbgprintf("PS2IP: System Initialised\n");
 
        result = 0;
    }
 
    return result;
}
 
static inline int InitLWIPStack(IPAddr *IP, IPAddr *NM, IPAddr *GW){
    int result;
    static struct NetManNetProtStack stack={
        &LinkStateUp,
        &LinkStateDown,
        &AllocRxPacket,
        &FreeRxPacket,
        &EnQRxPacket,
        &NextTxPacket,
        &DeQTxPacket,
        NULL,
        NULL
    };
 
    if((result = InitializeLWIP()) != 0)
        return result;
 
    netif_add(&NIF, IP, NM, GW, NULL, &SMapIFInit, tcpip_input);
    netif_set_default(&NIF);
 
    NetManRegisterNetworkStack(&stack);
    netif_set_up(&NIF);
 
    return 0;
}
 
int _start(int argc, char *argv[]){
    IPAddr IP, NM, GW;
 
    //Parse IP address arguments.
    if(argc>=4)
    {
        dbgprintf("SMAP: %s %s %s\n", argv[1],argv[2],argv[3]);
        IP.addr=inet_addr(argv[1]);
        NM.addr=inet_addr(argv[2]);
        GW.addr=inet_addr(argv[3]);
    }
    else
    {
        //Set some defaults.
        IP4_ADDR(&IP,192,168,0,80);
        IP4_ADDR(&NM,255,255,255,0);
        IP4_ADDR(&GW,192,168,0,1);
    }
 
    return InitLWIPStack(&IP, &NM, &GW)==0?MODULE_RESIDENT_END:MODULE_NO_RESIDENT_END;
}