ip_test.cpp

test internet中ip地址及其子网掩码

/****************************************************************************
*
* 文件名: ip_test.cpp
* 
* 文件描述:	判断IP地址的合法性并判断是否属于一个子网
*
* 创建人:
*  
* 版本号: 1.0
*
* 修改记录：
*
******************************************************************************/

#include <stdio.h>
#include <string.h>
#include <stdlib.h>
#include <ctype.h>

#define Max_len 50
//声明错误代码
#define ERR_NO_ERROR				0
#define ERR_EXIST_EMPTYPART			1
#define ERR_EXSIT_INVALIDSYMBOL		2
#define ERR_INVALID_EXPRESSION		3
#define ERR_OVERFLOW				4
#define ERR_MASK_OVERFLOW			5
#define ERR_MASK_INVALID_EXPRESSION 6

/*声明出错信息*/
char *errmsg[]={
	/* 0 */		"No error",
	/* 1 */		"Exist empty part number",
	/* 2 */		"Exist invalid symbol",
	/* 3 */		"Invalid expression",
	/* 4 */		"Over flow",
	/* 5 */		"Number of mask overflow"
	/* 6 */		"Invalid mask expression"
};

//声明错误代码全局变量
long errorno =0;

//打印出错信息函数
void perror(char *info)
{
	if (strlen(info)>0)
	{
		printf("%s: %s\n",info,errmsg[errorno]);
		return;
	}
	printf("Error: %s\n",errmsg[errorno]);
}

/*===================================================
*
* 函数名：split_str
* 
* 参  数：
*		   char *szSourse      :  源字符串
*		   char *szDesc        :  目的串
*		   unsigned int nStart :  起始位置
*		   char chSplit        :  分割字符
* 功能描述：
*			分割字符串
* 返回值：  void
*
* 抛出异常：
*
* 作  者：
*
=====================================================*/

void split_str(char *szSource,char *szDesc,unsigned int *nStart,char chSplit)
{
	char *p = NULL;
	if(*nStart >= strlen(szSource)) 
	{
		*szDesc = 0;
		printf("起始位置超出范围!");
		return;
	}
    p = szSource + *nStart;
	while(*p)
	{
		if(*p == chSplit) break;
		*szDesc = *p;
		p++;
		szDesc++;
		(*nStart)++;
	}
	(*nStart) = (*nStart) + 1;
	*szDesc = 0;
} 

/*====================================================================
*
* 函数名：ip_legal
* 
* 参  数：
*		   char *szSourse         :  ip地址字符串 202.113.114.43
*		   char *szDesc           :  从ip地址中取出的一部分，如：202
*		   unsigned int nStart    :  起始位置
*		   unsigned long szBuffer :  存放ip地址个部分的整数值
* 功能描述：
*			判断地址格式的合法性
* 返回值：  void
*
======================================================================*/

void ip_legal(char *szSourse,char * szDesc,unsigned int *nStart,unsigned long *szBuffer)
{
	int tag=0; //循环次数，每次循环得到地址中的一部分".255."
	unsigned long addr_num; //将每部分转换为整数
	
	//判断ip地址每一部分是否合法
	while(tag<4)
	{
		split_str(szSourse,szDesc,nStart,'.');
		addr_num=atol(szDesc);
		if (strlen(szDesc)==0) 
		{
			errorno=ERR_EXIST_EMPTYPART;
			perror("ip_legal");
			exit(0);
		}
		for (unsigned int idx=0;idx<strlen(szDesc);idx++)
		{
			if (isdigit(szDesc[idx])==0) 
			{
				errorno=ERR_EXSIT_INVALIDSYMBOL;
				perror("ip_legal");
				exit(0);
			}
		}
		if(tag==0 && addr_num==0)
		{
			errorno=ERR_INVALID_EXPRESSION;
			perror("ip_legal");
			exit(0);
		}
		if (addr_num<0 || addr_num>255) 
		{
			errorno=ERR_OVERFLOW;
			perror("ip_legal");
			exit(0);
		}
		else 
		{
			szBuffer[tag]=addr_num;
		}
		tag++;
	}
	split_str(szSourse,szDesc,nStart,'.');
	if (strlen(szDesc)!=0) 
	{
		errorno=ERR_INVALID_EXPRESSION;
		perror("ip_legal");
		exit(0);
	}

}


void main(int argc,char *args[])
{
	char szDesc[Max_len];
	char szSub_net[Max_len];
	char szMask[Max_len];
	unsigned int nStart = 0;
	unsigned long sub_net,ip,mask;
	unsigned long temp=0xffffffff;//
	int addr_num; //将每部分转换为整数
	unsigned long sub_addr[4],mask_addr[4],ip_addr[4];
	int tag=true; //tag标志子网掩码的表示方法，用"202.10.113.20"表示时tag=true;
	int mask_num;//子网掩码全为1的位数
	int len=3,Idx=0;
	int all=0xff;
	int sign=0; //临时变量，移位次数
	unsigned int check;//判断子网掩码的参数

	//判断参数个数
	if (argc!=3)
	{
		printf("usage error!");
		return;
	}
	
	//
	if (strstr(args[1],"/")==NULL)
	{
		printf("The sytle of usage error!\n");
		return;
	}

	//
	split_str(args[1],szDesc,&nStart,'/');
	strcpy(szSub_net,szDesc);
	//printf("%s\n",szDesc);
	split_str(args[1],szDesc,&nStart,'/');
	strcpy(szMask,szDesc);
	//printf("%s\n",szDesc);
	
	//判断子网
	nStart=0;
	ip_legal(szSub_net,szDesc,&nStart,sub_addr);
	
	//判断子网掩码
	if (strstr(szMask,".")!=NULL)
	{
		nStart=0;
		ip_legal(szMask,szDesc,&nStart,mask_addr);

		//判断是否符合子网掩码的标准
		while(mask_addr[len]==0) len--;
		while (Idx<len)
		{
			if (mask_addr[Idx]<255)
			{
				errorno=ERR_MASK_INVALID_EXPRESSION;
				perror("Mask");
				return;
			}
		}
		check=int(all<<sign);
		while (check!=mask_addr[len])
		{
			sign++;
			check=int(all<<sign);
		}
		if (sign>=8) 
		{
			printf("error!\n");
			return;
		}
	}
	else
	{
		mask_num=atoi(szMask);
		if (mask_num>30)
		{
			errorno=ERR_MASK_OVERFLOW;
			perror("mask");
			return;
		}
		else
		{
			tag=false;
		}
		
	}

	//判断IP地址
	nStart=0;
	ip_legal(args[2],szDesc,&nStart,ip_addr);
	
	//判断IP地址是否为子网成员
	sub_net=(sub_addr[0]<<24) | (sub_addr[1]<<16) | (sub_addr[2]<<8) | sub_addr[3];
	ip=	(ip_addr[0]<<24) | (ip_addr[1]<<16) | (ip_addr[2]<<8) | ip_addr[3];
	if (tag==false)
	{
		mask=temp<<(32-mask_num);
	}
	else
	{
		mask=temp<<(32-8*len+sign);
	}
	if ((ip & mask)!=sub_net)
	{
		printf("ip is not the subnet!\n");
		exit(0);
	}
	printf("The style is right!\n");
}