// SPDX-License-Identifier: GPL-2.0-or-later
/* mpi-sub-ui.c - Subtract an unsigned integer from an MPI.
 *
 * Copyright 1991, 1993, 1994, 1996, 1999-2002, 2004, 2012, 2013, 2015
 * Free Software Foundation, Inc.
 *
 * This file was based on the GNU MP Library source file:
 * https://gmplib.org/repo/gmp-6.2/file/510b83519d1c/mpz/aors_ui.h
 *
 * The GNU MP Library is free software; you can redistribute it and/or modify
 * it under the terms of either:
 *
 *   * the GNU Lesser General Public License as published by the Free
 *     Software Foundation; either version 3 of the License, or (at your
 *     option) any later version.
 *
 * or
 *
 *   * the GNU General Public License as published by the Free Software
 *     Foundation; either version 2 of the License, or (at your option) any
 *     later version.
 *
 * or both in parallel, as here.
 *
 * The GNU MP Library is distributed in the hope that it will be useful, but
 * WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY
 * or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License
 * for more details.
 *
 * You should have received copies of the GNU General Public License and the
 * GNU Lesser General Public License along with the GNU MP Library.  If not,
 * see https://www.gnu.org/licenses/.
 */

#include <linux/export.h>

#include "mpi-internal.h"

int mpi_sub_ui(MPI w, MPI u, unsigned long vval)
{
	if (u->nlimbs == 0) {
		if (mpi_resize(w, 1) < 0)
			return -ENOMEM;
		w->d[0] = vval;
		w->nlimbs = (vval != 0);
		w->sign = (vval != 0);
		return 0;
	}

	/* If not space for W (and possible carry), increase space. */
	if (mpi_resize(w, u->nlimbs + 1))
		return -ENOMEM;

	if (u->sign) {
		mpi_limb_t cy;

		cy = mpihelp_add_1(w->d, u->d, u->nlimbs, (mpi_limb_t) vval);
		w->d[u->nlimbs] = cy;
		w->nlimbs = u->nlimbs + cy;
		w->sign = 1;
	} else {
		/* The signs are different.  Need exact comparison to determine
		 * which operand to subtract from which.
		 */
		if (u->nlimbs == 1 && u->d[0] < vval) {
			w->d[0] = vval - u->d[0];
			w->nlimbs = 1;
			w->sign = 1;
		} else {
			mpihelp_sub_1(w->d, u->d, u->nlimbs, (mpi_limb_t) vval);
			/* Size can decrease with at most one limb. */
			w->nlimbs = (u->nlimbs - (w->d[u->nlimbs - 1] == 0));
			w->sign = 0;
		}
	}

	mpi_normalize(w);
	return 0;
}
EXPORT_SYMBOL_GPL(mpi_sub_ui);
