← Back to branch summary

~vcs-imports/samba/main

« back to all changes in this revision

Viewing changes to source/python/examples/tdbpack/oldtdbutil.py

  • Committer: jerry
  • Date: 2006-07-14 21:48:39 UTC
  • Revision ID: vcs-imports@canonical.com-20060714214839-586d8c489a8fcead
gutting trunk to move to svn:externals

Show diffs side-by-side

added added

removed removed

Lines of Context:
source/python/examples/tdbpack/oldtdbutil.py
1
 
#!/usr/bin/python
2
 
#############################################################
3
 
# tdbutil
4
 
# 
5
 
# Purpose:
6
 
#   Contains functions that are used to pack and unpack data
7
 
# from Samba's tdb databases.  Samba sometimes represents complex
8
 
# data structures as a single value in a database.  These functions
9
 
# allow other python scripts to package data types into a single python
10
 
# string and unpackage them.
11
 
#
12
 
#
13
 
# XXXXX: This code is no longer used; it's just here for testing
14
 
# compatibility with the new (much faster) C implementation.
15
 
#
16
 
############################################################## 
17
 
import string
18
 
 
19
 
def pack(format,list):
20
 
   retstring = ''
21
 
   listind = 0
22
 
   
23
 
   # Cycle through format entries
24
 
   for type in format:
25
 
      # Null Terminated String
26
 
      if (type == 'f' or type == 'P'):
27
 
         retstring = retstring + list[listind] + "\000"
28
 
      # 4 Byte Number
29
 
      if (type == 'd'):
30
 
         retstring = retstring + PackNum(list[listind],4)
31
 
      # 2 Byte Number
32
 
      if (type == 'w'):
33
 
         retstring = retstring + PackNum(list[listind],2)
34
 
      # Pointer Value
35
 
      if (type == 'p'):
36
 
         if (list[listind]):
37
 
            retstring = retstring + PackNum(1,4)
38
 
         else:
39
 
            retstring = retstring + PackNum(0,4)
40
 
      # Buffer and Length
41
 
      if (type == 'B'):
42
 
         # length
43
 
         length = list[listind]
44
 
         retstring = retstring + PackNum(length,4)
45
 
         length = int(length)
46
 
         listind = listind + 1
47
 
         # buffer
48
 
         retstring = retstring + list[listind][:length]
49
 
         
50
 
      listind = listind + 1
51
 
      
52
 
   return retstring
53
 
 
54
 
def unpack(format,buffer):
55
 
   retlist = []
56
 
   bufind = 0
57
 
   
58
 
   lasttype = ""
59
 
   for type in format:
60
 
      # Pointer Value
61
 
      if (type == 'p'):
62
 
         newvalue = UnpackNum(buffer[bufind:bufind+4])
63
 
         bufind = bufind + 4
64
 
         if (newvalue):
65
 
            newvalue = 1L
66
 
         else:
67
 
            newvalue = 0L
68
 
         retlist.append(newvalue)
69
 
      # Previous character till end of data
70
 
      elif (type == '$'):
71
 
         if (lasttype == 'f'):
72
 
            while (bufind < len(buffer)):
73
 
               newstring = ''
74
 
               while (buffer[bufind] != '\000'):
75
 
                  newstring = newstring + buffer[bufind]
76
 
                  bufind = bufind + 1
77
 
               bufind = bufind + 1
78
 
               retlist.append(newstring)
79
 
      # Null Terminated String
80
 
      elif (type == 'f' or type == 'P'):
81
 
         newstring = ''
82
 
         while (buffer[bufind] != '\000'):
83
 
            newstring = newstring + buffer[bufind]
84
 
            bufind = bufind + 1
85
 
         bufind = bufind + 1
86
 
         retlist.append(newstring)
87
 
      # 4 Byte Number
88
 
      elif (type == 'd'):
89
 
         newvalue = UnpackNum(buffer[bufind:bufind+4])
90
 
         bufind = bufind + 4
91
 
         retlist.append(newvalue)
92
 
      # 2 Byte Number
93
 
      elif (type == 'w'):
94
 
         newvalue = UnpackNum(buffer[bufind:bufind+2])
95
 
         bufind = bufind + 2
96
 
         retlist.append(newvalue)
97
 
      # Length and Buffer
98
 
      elif (type == 'B'):
99
 
         # Length
100
 
         length = UnpackNum(buffer[bufind:bufind+4])
101
 
         bufind = bufind + 4 
102
 
         retlist.append(length)
103
 
         length = int(length)
104
 
         # Buffer
105
 
         retlist.append(buffer[bufind:bufind+length])
106
 
         bufind = bufind + length
107
 
         
108
 
      lasttype = type
109
 
 
110
 
   return ((retlist,buffer[bufind:]))
111
 
 
112
 
def PackNum(myint,size):
113
 
    retstring = ''
114
 
    size = size * 2
115
 
    hint = hex(myint)[2:]
116
 
 
117
 
    # Check for long notation
118
 
    if (hint[-1:] == 'L'):
119
 
       hint = hint[:-1]
120
 
    
121
 
    addon = size - len(hint)
122
 
    for i in range(0,addon):
123
 
       hint = '0' + hint
124
 
    
125
 
    while (size > 0):
126
 
       val = string.atoi(hint[size-2:size],16)
127
 
       retstring = retstring + chr(val)
128
 
       size = size - 2
129
 
    
130
 
    return retstring
131
 
   
132
 
def UnpackNum(buffer):
133
 
   size = len(buffer)
134
 
   mystring = ''
135
 
 
136
 
   for i in range(size-1,-1,-1):
137
 
      val = hex(ord(buffer[i]))[2:]
138
 
      if (len(val) == 1):
139
 
         val = '0' + val
140
 
      mystring = mystring + val
141
 
   if (len(mystring) > 4):
142
 
      return string.atol(mystring,16)
143
 
   else:
144
 
      return string.atoi(mystring,16)