debugnet.c

/*
 *	debugnet library for PSP2 
 *	Copyright (C) 2010,2015 Antonio Jose Ramos Marquez (aka bigboss) @psxdev on twitter
 *  Repository https://github.com/psxdev/debugnet
 */

#include <stdio.h>
#include <stdarg.h>
#include <stdlib.h>
#include <string.h>
#include <psp2/net/net.h>
#include <psp2/net/netctl.h>
#include <psp2/types.h>
#include <psp2/kernel/clib.h>
#include <psp2/sysmodule.h>
#include "debugnet.h"

int debugnet_external_conf=0;
debugNetConfiguration *dconfig=NULL;
static void *net_memory = NULL;
static SceNetInAddr vita_addr;
struct SceNetSockaddrIn stSockAddr;


/**
 * UDP printf for debugnet library,
 * use debugNetPrintf() instead unless necessary
 *
 * @par Example:
 * @code
 * debugNetUDPPrintf("This is a %s test\n", "real");
 * @endcode
 */
void debugNetUDPPrintf(const char* fmt, ...)
{
  char buffer[0x800];
  va_list arg;
  va_start(arg, fmt);
  sceClibVsnprintf(buffer, sizeof(buffer), fmt, arg);
  va_end(arg);
  
  debugNetUDPSend(buffer);
}

/**
 * UDP Raw text send for debugnet library,
 * use debugNetPrintf() instead unless necessary
 *
 * @par Example:
 * @code
 * debugNetUDPSend("This is a test\n");
 * @endcode
 *
 * @param text - NULL-terminated buffer containing the raw text to send
 */
void debugNetUDPSend(const char *text)
{
	sceNetSend(dconfig->SocketFD, text, strlen(text), 0);
}

/**
 * Log Level printf for debugnet library 
 *
 * @par Example:
 * @code
 * debugNetPrintf(DBGNET_INFO,"This is a %s test\n", "real");
 * @endcode
 *
 * @param level - DBGNET_NONE,DBGNET_INFO,DBGNET_ERROR or DBGNET_DEBUG
 */
void debugNetPrintf(int level, const char* format, ...) 
{
	char msgbuf[0x800];
	va_list args;
	
		if (level>dconfig->logLevel)
		return;
       
	va_start(args, format);
       
	sceClibVsnprintf(msgbuf,2048, format, args);
	msgbuf[2047] = 0;
	va_end(args);
	if(level>dconfig->logLevel)
	{
		level=DBGNET_NONE;
	}
	switch(level)
	{
		case DBGNET_INFO:
	    	debugNetUDPPrintf("[VITA][INFO]: %s",msgbuf);  
	        break;
	   	case DBGNET_ERROR: 
	    	debugNetUDPPrintf("[VITA][ERROR]: %s",msgbuf);
	        break;
		case DBGNET_DEBUG:
	        debugNetUDPPrintf("[VITA][DEBUG]: %s",msgbuf);
	        break;
		case DBGNET_NONE:
			break;
	    default:
		    debugNetUDPPrintf("%s",msgbuf);
       
	}
}
/**
 * Set log level for debugnet library 
 *
 * @par Example:
 * @code
 * debugNetSetLogLevel(DBGNET_DEBUG);  
 * @endcode
 * @param level - DBGNET_DEBUG,DBGNET_ERROR,DBGNET_INFO or DBGNET_NONE 
 */
void debugNetSetLogLevel(int level)
{
	if(dconfig)
	{
		dconfig->logLevel=level;	
	}
}
/**
 * Init debugnet library 
 *
 * @par Example:
 * @code
 * #define LOGLEVEL 3  
 * int ret;
 * ret = debugNetInit("172.26.0.2", 18194, DBGNET_DEBUG);
 * @endcode
 *
 * @param serverIP - your pc/mac server ip
 * @param port - udp port server
 * @param level - DBGNET_DEBUG,DBGNET_ERROR,DBGNET_INFO or DBGNET_NONE 
 */
