Add return value check for avoid interger overflow

Add return value check for avoid interger overflow #40581