CCF头一次做出来第三题!!!HAPPY!!!
/*
* @Author: 365JHWZGo
* @Description: 202104-3 DHCP服务器
* @Date: 2021-09-01 11:10:38
* @FilePath: \VScodeC++practice\ccf\dhcp.cpp
* @LastEditTime: 2021-09-01 18:18:24
* @LastEditors: 365JHWZGo
*/
#include <iostream>
#include <vector>
using namespace std;
/*
N:地址池大小
Tdef:默认过期时间
Tmax:过期时间的上限
Tmin:过期时间的下限
H:运行 DHCP 服务器的主机名
num:收到的报文数量
t:收到报文的时刻
dhcp:dhcp数据报文
*/
int N, Tdef, Tmax, Tmin; //因为函数调用中有用到,需要提前声明
string H;
struct IP
{
int ipAddress; //ip地址
string ipState; //ip状态
string owner; //ip占用者
int expireTimeIP; //ip过期时间
};
struct dhcp
{
string sendName; //发送报文主机名
string receiveName; //接收报文主机名
string dataType; //数据类型
int ip; //IP
int expireTime; //过期时间
};
//判断发送报文主机名是否符合要求
bool judgeSN(string s)
{
bool flag = false;
for (int i = 0; i < s.size(); i++)
{
if (s[i] >= '0' && s[i] <= '9')
flag = true;
if (s[i] >= 'a' && s[i] <= 'z')
flag = true;
}
return flag;
}
//判断数据类型是否符合要求
bool judgeDataType(string s)
{
if (s != "OFR" && s != "REQ" && s != "ACK" && s != "NAK" && s != "DIS")
return false;
return true;
}
//初始化IP池
/*
no:未分配
prepare:待分配
own:占用
expire:过期
*/
void initIP(vector<IP> &pool, int n)
{
for (int i = 0; i < n; i++)
{
IP ip;
ip.ipAddress = i + 1;
ip.ipState = "no";
ip.owner = "";
ip.expireTimeIP = 0;
pool.push_back(ip);
}
}
//判断该数据报文是否需要处理
bool judgeIfNeedHandle(dhcp dp)
{
if (dp.receiveName != H && dp.receiveName != "*")
if (dp.dataType != "REQ")
return false;
if (dp.dataType != "REQ" && dp.dataType != "DIS")
return false;
if (dp.receiveName == "*" && dp.dataType != "DIS" || dp.receiveName == H && dp.dataType == "DIS")
return false;
return true * judgeSN(dp.sendName) * judgeDataType(dp.dataType);
}
//更新ip池
void updateIP(int t, vector<IP> &pool)
{
for (int i = 0; i < pool.size(); i++)
{
//处于待分配和占用状态的 IP 地址拥有一个大于零的过期时刻
//在到达该过期时刻时
if (pool[i].expireTimeIP <= t)
{
//若该地址的状态是待分配
if (pool[i].ipState == "prepare")
{
//则该地址的状态会自动变为未分配,且占用者清空,过期时刻清零
pool[i].ipState = "no";
pool[i].owner = "";
pool[i].expireTimeIP = 0;
}
if (pool[i].ipState == "own")
{
//该地址的状态会由占用自动变为过期,且过期时刻清零
pool[i].ipState = "expire";
pool[i].expireTimeIP = 0;
}
}
//此处标灰,否则运行超时,不需要验证这句话,因为ip已经初始化了
//处于未分配和过期状态的 IP 地址过期时刻为零,即没有过期时刻
// if (pool[i].ipState == "expire" || pool[i].ipState == "no")
// pool[i].expireTimeIP = 0;
}
}
//discover操作
void discover(int t, dhcp dp, vector<IP> &pool)
{
bool flag = true; //判断ip条件
int ipI = -1; //选取的ip坐标
updateIP(t, pool);
//检查此前是否给该客户端分配过 IP 地址,且该 IP 地址在此后没有被分配给其它客户端。
//如果是这样的情况,则直接将 IP 地址分配给它
//这句话挺重要的,值10分的测试点!!它在discover过程中没有说明,是在分配策略中说明的
for (int i = 0; i < pool.size(); i++)
{
if (pool[i].owner == dp.sendName)
{
flag = false;
ipI = i;
break;
}
}
//检查是否有占用者为发送主机的 IP 地址
//若有,则选取该 IP 地址
if (flag)
{
for (int i = 0; i < pool.size(); i++)
{
if (dp.ip == pool[i].ipAddress && pool[i].owner == dp.sendName)
{
flag = false;
ipI = i;
break;
}
}
}
//若没有,则选取最小的状态为未分配的 IP 地址
if (flag)
{
for (int i = 0; i < pool.size(); i++)
{
if (pool[i].ipState == "no")
{
flag = false;
ipI = i;
break;
}
}
}
//若没有,则选取最小的状态为过期的 IP 地址
if (flag)
{
for (int i = 0; i < pool.size(); i++)
{
if (pool[i].ipState == "expire")
{
flag = false;
ipI = i;
break;
}
}
}
//若没有,则不处理该报文,处理结束,此时flag==true
if (!flag)
{
//将该 IP 地址状态设置为待分配,占用者设置为发送主机
pool[ipI].ipState = "prepare";
pool[ipI].owner = dp.sendName;
//若报文中过期时刻为 0 ,则设置过期时刻为t + Tdef
if (dp.expireTime == 0)
pool[ipI].expireTimeIP = t + Tdef;
//否则根据报文中的过期时刻和收到报文的时刻计算过期时间,判断是否超过上下限
else
{
int cha = dp.expireTime - t;
//若没有超过,则设置过期时刻为报文中的过期时刻
if (cha <= Tmax && cha >= Tmin)
pool[ipI].expireTimeIP = dp.expireTime;
//否则则根据超限情况设置为允许的最早或最晚的过期时刻
else
{
if (cha > Tmax)
pool[ipI].expireTimeIP = Tmax + t;
else if (cha < Tmin)
{
pool[ipI].expireTimeIP = Tmin + t;
}
}
}
cout << H << " " << dp.sendName << " "<< "OFR" << " " << pool[ipI].ipAddress << " " << pool[ipI].expireTimeIP << endl;
}
}
//request操作
void request(int t, dhcp dp, vector<IP> &pool)
{
bool flag = true;
updateIP(t, pool);
//检查接收主机是否为本机
if (dp.receiveName == H)
{
for (int i = 0; i < pool.size(); i++)
{
//检查报文中的 IP 地址是否在地址池内,且其占用者为发送主机
if (dp.ip == pool[i].ipAddress && pool[i].owner == dp.sendName)
{
//无论该 IP 地址的状态为何,将该 IP 地址的状态设置为占用
pool[i].ipState = "own";
//若报文中过期时刻为 0 ,则设置过期时刻为t + Tdef
if (dp.expireTime == 0)
pool[i].expireTimeIP = t + Tdef;
//否则根据报文中的过期时刻和收到报文的时刻计算过期时间,判断是否超过上下
else
{
int cha = dp.expireTime - t;
//若没有超过,则设置过期时刻为报文中的过期时刻
if (cha <= Tmax && cha >= Tmin)
pool[i].expireTimeIP = dp.expireTime;
//否则则根据超限情况设置为允许的最早或最晚的过期时刻
else
{
if (cha > Tmax)
pool[i].expireTimeIP = Tmax + t;
else if (cha < Tmin)
{
pool[i].expireTimeIP = Tmin + t;
}
}
}
cout << H << " " << dp.sendName << " " << "ACK" << " " << pool[i].ipAddress << " " << pool[i].expireTimeIP << endl;
flag = false;
//if需要break;因为符合条件的只有一个,节省时间
break;
}
}
if (flag)
{
//若不是,则向发送主机发送 Nak 报文,处理结束
cout << H << " " << dp.sendName << " "<< "NAK" << " " << dp.ip << " " << 0 << endl;
}
}
//若不是
else
{
for (int i = 0; i < pool.size(); i++)
{
//占用者为发送主机的所有 IP 地址&&状态为待分配的
if (dp.sendName == pool[i].owner && pool[i].ipState == "prepare")
{
//将其状态设置为未分配,清空其占用者,清零其过期时刻
pool[i].ipState = "no";
pool[i].owner = "";
pool[i].expireTimeIP = 0;
}
}
}
}
int main()
{
std::ios::sync_with_stdio(false);
cin >> N >> Tdef >> Tmax >> Tmin >> H;
vector<IP> ipPool;
int num, t;
cin >> num;
initIP(ipPool, N);
for (int i = 0; i < num; i++)
{
dhcp dhcp;
cin >> t >> dhcp.sendName >> dhcp.receiveName >> dhcp.dataType >> dhcp.ip >> dhcp.expireTime;
bool a = dhcp.ip >= 0 ? true : false;
bool b = dhcp.expireTime >= 0 ? true : false;
bool flag = judgeIfNeedHandle(dhcp);
if (flag * a * b)
{
//对于 Discover 报文,按照下述方法处理
if (dhcp.dataType == "DIS")
{
discover(t, dhcp, ipPool);
}
//对于 Request 报文,按照下述方法处理
if (dhcp.dataType == "REQ")
{
request(t, dhcp, ipPool);
}
}
else
continue;
}
return 0;
}
使用std::ios::sync_with_stdio(false);
不使用std::ios::sync_with_stdio(false);
std::ios::sync_with_stdio(false);确实能节省时间