function startShare(address tokenAddress, uint256 tokenAmount) external payable notContract whenRunning nonReentrant {
if (isRequireFee && minShareFee > 0) {
require(msg.value >= minShareFee, 'Invalid share fee');
payable(feeAddress).transfer(msg.value);
}

require(tokenAmount > 0, "Invalid token amount");
require(tokenAddress != address(0), "Zero address");
require(IERC20(tokenAddress).balanceOf(msg.sender) >= tokenAmount, "Not enough token balance");
IERC20(tokenAddress).safeTransferFrom(address(msg.sender), treasuryAddress, tokenAmount);

uint256 _id = ++shareId;
ShareItem memory newShareItem = ShareItem({
id: _id,
userAddress: msg.sender,
tokenAddress: tokenAddress,
tokenAmount: tokenAmount,
timestamp: block.timestamp,
pendingCompleteTimestamp: block.timestamp + (maxDayInterest * 1 days),
expireCompleteTimestamp: block.timestamp + ((maxDayInterest + pendingCompleteDay) * 1 days),
completeTimestamp: 0,
finishTimestamp: 0,
status: uint8(STATUS.INACTIVE),
mode: uint8(MODE.STANDARD),
tokenType: uint8(TOKEN_TYPE.ERC20)
});
listShareItem[_id] = newShareItem;
allShareId.push(_id);
shareIdByAddress[msg.sender].push(_id);
emit StartShare(msg.sender, tokenAddress, tokenAmount, _id);
}

/**
* @notice Start share main
* @param amount: amount
* @dev Callable by users
*/
function startShareMain(uint256 amount) external payable notContract whenRunning nonReentrant {
require(amount > 0 && msg.value >= amount, "Invalid amount");
uint256 feeAmount;
if (isRequireFee && minShareFee > 0) {
feeAmount = (msg.value).sub(amount);
require(feeAmount >= minShareFee, 'Invalid share fee');
payable(feeAddress).transfer(feeAmount);
}
payable(treasuryAddress).transfer(amount);

uint256 _id = ++shareId;
ShareItem memory newShareItem = ShareItem({
id: _id,
userAddress: msg.sender,
tokenAddress: zeroAddress,
tokenAmount: amount,
timestamp: block.timestamp,
pendingCompleteTimestamp: block.timestamp + (maxDayInterest * 1 days),
expireCompleteTimestamp: block.timestamp + ((maxDayInterest + pendingCompleteDay) * 1 days),
completeTimestamp: 0,
finishTimestamp: 0,
status: uint8(STATUS.INACTIVE),
mode: uint8(MODE.STANDARD),
tokenType: uint8(TOKEN_TYPE.MAIN)
});
listShareItem[_id] = newShareItem;
allShareId.push(_id);
shareIdByAddress[msg.sender].push(_id);
emit StartShareMain(msg.sender, amount, feeAmount, _id);
}

Audit Certificate

Disclaimer

Scan to verify