torsava / rpms / python3

Forked from rpms/python3 6 years ago
Clone
aa2fa3b
--- python/branches/py3k/Modules/audioop.c	2010/05/09 15:52:27	81032
aa2fa3b
+++ python/branches/py3k/Modules/audioop.c	2010/05/11 13:09:58	81081
aa2fa3b
@@ -834,7 +834,7 @@
aa2fa3b
 audioop_tostereo(PyObject *self, PyObject *args)
aa2fa3b
 {
aa2fa3b
     signed char *cp, *ncp;
aa2fa3b
-    int len, new_len, size, val1, val2, val = 0;
aa2fa3b
+    int len, size, val1, val2, val = 0;
aa2fa3b
     double fac1, fac2, fval, maxval;
aa2fa3b
     PyObject *rv;
aa2fa3b
     int i;
aa2fa3b
@@ -851,14 +851,13 @@
aa2fa3b
         return 0;
aa2fa3b
     }
aa2fa3b
 
aa2fa3b
-    new_len = len*2;
aa2fa3b
-    if (new_len < 0) {
aa2fa3b
+    if (len > INT_MAX/2) {
aa2fa3b
         PyErr_SetString(PyExc_MemoryError,
aa2fa3b
                         "not enough memory for output buffer");
aa2fa3b
         return 0;
aa2fa3b
     }
aa2fa3b
 
aa2fa3b
-    rv = PyBytes_FromStringAndSize(NULL, new_len);
aa2fa3b
+    rv = PyBytes_FromStringAndSize(NULL, len*2);
aa2fa3b
     if ( rv == 0 )
aa2fa3b
         return 0;
aa2fa3b
     ncp = (signed char *)PyBytes_AsString(rv);
aa2fa3b
@@ -1021,7 +1020,7 @@
aa2fa3b
 {
aa2fa3b
     signed char *cp;
aa2fa3b
     unsigned char *ncp;
aa2fa3b
-    int len, new_len, size, size2, val = 0;
aa2fa3b
+    int len, size, size2, val = 0;
aa2fa3b
     PyObject *rv;
aa2fa3b
     int i, j;
aa2fa3b
 
aa2fa3b
@@ -1035,13 +1034,12 @@
aa2fa3b
         return 0;
aa2fa3b
     }
aa2fa3b
 
aa2fa3b
-    new_len = (len/size)*size2;
aa2fa3b
-    if (new_len < 0) {
aa2fa3b
+    if (len/size > INT_MAX/size2) {
aa2fa3b
         PyErr_SetString(PyExc_MemoryError,
aa2fa3b
                         "not enough memory for output buffer");
aa2fa3b
         return 0;
aa2fa3b
     }
aa2fa3b
-    rv = PyBytes_FromStringAndSize(NULL, new_len);
aa2fa3b
+    rv = PyBytes_FromStringAndSize(NULL, (len/size)*size2);
aa2fa3b
     if ( rv == 0 )
aa2fa3b
         return 0;
aa2fa3b
     ncp = (unsigned char *)PyBytes_AsString(rv);
aa2fa3b
@@ -1077,7 +1075,6 @@
aa2fa3b
     int chan, d, *prev_i, *cur_i, cur_o;
aa2fa3b
     PyObject *state, *samps, *str, *rv = NULL;
aa2fa3b
     int bytes_per_frame;
aa2fa3b
-    size_t alloc_size;
aa2fa3b
 
aa2fa3b
     weightA = 1;
aa2fa3b
     weightB = 0;
aa2fa3b
@@ -1120,14 +1117,13 @@
aa2fa3b
     inrate /= d;
aa2fa3b
     outrate /= d;
aa2fa3b
 
aa2fa3b
-    alloc_size = sizeof(int) * (unsigned)nchannels;
aa2fa3b
-    if (alloc_size < (unsigned)nchannels) {
aa2fa3b
+    if ((size_t)nchannels > PY_SIZE_MAX/sizeof(int)) {
aa2fa3b
         PyErr_SetString(PyExc_MemoryError,
aa2fa3b
                         "not enough memory for output buffer");
aa2fa3b
         return 0;
aa2fa3b
     }
aa2fa3b
-    prev_i = (int *) malloc(alloc_size);
aa2fa3b
-    cur_i = (int *) malloc(alloc_size);
aa2fa3b
+    prev_i = (int *) malloc(nchannels * sizeof(int));
aa2fa3b
+    cur_i = (int *) malloc(nchannels * sizeof(int));
aa2fa3b
     if (prev_i == NULL || cur_i == NULL) {
aa2fa3b
         (void) PyErr_NoMemory();
aa2fa3b
         goto exit;
aa2fa3b
@@ -1164,25 +1160,16 @@
aa2fa3b
            ceiling(len*outrate/inrate) output frames, and each frame
aa2fa3b
            requires bytes_per_frame bytes.  Computing this
aa2fa3b
            without spurious overflow is the challenge; we can
aa2fa3b
-           settle for a reasonable upper bound, though. */
aa2fa3b
-        int ceiling;   /* the number of output frames */
aa2fa3b
-        int nbytes;    /* the number of output bytes needed */
aa2fa3b
-        int q = len / inrate;
aa2fa3b
-        /* Now len = q * inrate + r exactly (with r = len % inrate),
aa2fa3b
-           and this is less than q * inrate + inrate = (q+1)*inrate.
aa2fa3b
-           So a reasonable upper bound on len*outrate/inrate is
aa2fa3b
-           ((q+1)*inrate)*outrate/inrate =
aa2fa3b
-           (q+1)*outrate.
aa2fa3b
-        */
aa2fa3b
-        ceiling = (q+1) * outrate;
aa2fa3b
-        nbytes = ceiling * bytes_per_frame;
aa2fa3b
-        /* See whether anything overflowed; if not, get the space. */
aa2fa3b
-        if (q+1 < 0 ||
aa2fa3b
-            ceiling / outrate != q+1 ||
aa2fa3b
-            nbytes / bytes_per_frame != ceiling)
aa2fa3b
+           settle for a reasonable upper bound, though, in this
aa2fa3b
+           case ceiling(len/inrate) * outrate. */
aa2fa3b
+
aa2fa3b
+        /* compute ceiling(len/inrate) without overflow */
aa2fa3b
+        int q = len > 0 ? 1 + (len - 1) / inrate : 0;
aa2fa3b
+        if (outrate > INT_MAX / q / bytes_per_frame)
aa2fa3b
             str = NULL;
aa2fa3b
         else
aa2fa3b
-            str = PyBytes_FromStringAndSize(NULL, nbytes);
aa2fa3b
+            str = PyBytes_FromStringAndSize(NULL,
aa2fa3b
+                                            q * outrate * bytes_per_frame);
aa2fa3b
 
aa2fa3b
         if (str == NULL) {
aa2fa3b
             PyErr_SetString(PyExc_MemoryError,
aa2fa3b
@@ -1300,7 +1287,7 @@
aa2fa3b
     unsigned char *cp;
aa2fa3b
     unsigned char cval;
aa2fa3b
     signed char *ncp;
aa2fa3b
-    int len, new_len, size, val;
aa2fa3b
+    int len, size, val;
aa2fa3b
     PyObject *rv;
aa2fa3b
     int i;
aa2fa3b
 
aa2fa3b
@@ -1313,18 +1300,17 @@
aa2fa3b
         return 0;
aa2fa3b
     }
aa2fa3b
 
aa2fa3b
-    new_len = len*size;
aa2fa3b
-    if (new_len < 0) {
aa2fa3b
+    if (len > INT_MAX/size) {
aa2fa3b
         PyErr_SetString(PyExc_MemoryError,
aa2fa3b
                         "not enough memory for output buffer");
aa2fa3b
         return 0;
aa2fa3b
     }
aa2fa3b
-    rv = PyBytes_FromStringAndSize(NULL, new_len);
aa2fa3b
+    rv = PyBytes_FromStringAndSize(NULL, len*size);
aa2fa3b
     if ( rv == 0 )
aa2fa3b
         return 0;
aa2fa3b
     ncp = (signed char *)PyBytes_AsString(rv);
aa2fa3b
 
aa2fa3b
-    for ( i=0; i < new_len; i += size ) {
aa2fa3b
+    for ( i=0; i < len*size; i += size ) {
aa2fa3b
         cval = *cp++;
aa2fa3b
         val = st_ulaw2linear16(cval);
aa2fa3b
 
aa2fa3b
@@ -1374,7 +1360,7 @@
aa2fa3b
     unsigned char *cp;
aa2fa3b
     unsigned char cval;
aa2fa3b
     signed char *ncp;
aa2fa3b
-    int len, new_len, size, val;
aa2fa3b
+    int len, size, val;
aa2fa3b
     PyObject *rv;
aa2fa3b
     int i;
aa2fa3b
 
aa2fa3b
@@ -1387,18 +1373,17 @@
aa2fa3b
         return 0;
aa2fa3b
     }
aa2fa3b
 
aa2fa3b
-    new_len = len*size;
aa2fa3b
-    if (new_len < 0) {
aa2fa3b
+    if (len > INT_MAX/size) {
aa2fa3b
         PyErr_SetString(PyExc_MemoryError,
aa2fa3b
                         "not enough memory for output buffer");
aa2fa3b
         return 0;
aa2fa3b
     }
aa2fa3b
-    rv = PyBytes_FromStringAndSize(NULL, new_len);
aa2fa3b
+    rv = PyBytes_FromStringAndSize(NULL, len*size);
aa2fa3b
     if ( rv == 0 )
aa2fa3b
         return 0;
aa2fa3b
     ncp = (signed char *)PyBytes_AsString(rv);
aa2fa3b
 
aa2fa3b
-    for ( i=0; i < new_len; i += size ) {
aa2fa3b
+    for ( i=0; i < len*size; i += size ) {
aa2fa3b
         cval = *cp++;
aa2fa3b
         val = st_alaw2linear16(cval);
aa2fa3b
 
aa2fa3b
@@ -1523,7 +1508,7 @@
aa2fa3b
 {
aa2fa3b
     signed char *cp;
aa2fa3b
     signed char *ncp;
aa2fa3b
-    int len, new_len, size, valpred, step, delta, index, sign, vpdiff;
aa2fa3b
+    int len, size, valpred, step, delta, index, sign, vpdiff;
aa2fa3b
     PyObject *rv, *str, *state;
aa2fa3b
     int i, inputbuffer = 0, bufferstep;
aa2fa3b
 
aa2fa3b
@@ -1545,13 +1530,12 @@
aa2fa3b
     } else if ( !PyArg_ParseTuple(state, "ii", &valpred, &index) )
aa2fa3b
         return 0;
aa2fa3b
 
aa2fa3b
-    new_len = len*size*2;
aa2fa3b
-    if (new_len < 0) {
aa2fa3b
+    if (len > (INT_MAX/2)/size) {
aa2fa3b
         PyErr_SetString(PyExc_MemoryError,
aa2fa3b
                         "not enough memory for output buffer");
aa2fa3b
         return 0;
aa2fa3b
     }
aa2fa3b
-    str = PyBytes_FromStringAndSize(NULL, new_len);
aa2fa3b
+    str = PyBytes_FromStringAndSize(NULL, len*size*2);
aa2fa3b
     if ( str == 0 )
aa2fa3b
         return 0;
aa2fa3b
     ncp = (signed char *)PyBytes_AsString(str);
aa2fa3b
@@ -1559,7 +1543,7 @@
aa2fa3b
     step = stepsizeTable[index];
aa2fa3b
     bufferstep = 0;
aa2fa3b
 
aa2fa3b
-    for ( i=0; i < new_len; i += size ) {
aa2fa3b
+    for ( i=0; i < len*size*2; i += size ) {
aa2fa3b
         /* Step 1 - get the delta value and compute next index */
aa2fa3b
         if ( bufferstep ) {
aa2fa3b
             delta = inputbuffer & 0xf;