int debugNetInit(const char *serverIp, int port, int level)
{
    int ret=0;
    SceNetInitParam initparam;
    SceNetCtlInfo info;
	
	if(debugNetCreateConf())
	{
		return dconfig->debugnet_initialized;
	}
	
	debugNetSetLogLevel(level);
    
    
	if (sceSysmoduleIsLoaded(SCE_SYSMODULE_NET) != SCE_SYSMODULE_LOADED)
	ret=sceSysmoduleLoadModule(SCE_SYSMODULE_NET);
	
	if (ret >=0) {
		
	
    /*net initialazation code from xerpi at https://github.com/xerpi/FTPVita/blob/master/ftp.c*/
    /* Init Net */
    if (sceNetShowNetstat() == SCE_NET_ERROR_ENOTINIT) {
        net_memory = malloc(NET_INIT_SIZE);

        initparam.memory = net_memory;
        initparam.size = NET_INIT_SIZE;
        initparam.flags = 0;

        ret = sceNetInit(&initparam);
        //printf("sceNetInit(): 0x%08X\n", ret);
    } else {
        //printf("Net is already initialized.\n");
    }

    /* Init NetCtl */
    ret = sceNetCtlInit();
    //printf("sceNetCtlInit(): 0x%08X\n", ret);
   

    /* Get IP address */
    ret = sceNetCtlInetGetInfo(SCE_NETCTL_INFO_GET_IP_ADDRESS, &info);
    //printf("sceNetCtlInetGetInfo(): 0x%08X\n", ret);


    /* Save the IP of PSVita to a global variable */
    sceNetInetPton(SCE_NET_AF_INET, info.ip_address, &vita_addr);
	
	/* Create datagram udp socket*/
    dconfig->SocketFD = sceNetSocket("debugnet_socket",
        SCE_NET_AF_INET , SCE_NET_SOCK_DGRAM, SCE_NET_IPPROTO_UDP);
   
    memset(&stSockAddr, 0, sizeof stSockAddr);
	
	
	/*Populate SceNetSockaddrIn structure values*/
    stSockAddr.sin_family = SCE_NET_AF_INET;
    stSockAddr.sin_port = sceNetHtons(port);
	sceNetInetPton(SCE_NET_AF_INET, serverIp, &stSockAddr.sin_addr);

	/*Connect socket to server*/
    sceNetConnect(dconfig->SocketFD, (struct SceNetSockaddr *)&stSockAddr, sizeof stSockAddr);

	/*Show log on pc/mac side*/
	debugNetUDPPrintf("debugnet initialized\n");
	debugNetUDPPrintf("Copyright (C) 2010,2015 Antonio Jose Ramos Marquez aka bigboss @psxdev\n");
	debugNetUDPPrintf("This Program is subject to the terms of the Mozilla Public\n"
		"License, v. 2.0. If a copy of the MPL was not distributed with this\n"
		"file, You can obtain one at http://mozilla.org/MPL/2.0/.\n");
    debugNetUDPPrintf("ready to have a lot of fun...\n");

	/*library debugnet initialized*/
    dconfig->debugnet_initialized = 1;
	}

    return dconfig->debugnet_initialized;
}

debugNetConfiguration *debugNetGetConf()
{	
	if(dconfig)
	{
		return dconfig;
	}
	
	return NULL;	
}
int debugNetSetConf(debugNetConfiguration *conf)
{	
	if(conf)
	{
		dconfig=conf;
		debugnet_external_conf=1;
		return dconfig->debugnet_initialized;
	}
	
	return 0;	
}
int debugNetInitWithConf(debugNetConfiguration *conf)
{
	int ret;
	ret=debugNetSetConf(conf);
	if(ret)
	{
		debugNetPrintf(DBGNET_INFO,"debugnet already initialized using configuration from psp2link\n");
		debugNetPrintf(DBGNET_INFO,"debugnet_initialized=%d SocketFD=%d logLevel=%d\n",dconfig->debugnet_initialized,dconfig->SocketFD,dconfig->logLevel);
		debugNetPrintf(DBGNET_INFO,"ready to have a lot of fun...\n");
		return dconfig->debugnet_initialized;
	}
	else
	{
		return 0;
	}
	
}
int debugNetCreateConf()
{	
	if(!dconfig)
	{
		dconfig=malloc(sizeof(debugNetConfiguration));
		dconfig->debugnet_initialized=0;
		dconfig->SocketFD = -1;
		dconfig->logLevel=DBGNET_INFO;	
		return 0;
	}
	
	if(dconfig->debugnet_initialized)
	{
		return 1;
	}
	return 0;			
}


/**
 * Finish debugnet library 
 *
 * @par Example:
 * @code
 * debugNetFinish();
 * @endcode
 */
void debugNetFinish()
{
	
	if(!debugnet_external_conf)
	{
    	if (dconfig->debugnet_initialized) {
        	dconfig->debugnet_initialized = 0;
			dconfig->SocketFD=-1;
			
			//sceNetCtlTerm();

			//sceNetTerm();
			
			//sceSysmoduleUnloadModule(SCE_SYSMODULE_NET);
			
	        if (net_memory) {
	            free(net_memory);
	            net_memory = NULL;
	        }
    	}
	}
   
}