message.c 3.7 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159
  1. /* See http://www.python-ldap.org/ for details.
  2. * $Id: message.c,v 1.16 2009/08/17 01:49:47 leonard Exp $ */
  3. #include "common.h"
  4. #include "message.h"
  5. #include "berval.h"
  6. #include "errors.h"
  7. /*
  8. * Converts an LDAP message into a Python structure.
  9. *
  10. * On success, returns a list of dictionaries.
  11. * On failure, returns NULL, and sets an error.
  12. *
  13. * The message m is always freed, regardless of return value.
  14. */
  15. PyObject *
  16. LDAPmessage_to_python(LDAP *ld, LDAPMessage *m)
  17. {
  18. /* we convert an LDAP message into a python structure.
  19. * It is always a list of dictionaries.
  20. * We always free m.
  21. */
  22. PyObject* result;
  23. LDAPMessage* entry;
  24. result = PyList_New(0);
  25. if (result == NULL) {
  26. ldap_msgfree( m );
  27. return NULL;
  28. }
  29. for(entry = ldap_first_entry(ld,m);
  30. entry != NULL;
  31. entry = ldap_next_entry(ld,entry))
  32. {
  33. char *dn;
  34. char *attr;
  35. BerElement *ber = NULL;
  36. PyObject* entrytuple;
  37. PyObject* attrdict;
  38. dn = ldap_get_dn( ld, entry );
  39. if (dn == NULL) {
  40. Py_DECREF(result);
  41. ldap_msgfree( m );
  42. return LDAPerror( ld, "ldap_get_dn" );
  43. }
  44. attrdict = PyDict_New();
  45. if (attrdict == NULL) {
  46. Py_DECREF(result);
  47. ldap_msgfree( m );
  48. ldap_memfree(dn);
  49. return NULL;
  50. }
  51. /* Fill attrdict with lists */
  52. for( attr = ldap_first_attribute( ld, entry, &ber );
  53. attr != NULL;
  54. attr = ldap_next_attribute( ld, entry, ber )
  55. ) {
  56. PyObject* valuelist;
  57. struct berval ** bvals =
  58. ldap_get_values_len( ld, entry, attr );
  59. /* Find which list to append to */
  60. if ( PyMapping_HasKeyString( attrdict, attr ) ) {
  61. valuelist = PyMapping_GetItemString( attrdict, attr );
  62. } else {
  63. valuelist = PyList_New(0);
  64. if (valuelist != NULL && PyMapping_SetItemString(attrdict,
  65. attr, valuelist) == -1) {
  66. Py_DECREF(valuelist);
  67. valuelist = NULL; /* catch error later */
  68. }
  69. }
  70. if (valuelist == NULL) {
  71. Py_DECREF(attrdict);
  72. Py_DECREF(result);
  73. if (ber != NULL)
  74. ber_free(ber, 0);
  75. ldap_msgfree( m );
  76. ldap_memfree(attr);
  77. ldap_memfree(dn);
  78. return NULL;
  79. }
  80. if (bvals != NULL) {
  81. Py_ssize_t i;
  82. for (i=0; bvals[i]; i++) {
  83. PyObject *valuestr;
  84. valuestr = LDAPberval_to_object(bvals[i]);
  85. if (PyList_Append( valuelist, valuestr ) == -1) {
  86. Py_DECREF(attrdict);
  87. Py_DECREF(result);
  88. Py_DECREF(valuestr);
  89. Py_DECREF(valuelist);
  90. if (ber != NULL)
  91. ber_free(ber, 0);
  92. ldap_msgfree( m );
  93. ldap_memfree(attr);
  94. ldap_memfree(dn);
  95. return NULL;
  96. }
  97. Py_DECREF(valuestr);
  98. }
  99. ldap_value_free_len(bvals);
  100. }
  101. Py_DECREF( valuelist );
  102. ldap_memfree(attr);
  103. }
  104. entrytuple = Py_BuildValue("(sO)", dn, attrdict);
  105. ldap_memfree(dn);
  106. Py_DECREF(attrdict);
  107. PyList_Append(result, entrytuple);
  108. Py_DECREF(entrytuple);
  109. if (ber != NULL)
  110. ber_free(ber, 0);
  111. }
  112. for(entry = ldap_first_reference(ld,m);
  113. entry != NULL;
  114. entry = ldap_next_reference(ld,entry))
  115. {
  116. char **refs = NULL;
  117. PyObject* entrytuple;
  118. PyObject* reflist = PyList_New(0);
  119. if (reflist == NULL) {
  120. Py_DECREF(result);
  121. ldap_msgfree( m );
  122. return NULL;
  123. }
  124. if (ldap_parse_reference(ld, entry, &refs, NULL, 0) != LDAP_SUCCESS) {
  125. Py_DECREF(result);
  126. ldap_msgfree( m );
  127. return LDAPerror( ld, "ldap_parse_reference" );
  128. }
  129. if (refs) {
  130. Py_ssize_t i;
  131. for (i=0; refs[i] != NULL; i++) {
  132. PyObject *refstr = PyString_FromString(refs[i]);
  133. PyList_Append(reflist, refstr);
  134. Py_DECREF(refstr);
  135. }
  136. ber_memvfree( (void **) refs );
  137. }
  138. entrytuple = Py_BuildValue("(sO)", NULL, reflist);
  139. Py_DECREF(reflist);
  140. PyList_Append(result, entrytuple);
  141. Py_DECREF(entrytuple);
  142. }
  143. ldap_msgfree( m );
  144. return result;
  145. }