limit the size of the uids cookie

adapters/conversant/conversant_test.go

@@ -691,7 +691,7 @@ func ParseRequest(req *pbs.PBSRequest) (*pbs.PBSRequest, error) {
 	// Need to pass the conversant user id thru uid cookie
 
 	httpReq := httptest.NewRequest("POST", "/foo", body)
-	cookie := usersync.NewPBSCookie()
+	cookie := usersync.NewPBSCookie(0)
 	_ = cookie.TrySync("conversant", ExpectedBuyerUID)
 	httpReq.Header.Set("Cookie", cookie.ToHTTPCookie(90*24*time.Hour).String())
 	httpReq.Header.Add("Referer", "http://example.com")

adapters/openrtb_util_test.go

@@ -341,7 +341,7 @@ func TestOpenRTBEmptyUser(t *testing.T) {
 }
 
 func TestOpenRTBUserWithCookie(t *testing.T) {
-	pbsCookie := usersync.NewPBSCookie()
+	pbsCookie := usersync.NewPBSCookie(0)
 	pbsCookie.TrySync("test", "abcde")
 	pbReq := pbs.PBSRequest{
 		User: &openrtb.User{},

config/config.go

@@ -168,12 +168,13 @@ type FileLogs struct {
 }
 
 type HostCookie struct {
-	Domain       string `mapstructure:"domain"`
-	Family       string `mapstructure:"family"`
-	CookieName   string `mapstructure:"cookie_name"`
-	OptOutURL    string `mapstructure:"opt_out_url"`
-	OptInURL     string `mapstructure:"opt_in_url"`
-	OptOutCookie Cookie `mapstructure:"optout_cookie"`
+	Domain             string `mapstructure:"domain"`
+	Family             string `mapstructure:"family"`
+	CookieName         string `mapstructure:"cookie_name"`
+	OptOutURL          string `mapstructure:"opt_out_url"`
+	OptInURL           string `mapstructure:"opt_in_url"`
+	MaxCookieSizeBytes int    `mapstructure:"max_cookie_size_bytes"`
+	OptOutCookie       Cookie `mapstructure:"optout_cookie"`
 	// Cookie timeout in days
 	TTL int64 `mapstructure:"ttl_days"`
 }
@@ -500,6 +501,7 @@ func SetupViper(v *viper.Viper, filename string) {
 	v.SetDefault("host_cookie.optout_cookie.name", "")
 	v.SetDefault("host_cookie.value", "")
 	v.SetDefault("host_cookie.ttl_days", 90)
+	v.SetDefault("host_cookie.max_cookie_size_bytes", 0)
 	v.SetDefault("http_client.max_idle_connections", 400)
 	v.SetDefault("http_client.max_idle_connections_per_host", 10)
 	v.SetDefault("http_client.idle_connection_timeout_seconds", 60)

config/config_test.go

@@ -22,6 +22,7 @@ func TestDefaults(t *testing.T) {
 	cmpInts(t, "auction_timeouts_ms.max", int(cfg.AuctionTimeouts.Max), 0)
 	cmpInts(t, "max_request_size", int(cfg.MaxRequestSize), 1024*256)
 	cmpInts(t, "host_cookie.ttl_days", int(cfg.HostCookie.TTL), 90)
+	cmpInts(t, "host_cookie.max_cookie_size_bytes", cfg.HostCookie.MaxCookieSizeBytes, 0)
 	cmpStrings(t, "datacache.type", cfg.DataCache.Type, "dummy")
 	cmpStrings(t, "adapters.pubmatic.endpoint", cfg.Adapters[string(openrtb_ext.BidderPubmatic)].Endpoint, "http://hbopenbid.pubmatic.com/translator?source=prebid-server")
 	cmpInts(t, "currency_converter.fetch_interval_seconds", cfg.CurrencyConverter.FetchIntervalSeconds, 1800)
@@ -39,6 +40,7 @@ host_cookie:
   domain: cookies.prebid.org
   opt_out_url: http://prebid.org/optout
   opt_in_url: http://prebid.org/optin
+  max_cookie_size_bytes: 32768
 external_url: http://prebid-server.prebid.org/
 host: prebid-server.prebid.org
 port: 1234

endpoints/cookie_sync_test.go

@@ -127,7 +127,7 @@ func doConfigurablePost(body string, existingSyncs map[string]string, gdprHostCo
 	router.POST("/cookie_sync", endpoint)
 	req, _ := http.NewRequest("POST", "/cookie_sync", strings.NewReader(body))
 	if len(existingSyncs) > 0 {
-		pcs := usersync.NewPBSCookie()
+		pcs := usersync.NewPBSCookie(0)
 		for bidder, uid := range existingSyncs {
 			pcs.TrySync(bidder, uid)
 		}

endpoints/setuid_test.go

@@ -83,7 +83,7 @@ func TestBadRequests(t *testing.T) {
 
 func TestOptedOut(t *testing.T) {
 	request := httptest.NewRequest("GET", "/setuid?bidder=pubmatic&uid=123", nil)
-	cookie := usersync.NewPBSCookie()
+	cookie := usersync.NewPBSCookie(0)
 	cookie.SetPreference(false)
 	addCookie(request, cookie)
 	response := doRequest(request, true, false)
@@ -111,7 +111,7 @@ func assertBadRequest(t *testing.T, uri string, errMsg string) {
 func makeRequest(uri string, existingSyncs map[string]string) *http.Request {
 	request := httptest.NewRequest("GET", uri, nil)
 	if len(existingSyncs) > 0 {
-		pbsCookie := usersync.NewPBSCookie()
+		pbsCookie := usersync.NewPBSCookie(0)
 		for family, value := range existingSyncs {
 			pbsCookie.TrySync(family, value)
 		}
@@ -146,7 +146,7 @@ func parseCookieString(t *testing.T, response *httptest.ResponseRecorder) *users
 		Name:  "uids",
 		Value: res[1],
 	}
-	return usersync.ParsePBSCookie(&httpCookie)
+	return usersync.ParsePBSCookie(&httpCookie, 0)
 }
 
 func assertIntsMatch(t *testing.T, expected int, actual int) {

exchange/legacy.go

@@ -100,7 +100,7 @@ func (bidder *adaptedAdapter) toLegacyRequest(req *openrtb.BidRequest) (*pbs.PBS
 		domain = req.Site.Domain
 	}
 
-	cookie := usersync.NewPBSCookie()
+	cookie := usersync.NewPBSCookie(0)
 	if req.User != nil {
 		if req.User.BuyerUID != "" {
 			cookie.TrySync(bidder.adapter.Name(), req.User.BuyerUID)

usersync/cookie.go

@@ -4,6 +4,7 @@ import (
 	"encoding/base64"
 	"encoding/json"
 	"errors"
+	"math"
 	"net/http"
 	"time"
 
@@ -30,9 +31,10 @@ var bidderToFamilyNames = map[openrtb_ext.BidderName]string{
 // To get an instance of this from a request, use ParsePBSCookieFromRequest.
 // To write an instance onto a response, use SetCookieOnResponse.
 type PBSCookie struct {
-	uids     map[string]uidWithExpiry
-	optOut   bool
-	birthday *time.Time
+	uids         map[string]uidWithExpiry
+	optOut       bool
+	birthday     *time.Time
+	maxSizeBytes int
 }
 
 // uidWithExpiry bundles the UID with an Expiration date.
@@ -49,17 +51,17 @@ func ParsePBSCookieFromRequest(r *http.Request, cookie *config.HostCookie) *PBSC
 	if cookie.OptOutCookie.Name != "" {
 		optOutCookie, err1 := r.Cookie(cookie.OptOutCookie.Name)
 		if err1 == nil && optOutCookie.Value == cookie.OptOutCookie.Value {
-			pc := NewPBSCookie()
+			pc := NewPBSCookie(cookie.MaxCookieSizeBytes)
 			pc.SetPreference(false)
 			return pc
 		}
 	}
 	var parsed *PBSCookie
 	uidCookie, err2 := r.Cookie(UID_COOKIE_NAME)
 	if err2 == nil {
-		parsed = ParsePBSCookie(uidCookie)
+		parsed = ParsePBSCookie(uidCookie, cookie.MaxCookieSizeBytes)
 	} else {
-		parsed = NewPBSCookie()
+		parsed = NewPBSCookie(cookie.MaxCookieSizeBytes)
 	}
 	// Fixes #582
 	if uid, _, _ := parsed.GetUID(cookie.Family); uid == "" && cookie.CookieName != "" {
@@ -71,8 +73,8 @@ func ParsePBSCookieFromRequest(r *http.Request, cookie *config.HostCookie) *PBSC
 }
 
 // ParsePBSCookie parses the UserSync cookie from a raw HTTP cookie.
-func ParsePBSCookie(uidCookie *http.Cookie) *PBSCookie {
-	pc := NewPBSCookie()
+func ParsePBSCookie(uidCookie *http.Cookie, MaxSizeBytes int) *PBSCookie {
+	pc := NewPBSCookie(MaxSizeBytes)
 
 	j, err := base64.URLEncoding.DecodeString(uidCookie.Value)
 	if err != nil {
@@ -87,10 +89,11 @@ func ParsePBSCookie(uidCookie *http.Cookie) *PBSCookie {
 }
 
 // NewPBSCookie returns an empty PBSCookie
-func NewPBSCookie() *PBSCookie {
+func NewPBSCookie(MaxSizeBytes int) *PBSCookie {
 	return &PBSCookie{
-		uids:     make(map[string]uidWithExpiry),
-		birthday: timestamp(),
+		uids:         make(map[string]uidWithExpiry),
+		birthday:     timestamp(),
+		maxSizeBytes: MaxSizeBytes,
 	}
 }
 
@@ -162,9 +165,30 @@ func (cookie *PBSCookie) GetId(bidderName openrtb_ext.BidderName) (id string, ex
 // SetCookieOnResponse is a shortcut for "ToHTTPCookie(); cookie.setDomain(domain); setCookie(w, cookie)"
 func (cookie *PBSCookie) SetCookieOnResponse(w http.ResponseWriter, domain string, ttl time.Duration) {
 	httpCookie := cookie.ToHTTPCookie(ttl)
+	var now time.Time = time.Now()
+
 	if domain != "" {
 		httpCookie.Domain = domain
 	}
+
+	var currSize int = len([]byte(httpCookie.String()))
+	for cookie.maxSizeBytes > 0 && currSize > cookie.maxSizeBytes {
+		var oldestElem string = ""
+		var oldestDate int64 = math.MaxInt64
+		for key, value := range cookie.uids {
+			timeUntilExpiration := value.Expires.Sub(now)
+			if timeUntilExpiration < time.Duration(oldestDate) {
+				oldestElem = key
+				oldestDate = int64(timeUntilExpiration)
+			}
+		}
+		delete(cookie.uids, oldestElem)
+		httpCookie = cookie.ToHTTPCookie(ttl)
+		if domain != "" {
+			httpCookie.Domain = domain
+		}
+		currSize = len([]byte(httpCookie.String()))
+	}
 	http.SetCookie(w, httpCookie)
 }