From: Mark Wooding <mdw@distorted.org.uk>
Date: Sat, 23 Nov 2019 14:22:30 +0000 (+0000)
Subject: buffer.c: Refactor `WriteBuffer.putecptraw'.
X-Git-Url: https://git.distorted.org.uk/~mdw/catacomb-python/commitdiff_plain/4e5cc42b1384e02fcbdd2c5673f8a33e9fd807b9

buffer.c: Refactor `WriteBuffer.putecptraw'.

This will better match a function to be added soon.  No functional
change.
---

diff --git a/buffer.c b/buffer.c
index 0d3d165..c616403 100644
--- a/buffer.c
+++ b/buffer.c
@@ -465,12 +465,14 @@ static PyObject *wbmeth_putecpt(PyObject *me, PyObject *arg)
 static PyObject *wbmeth_putecptraw(PyObject *me, PyObject *arg)
 {
   PyObject *ptobj;
+  ec_curve *cc;
   ec pt = EC_INIT;
   if (!PyArg_ParseTuple(arg, "O!:putecptraw", ecptcurve_pytype, &ptobj))
     return (0);
-  EC_OUT(ECPT_C(ptobj), &pt, ECPT_P(ptobj));
-  if (ensurebuf(me, ECPT_C(ptobj)->f->noctets * 2 + 1)) return (0);
-  ec_putraw(ECPT_C(ptobj), BUF_B(me), &pt); assert(BOK(BUF_B(me)));
+  cc = ECPT_C(ptobj);
+  EC_OUT(cc, &pt, ECPT_P(ptobj));
+  if (ensurebuf(me, 2*cc->f->noctets + 1)) return (0);
+  ec_putraw(cc, BUF_B(me), &pt); assert(BOK(BUF_B(me)));
   EC_DESTROY(&pt);
   RETURN_ME;
 }